arXiv:2502.13177cs.LGcs.AI2025-02NeurIPS被引 7

让偏好优化的惩罚力度随数据对自适应调整,提升模型对齐效果。

KL Penalty Control via Perturbation for Direct Preference Optimization

论文配图:KL Penalty Control via Perturbation for Direct Preference Optimization
图 1 · 摘自论文原文
  • 基于对数概率单调性变化动态调节每对偏好样本的KL惩罚强度
  • 在多个对话基准上优于现有直接对齐方法,显著提升性能
  • 适合关注模型对齐精度与训练稳定性研究的读者

直接偏好优化(DPO)通过离线数据集实现大语言模型与人类偏好的对齐,但其缺点在于KL惩罚项在整个训练过程中保持静态。尽管已有方法尝试将静态惩罚转为动态,却无法为每对偏好样本自适应分配不同惩罚强度。本文提出ε-DPO,可针对每个偏好对自适应控制KL惩罚强度β。具体而言,ε-DPO基于训练过程中β扰动下对数概率的单调性变化来动态调节β,等价于仅利用当前策略与参考模型的对数概率,判断训练时温度变化是否能增强偏好置信度。实验表明,该简单判据显著提升DPO性能,在通用聊天机器人基准上优于多数现有直接对齐算法,并能有效反映偏好模型的混淆状态,提供高效KL权衡,凸显实例级自适应KL惩罚控制的重要性。

原文摘要 · Abstract (English)

Direct Preference Optimization (DPO) demonstrates the advantage of aligning a large language model with human preference using only an offline dataset. However, DPO has the limitation that the KL penalty, which prevents excessive deviation from the reference model, is static throughout the training process. Several methods claim to change this static KL penalty of DPO into a dynamic one, but no approach can adaptively assign different KL penalties for each preference pair. In this paper, we propose $\varepsilon$-Direct Preference Optimization ($\varepsilon$-DPO), which allows adaptive control of the KL penalty strength $β$ for each preference pair. Specifically, $\varepsilon$-DPO adaptively controls $β$ for each preference pair based on the monotonicity of logits as a preference model under the perturbation of $β$ during training. This is equivalent to adjusting the KL penalty by checking whether the change in training-time temperature can lead to better preference confidence as preference models by simply reusing the logit of the current policy and the reference policy. Experimental results show that the simple criterion of $\varepsilon$-DPO for KL penalty relaxation significantly improves DPO compared to most existing direct alignment algorithms on general chatbot benchmarks and reveal that this KL penalty control criterion can reflect confusion as a preference model and provide an efficient KL trade-off, highlighting the significance of instance-level adaptive KL penalty control in DPO.

偏好优化强化学习模型对齐KL惩罚

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。