On the Sample Complexity of Differentially Private Policy Optimization
Yi He, Xingyu Zhou
Abstract
Policy optimization (PO) is a cornerstone of modern reinforcement learning (RL), with diverse applications spanning robotics, healthcare, and large language model training. The increasing deployment of PO in sensitive domains, however, raises significant privacy concerns. In this paper, we initiate a theoretical study of differentially private policy optimization, focusing explicitly on its sample complexity. We first formalize an appropriate definition of differential privacy (DP) tailored to PO, addressing the inherent challenges arising from on-policy learning dynamics and the subtlety involved in defining the unit of privacy. We then systematically analyze the sample complexity of widely-used PO algorithms, including policy gradient (PG), natural policy gradient (NPG) and more, under DP constraints and various settings, via a unified framework. Our theoretical results demonstrate that privacy costs can often manifest as lower-order terms in the sample complexity, while also highlighting subtle yet important observations in private PO settings. These offer valuable practical insights for privacy-preserving PO algorithms.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 9052edc9-fe29-49d4-9751-6a2bc3c1cac4Cited by top-tier papers2
- Improved Bounds for Private and Robust AlignmentWenqian Weng, Yi He, Xingyu ZhouICML 2026 · 3 citations
- Private Direct Preference Optimization for LLM AlignmentYangfan Jiang, Fei Wei, Ergute Bao, Xiaokui Xiao et al.CCS 2026
Builds on29
- Training language models to follow instructions with human feedbackLong Ouyang, Jeffrey Wu, Xu Jiang, Diogo Almeida et al.NeurIPS 2022 · 24,707 citations
- Direct Preference Optimization: Your Language Model is Secretly a Reward ModelRafael Rafailov, Archit Sharma, Eric Mitchell, Christopher D. Manning et al.NeurIPS 2023 · 10,924 citations
- Provably Efficient Exploration in Policy OptimizationQi Cai, Zhuoran Yang, Chi Jin, Zhaoran WangICML 2020 · 304 citations
- Neural Policy Gradient Methods: Global Optimality and Rates of ConvergenceLingxiao Wang, Qi Cai, Zhuoran Yang, Zhaoran WangICLR 2020 · 270 citations
- Adaptive Trust Region Policy Optimization: Global Convergence and Faster Rates for Regularized MDPsLior Shani, Yonathan Efroni, Shie MannorAAAI 2020 · 201 citations
Related papers
- Differentially Private Regret Minimization in Episodic Markov Decision ProcessesSayak Ray Chowdhury, Xingyu ZhouAAAI 2022 · 26 citations
- Private Reinforcement Learning with PAC and Regret GuaranteesGiuseppe Vietri, Borja Balle, Akshay Krishnamurthy, Zhiwei Steven WuICML 2020 · 70 citations
- Local Differential Privacy for Regret Minimization in Reinforcement LearningEvrard Garcelon, Vianney Perchet, Ciara Pike-Burke, Matteo PirottaNeurIPS 2021 · 47 citations
- Optimistic Natural Policy Gradient: a Simple Efficient Policy Optimization Framework for Online RLQinghua Liu, Gellért Weisz, András György, Chi Jin et al.NeurIPS 2023 · 16 citations
- Offline Reinforcement Learning with Differential PrivacyDan Qiao, Yu-Xiang WangNeurIPS 2023 · 34 citations
