Efficient Differentially Private Fine-Tuning of LLMs via Reinforcement Learning
Not on the map. Why?
Abstract (arXiv)
The tension between data privacy and model utility has become the defining bottleneck for the practical deployment of large language models (LLMs) trained on sensitive corpora including healthcare. Differentially private stochastic gradient descent (DP-SGD) guarantees formal privacy, yet it does so at a pronounced cost: gradients are forcibly clipped and perturbed with noise, degrading sample efficiency and final accuracy. Numerous variants have been proposed to soften this trade-off, but they all share a handicap: their control knobs are hard-coded, global, and oblivious to the evolving optimization landscape. Consequently, practitioners are forced either to over-spend privacy budget in pursuit of utility, or to accept mediocre models in order to stay within privacy constraints. We present RLDP, the first framework to cast DP optimization itself as a closed-loop control problem amenable to modern deep reinforcement learning (RL). RLDP continuously senses rich statistics of the learning dynamics and acts by selecting fine-grained per parameter gradient-clipping thresholds as well as the magnitude of injected Gaussian noise. A soft actor-critic (SAC) hyper-policy is trained online during language model fine-tuning; it learns, from scratch, how to allocate the privacy budget where it matters and when it matters. Across more than 1,600 ablation experiments on GPT2-small, Llama-1B, Llama-3B, and Mistral-7B, RLDP delivers perplexity reductions of 1.3-30.5% (mean 5.4%) and an average 5.6% downstream utility gain. RLDP reaches each baseline's final utility after only 13-43% of the gradient-update budget (mean speed-up 71%), all while honoring the same (ε, δ)-DP contract and exhibiting equal or lower susceptibility to membership-inference and canary-extraction attacks.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 17 of its 27 references that are in the map. First posted on arXiv 30 Jul 2025.
Builds on
17 of 27 references.
References · 17
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,885
- 2016Deep Learning with Differential Privacycited by 2,728
- 2021Membership Inference Attacks From First Principlescited by 878
- 2017Renyi Differential Privacycited by 650
- 2021Large Language Models Can Be Strong Differentially Private Learnerscited by 412
- 2018Scalable Private Learning with PATEcited by 269
- 2019Differentially Private Learning with Adaptive Clippingcited by 220
- 2022Automatic Clipping: Differentially Private Deep Learning Made Easier and Strongercited by 88
- 2019AdaCliP: Adaptive Clipping for Private SGDcited by 77
- 2017Multi-step Reinforcement Learning: A Unifying Algorithmcited by 42
- 2023Differentially Private Low-Rank Adaptation of Large Language Model Using Federated Learningcited by 41
5 more
- 2022Differentially Private Learning with Per-Sample Adaptive Clippingcited by 21
- 2022Wind Power Forecasting Considering Data Privacy Protection: A Federated Deep Reinforcement Learning Approachcited by 12
- 2025DC-SGD: Differentially Private SGD with Dynamic Clipping through Gradient Norm Distribution Estimationcited by 6
- 2021Efficient Hyperparameter Optimization for Differentially Private Deep Learningcited by 5
- 2025GeoClip: Geometry-Aware Clipping for Differentially Private SGDcited by 3
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2021An Efficient DP-SGD Mechanism for Large Scale NLP Models
- 2025On the Sample Complexity of Differentially Private Policy Optimization
- 2023DP-Forward: Fine-tuning and Inference on Language Models with Differential Privacy in Forward Pass
- 2025RL-Finetuned LLMs for Privacy-Preserving Synthetic Rewriting
- 2026Privacy-Preserving Reinforcement Learning from Human Feedback via Decoupled Reward Modeling
- 2026DP-OPD: Differentially Private On-Policy Distillation for Language Models
- 2023Privately Aligning Language Models with Reinforcement Learning
- 2025Differentially Private Policy Gradient