Reward-Driven Learning under Prompt-Level Differential Privacy
cs.LG, cs.CR
Submitted: 2026-10-05
Updated: 2026-10-05
Code: https://github.com/google/differential-privacy
License: http://creativecommons.org/licenses/by/4.0/
The gist: Reinforcement learning with verifiable rewards (RLVR) trains a language model on problems that may themselves be confidential, and the trained model can reveal which problems it saw.
Terminology
Abstract
Reinforcement learning with verifiable rewards (RLVR) trains a language model on problems that may themselves be confidential, and the trained model can reveal which problems it saw. We study RLVR under prompt-level differential privacy: the released weights must be (ε,δ)-differentially private with respect to the presence of any one training problem. Taking the group of responses to one prompt as the privacy record, our method aggregates their gradients, clips the prompt's contribution once, adds Gaussian noise, and composes the privacy loss across updates, so the budget depends on neither the number of responses per prompt nor the clipping norm; to our knowledge this is the first differential privacy guarantee for RLVR training. We train Qwen2.5-1.5B-Instruct with LoRA at a per-run budget of ε=8 and compare, on the same prompts and at the same budget, a control that removes only the reward signal and two private supervised fine-tuning recipes. The reward signal improves accuracy over the control by 2.65 points on MATH and 3.24 on GSM8K, in every seed; the improvement survives a format-robust scorer, at 1.3 points on MATH, and is not explained by response length. At the same budget the private model outperforms both supervised recipes on MATH and GSM8K by 2.3 to 3.8 points, retains 85--90% of the gain of non-private GRPO on these tasks, and on MATH the noise of an eightfold tighter budget costs at most 1.2 points. The reward effect also carries to CommonsenseQA, an exploratory non-mathematical task. Verifier feedback thus remains a usable learning signal under prompt-level privacy.
Sources
- DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models
- Auditing Data Membership in Reinforcement Learning With Verifiable Rewards
- Towards User-level Private Reinforcement Learning with Human Feedback
- Improved Algorithms for Differentially Private Language Model Alignment
- Differentially Private Policy Gradient
- Privacy-Preserving Instructions for Aligning Large Language Models
- Reinforcement Learning on Benign Facts Amplifies Leakage of Memorized Private Data
- KL-regularization Itself is Differentially Private in Bandits and RLHF
- Towards Differentially Private Reinforcement Learning with General Function Approximation
- DP-OPD: Differentially Private On-Policy Distillation for Language Models
- Efficient Differentially Private Fine-Tuning of LLMs via Reinforcement Learning
- GRASP: Reinforcing Language Model Anonymizers with Group Relative Policy Optimization
- Demystifying Group Relative Policy Optimization: Its Policy Gradient is a U-Statistic
- Unlocking High-Accuracy Differentially Private Image Classification through Scale
- Understanding R1-Zero-Like Training: A Critical Perspective
- Qwen2.5 Technical Report
- Scaling Relationship on Learning Mathematical Reasoning with Large Language Models
- Teaching Large Language Models to Reason with Reinforcement Learning
- Proximal Policy Optimization Algorithms
- On a few pitfalls in KL divergence gradient estimation for RL
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks