改进DPO算法,让大模型生成更符合人类偏好的结果
Entropy Controllable Direct Preference Optimization
- 通过调节策略熵提升DPO的模式捕捉能力
- 数学任务上pass@$k$指标优于原版DPO
- 仅修改损失函数,易于在现有框架中部署
在大语言模型后训练中,基于人类反馈的强化学习(RLHF)是实现生成内容与人类偏好对齐的有效方法。直接偏好优化(DPO)通过二元交叉熵损失进行策略训练,无需奖励模型。其目标受反向KL散度正则化,鼓励对参考策略的模式搜索拟合。然而,我们指出最小化反向KL可能导致无法捕捉参考分布的某一模式,从而损害策略性能。为此,我们提出一种简单改进——H-DPO,可控制最终策略的熵,增强分布锐度,从而更有效地实现模式搜索拟合。实验表明,H-DPO在多个任务中均优于DPO,尤其在数学任务的pass@$k$评估中表现更优。此外,H-DPO实现简单,仅需微调DPO的损失计算,具备高度实用性与广泛应用前景。
原文摘要 · Abstract (English)
In the post-training of large language models (LLMs), Reinforcement Learning from Human Feedback (RLHF) is an effective approach to achieve generation aligned with human preferences. Direct Preference Optimization (DPO) allows for policy training with a simple binary cross-entropy loss without a reward model. The objective of DPO is regularized by reverse KL divergence that encourages mode-seeking fitting to the reference policy. Nonetheless, we indicate that minimizing reverse KL divergence could fail to capture a mode of the reference distribution, which may hurt the policy's performance. Based on this observation, we propose a simple modification to DPO, H-DPO, which allows for control over the entropy of the resulting policy, enhancing the distribution's sharpness and thereby enabling mode-seeking fitting more effectively. In our experiments, we show that H-DPO outperformed DPO across various tasks, demonstrating superior results in pass@$k$ evaluations for mathematical tasks. Moreover, H-DPO is simple to implement, requiring only minor modifications to the loss calculation of DPO, which makes it highly practical and promising for wide-ranging applications in the training of LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。