Improved Bounds for Private and Robust Alignment
Wenqian Weng, Yi He, Xingyu Zhou
Abstract
In this paper, we study the private and robust alignment of language models from a theoretical perspective by establishing upper bounds on the suboptimality gap in both offline and online settings. We consider preference labels subject to privacy constraints and/or adversarial corruption, and analyze two distinct interplays between them: privacy-first and corruption-first. For the privacy-only setting, we show that log loss with an MLE-style algorithm achieves near-optimal rates, in contrast to conventional wisdom. For the joint privacy-and-corruption setting, we first demonstrate that existing offline algorithms in fact provide stronger guarantees---simultaneously in terms of corruption level and privacy parameters---than previously known, which further yields improved bounds in the corruption-only regime. In addition, we also present the first set of results for private and robust online alignment. Our results are enabled by new uniform convergence guarantees for log loss and square loss under privacy and corruption, which we believe have broad applicability across learning theory and statistics.
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.
Cited by top-tier papers2
- Near-Optimal Regret for KL-Regularized Multi-Armed BanditsKaixuan Ji, Qingyue Zhao, Heyang Zhao, Qiwei Di et al.ICML 2026 · 3 citations
- Private Direct Preference Optimization for LLM AlignmentYangfan Jiang, Fei Wei, Ergute Bao, Xiaokui Xiao et al.CCS 2026
Builds on32
- 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
- Iterative Preference Learning from Human Feedback: Bridging Theory and Practice for RLHF under KL-constraintWei Xiong, Hanze Dong, Chenlu Ye, Ziqi Wang et al.ICML 2024 · 346 citations
- Principled Reinforcement Learning with Human Feedback from Pairwise or K-wise ComparisonsBanghua Zhu, Michael I. Jordan, Jiantao JiaoICML 2023 · 273 citations
- FLAMBE: Structural Complexity and Representation Learning of Low Rank MDPsAlekh Agarwal, Sham M. Kakade, Akshay Krishnamurthy, Wen SunNeurIPS 2020 · 271 citations
Related papers
- SquareχPO: Differentially Private and Robust χ2-Preference Optimization in Offline Direct AlignmentXingyu Zhou, Yulian Wu, Wenqian Weng, Francesco OrabonaICML 2025
- A Unified Theoretical Analysis of Private and Robust Offline Alignment: from RLHF to DPOXingyu Zhou, Yulian Wu, Francesco OrabonaICML 2025
- Locally Private and Robust Multi-Armed BanditsXingyu Zhou, Komo (Wei) ZhangNeurIPS 2024 · 5 citations
- Towards Robust Model-Based Reinforcement Learning Against Adversarial CorruptionChenlu Ye, Jiafan He, Quanquan Gu, Tong ZhangICML 2024 · 10 citations
- Robust and differentially private mean estimationXiyang Liu, Weihao Kong, Sham M. Kakade, Sewoong OhNeurIPS 2021 · 87 citations
