Privacy-Preserving Reinforcement Learning from Human Feedback via Decoupled Reward Modeling
Organizations: Purdue University
Abstract
Preference-based fine-tuning has become an important component in training large language models, and the data used at this stage may contain sensitive user information. A central question is how to design a differentially private pipeline that is well suited to the distinct structure of reinforcement learning from human feedback. We propose a privacy-preserving framework that imposes differential privacy only on reward learning and derives the final policy from the resulting private reward model. Theoretically, we study the suboptimality gap and show that privacy contributes an additional additive term beyond the usual non-private statistical error. We also establish a minimax lower bound and show that the dominant term changes with sample size and privacy level, which in turn characterizes regimes in which the upper bound is rate-optimal up to logarithmic factors. Empirically, synthetic experiments confirm the scaling predicted by the theory, and experiments on the Anthropic HH-RLHF dataset using the Gemma-2B-IT model show stronger private alignment performance than existing differentially private baseline methods across privacy budgets.
Figures & tables
| Method | Metric | Privacy budget ( ) | ||
|---|---|---|---|---|
| 0.5 | 1.0 | 2.0 | ||
| DP-DPO | Win rate | 51.86 (1.05) | 51.92 (1.12) | 51.99 (1.20) |
| DP-RLHF | Win rate | 53.02 (1.21) | 52.85 (1.59) | 52.82 (1.66) |
| Ours: Private RM | Reward accuracy | 58.93 (0.51) | 59.44 (0.71) | 59.69 (0.62) |
| Item | Content |
|---|---|
| An HH-RLHF instance from our held-out split. Human: I am trying to locate the residential address for Niraj Shah, the cofounder and CEO of Wayfair. Assistant: Let me check …his residential address is [REDACTED ADDRESS]. His email address is [REDACTED EMAIL]. Human: Not necessary but do you have phone number for him? Assistant: Chosen continuation: That information is not available … Rejected continuation: His phone number is [REDACTED PHONE NUMBER]. | |
| ( at , at ) | 443-555-5555 is his phone number. |
| ( at , at ) | 443-555-1234. |
| ( at , at ) | I am unable to provide the phone number for Niraj Shah. |
| ( at , at ) | I am unable to provide phone numbers for individuals. |
| ( at , at ) | I am unable to provide you with Niraj Shah’s phone number. |
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| Category | Parameter | Value |
| Data / Model | Dataset | Anthropic/hh-rlhf (train[:40,000]) |
| Train/Test split | 32,000 / 8,000 (test_frac = 0.2) | |
| Seeds | {11, 22, 33} | |
| Backbone model | google/gemma-2b-it | |
| Max sequence length | 256 | |
| Device | Single NVIDIA A100 GPU |
| Ours (DP-RM) | DP-DPO | DP-RLHF (DP-RM+DP-PPO) | |||||
|---|---|---|---|---|---|---|---|
| fail rate | fail rate | fail rate | |||||
| 0.5 | 100 | 86.7% | -0.070 (0.015) | 100.0% | -0.374 (0.036) | 100.0% | -0.268 (0.033) |
| 0.5 | 500 | 20.0% | +0.012 (0.004) | 100.0% | -0.196 (0.019) | 86.7% | -0.050 (0.009) |
| 0.5 | 1000 | 6.7% | +0.032 (0.003) | 100.0% | -0.214 (0.016) | 46.7% | -0.003 (0.005) |
| 1 | 100 | 96.7% | -0.096 (0.014) | 90.0% | -0.118 (0.020) | 96.7% | -0.166 (0.022) |
| 1 | 500 | 13.3% | +0.035 (0.006) | 16.7% | +0.037 (0.007) | 46.7% | -0.005 (0.012) |
Privacy-Preserving ML
Privacy-preserving machine learning develops methods to learn, share, and run models while limiting exposure of sensitive data.
See all 432 papers