改进DPO对齐方法,防止偏好响应概率异常下降
C2-DPO: Constrained Controlled Direct Preference Optimization
- 从样本内KL约束出发重构优化目标,揭示DPO内在机制
- 发现原始DPO会使偏好与拒绝响应概率同时下降
- 引入概率质量守恒约束,提升对齐稳定性与效果
直接偏好优化(DPO)已成为解决人工智能对齐问题的有前景方法。本文提出两个反直觉观察:首先,我们证明DPO损失可由仅在样本内响应上定义KL约束的替代优化问题推导得出,不同于原始强化学习人类反馈(RLHF)中在整个分布上定义约束;其次,我们证明该替代优化问题存在一个意外性质——在最优策略下,偏好与拒绝响应的概率均趋于下降,这正是实践中DPO的表现。为控制此现象,我们设计一组约束,限制参考策略与目标策略间偏好与拒绝响应的概率质量转移。由此提出的算法称为受限可控直接偏好优化(C2-DPO),具有明确的RLHF解释意义。在标准偏好数据集上对多个语言模型进行对齐时,相较于原始DPO,C2-DPO展现出显著的实用改进。
原文摘要 · Abstract (English)
Direct preference optimization (\texttt{DPO}) has emerged as a promising approach for solving the alignment problem in AI. In this paper, we make two counter-intuitive observations about \texttt{DPO}. First, we show that \texttt{DPO} loss could be derived by starting from an alternative optimization problem that only defines the KL guardrail on in-sample responses, unlike the original RLHF problem where guardrails are defined on the entire distribution. Second, we prove a surprising property of this alternative optimization problem, namely that under its optimal policy, both preferred and rejected responses tend to decrease in probability, a phenomenon typically displayed by DPO in practice. To control this behavior, we propose a set of constraints designed to limit the displacement of probability mass between the preferred and rejected responses in the reference and target policies. The resulting algorithm, which we call Constrained Controlled DPO (\texttt{C2-DPO}), has a meaningful RLHF interpretation. By hedging against the displacement, \texttt{C2-DPO} provides practical improvements over vanilla \texttt{DPO} when aligning several language models using standard preference datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。