在隐私保护下研究人类反馈强化学习的离线与在线算法,理论证明其有效性。
Offline and Online KL-Regularized RLHF under Differential Privacy
- 基于悲观原则设计离线算法,结合差分隐私保护人类偏好标签。
- 离线场景下误差上界为 $\tilde{O}(1/[(e^ε-1)^2 n])$,在线场景下对数级累积损失。
- 首次在差分隐私下分析该问题,适用于注重数据隐私的对齐模型训练。
本文研究在人类反馈强化学习(RLHF)中引入KL正则化,并在人类偏好标签的局部差分隐私(ε-LDP)约束下,分别考虑离线与在线两种设置。离线情形下,基于悲观原则设计算法,在单策略集中性假设下,得到KL正则化目标的子优性差距为 $\tilde{O}(1/[(e^ε-1)^2 n])$,并给出匹配的下界证明其最优性,其中 $n$ 为样本数量。在线情形下,首次从理论上分析带差分隐私的KL正则化RLHF,设计基于乐观原则的算法,得到对数级遗憾界 $O(d_{\mathcal{F}}\log (N_{\mathcal{F}}\cdot T) /(e^ε-1)^2 )$,其中 $T$ 为总时间步,$N_{\mathcal{F}}$ 为奖励函数空间 $\mathcal{F}$ 的基数,$d_{\mathcal{F}}$ 为面向RLHF的变体弹性维数。分析结果还首次揭示了无隐私限制下的在线KL正则化RLHF的理论性质。我们实现了离线算法以验证理论结果,并开源代码于:https://github.com/rushil-thareja/PPKL-RLHF-Official。
原文摘要 · Abstract (English)
In this paper, we study the offline and online settings of reinforcement learning from human feedback (RLHF) with KL-regularization -- a widely used objective function in large language model alignment -- under the $ε$ local differential privacy ($ε$-LDP) model on the label of the human preference. In the offline setting, we design an algorithm based on the principle of pessimism and derive a new suboptimality gap of $\tilde{O}(1/[(e^ε-1)^2 n])$ on the KL-regularized objective under single-policy concentrability. We also prove its optimality by providing a matching lower bound where $n$ is the sample size. In the online setting, we are the first one to theoretically investigate the problem of KL-regularized RLHF with LDP. We design an optimism-based algorithm and derive a logarithmic regret bound of $O(d_{\mathcal{F}}\log (N_{\mathcal{F}}\cdot T) /(e^ε-1)^2 )$, where $T$ is the total time step, $N_{\mathcal{F}}$ is cardinality of the reward function space $\mathcal{F}$ and $d_{\mathcal{F}}$ is a variant of eluder dimension for RLHF. As a by-product of our analysis, our results also imply the first analysis for online KL-regularized RLHF without privacy. We implement our algorithm in the offline setting to verify our theoretical results and release our open source code at: https://github.com/rushil-thareja/PPKL-RLHF-Official.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。