arXiv:2410.10148cs.LGcs.AI2024-10ICML被引 25

提出自适应奖励边界方法,提升大模型对齐效果与训练稳定性。

AlphaDPO: Adaptive Reward Margin for Direct Preference Optimization

  • 引入动态奖励边界,根据数据自适应调整偏好优化策略。
  • 在AlpacaEval 2和Arena-Hard上胜率显著高于DPO和SimPO。
  • 适合追求高效稳定对齐的大模型训练者使用。

对齐大型语言模型(LLM)与人类价值观和意图对于其可用性、诚实性和安全性至关重要。强化学习从人类反馈(RLHF)是实现对齐的常用方法,但面临计算效率低和训练不稳定的问题。近期的直接偏好优化(DPO)和简单偏好优化(SimPO)提出了离线替代方案,通过重参数化奖励函数简化流程。然而,DPO依赖潜在次优的参考模型,而SimPO假设固定的目标奖励边界,在多样化数据设置下可能导致次优决策。本文提出α-DPO,一种自适应偏好优化算法,通过引入动态奖励边界解决上述问题。具体而言,α-DPO采用自适应偏好分布,平衡策略模型与参考模型,实现个性化奖励边界。我们为α-DPO提供了理论保证,证明其作为替代优化目标的有效性,并能通过控制KL散度平衡对齐与多样性。在AlpacaEval 2和Arena-Hard上的实验表明,α-DPO在多种模型设置下持续优于DPO和SimPO,展现出强大的鲁棒性,显著提升了胜率。代码已开源。

原文摘要 · Abstract (English)

Aligning large language models (LLMs) with human values and intentions is crucial for their utility, honesty, and safety. Reinforcement learning from human feedback (RLHF) is a popular approach to achieve this alignment, but it faces challenges in computational efficiency and training stability. Recent methods like Direct Preference Optimization (DPO) and Simple Preference Optimization (SimPO) have proposed offline alternatives to RLHF, simplifying the process by reparameterizing the reward function. However, DPO depends on a potentially suboptimal reference model, and SimPO's assumption of a fixed target reward margin may lead to suboptimal decisions in diverse data settings. In this work, we propose $α$-DPO, an adaptive preference optimization algorithm designed to address these limitations by introducing a dynamic reward margin. Specifically, $α$-DPO employs an adaptive preference distribution, balancing the policy model and the reference model to achieve personalized reward margins. We provide theoretical guarantees for $α$-DPO, demonstrating its effectiveness as a surrogate optimization objective and its ability to balance alignment and diversity through KL divergence control. Empirical evaluations on AlpacaEval 2 and Arena-Hard show that $α$-DPO consistently outperforms DPO and SimPO across various model settings, establishing it as a robust approach for fine-tuning LLMs. Our method achieves significant improvements in win rates, highlighting its potential as a powerful tool for LLM alignment. The code is available at https://github.com/junkangwu/alpha-DPO

大模型对齐偏好优化自适应机制

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