arXiv:2502.14340cs.CL2025-02ICLR被引 10

让早期词元权重更高,改进大模型对齐效果

Earlier Tokens Contribute More: Learning Direct Preference Optimization From Temporal Decay Perspective

  • 引入时间衰减机制,按位置动态调整奖励权重
  • 在AlpacaEval2上提升5.9-8.8分,Arena-Hard上提升3.3-9.7分
  • 适合追求高效对齐且关注生成质量的研究者

直接偏好优化(DPO)作为强化学习从人类反馈中学习(RLHF)的高效替代方案,被广泛用于对齐大语言模型与人类偏好。尽管如此,DPO仍存在长度偏差问题,生成响应普遍长于参考模型。现有方法如SimPO和SamPO虽部分缓解此问题,但对序列中各位置奖励的贡献采用统一处理,忽略了时序动态特性。为此,我们提出一种新方法,引入由伽马参数控制的时间衰减因子,使奖励影响随位置动态变化,更重视对齐关键的早期词元。该机制通过聚焦相关反馈,减轻对不重要数据的过拟合,同时保持对人类偏好演化的敏感性。在多个基准测试中,该方法在AlpacaEval 2上比原始DPO提升5.9–8.8分,在Arena-Hard上提升3.3–9.7分,覆盖不同架构与规模的模型。数学与推理任务(MMLU、GSM8K、MATH)上的额外实验表明,该方法提升性能的同时未损害通用能力。代码将开源至\url{https://github.com/LotuSrc/D2PO}。

原文摘要 · Abstract (English)

Direct Preference Optimization (DPO) has gained attention as an efficient alternative to reinforcement learning from human feedback (RLHF) for aligning large language models (LLMs) with human preferences. Despite its advantages, DPO suffers from a length bias, generating responses longer than those from the reference model. Existing solutions like SimPO and SamPO address this issue but uniformly treat the contribution of rewards across sequences, overlooking temporal dynamics. To this end, we propose an enhanced preference optimization method that incorporates a temporal decay factor controlled by a gamma parameter. This dynamic weighting mechanism adjusts the influence of each reward based on its position in the sequence, prioritizing earlier tokens that are more critical for alignment. By adaptively focusing on more relevant feedback, our approach mitigates overfitting to less pertinent data and remains responsive to evolving human preferences. Experimental results on several benchmarks show that our approach consistently outperforms vanilla DPO by 5.9-8.8 points on AlpacaEval 2 and 3.3-9.7 points on Arena-Hard across different model architectures and sizes. Furthermore, additional experiments on mathematical and reasoning benchmarks (MMLU, GSM8K, and MATH) confirm that our method enhances performance without compromising general capabilities. Our codebase would be available at \url{https://github.com/LotuSrc/D2PO}.

偏好优化大模型对齐时序建模

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