解决大模型对齐中奖励生成偏差问题,提升指令响应质量。
Towards Bridging the Reward-Generation Gap in Direct Alignment Algorithms
- 从令牌级马尔可夫决策过程视角分析对齐算法缺陷
- 提出截断策略使优劣回复长度一致,提升评估得分11.8点
- 适用于追求高效对齐的LLM开发者与研究者
直接对齐算法(DAAs),如直接偏好优化(DPO)和简单偏好优化(SimPO),已成为替代基于人类反馈强化学习(RLHF)的有效方法。然而,这类算法存在一个我们称之为“奖励生成差距”的根本性局限,即训练目标与自回归解码动态之间的不一致。本文指出,这一差距的部分原因在于前缀令牌在生成过程中的重要性与其在DAAs隐式奖励函数中的体现不匹配。为此,我们从令牌级马尔可夫决策过程视角分析其局限,并提出一种简单有效的改进方法——前缀导向等长训练(POET),该方法将优选与非优选回复均截断至较短者长度。在DPO和SimPO上的实验表明,相比标准实现,POET在AlpacaEval 2上最高提升11.8分,并在下游任务中实现整体性能提升。结果凸显了通过更好对齐训练目标与自回归解码动态来缓解奖励生成差距的重要性。
原文摘要 · Abstract (English)
Direct Alignment Algorithms (DAAs), such as Direct Preference Optimization (DPO) and Simple Preference Optimization (SimPO), have emerged as efficient alternatives to Reinforcement Learning from Human Feedback (RLHF) algorithms for aligning large language models (LLMs) with human preferences. However, DAAs suffer from a fundamental limitation we identify as the "reward-generation gap", a discrepancy between training objectives and autoregressive decoding dynamics. In this paper, we consider that one contributor to the reward-generation gap is the mismatch between the inherent importance of prefix tokens during the LLM generation process and how this importance is reflected in the implicit reward functions of DAAs. To bridge the gap, we adopt a token-level MDP perspective of DAAs to analyze its limitations and introduce a simple yet effective approach called Prefix-Oriented Equal-length Training (POET), which truncates both preferred and dispreferred responses to match the shorter one's length. We conduct experiments with DPO and SimPO, two representative DAAs, demonstrating that POET improves over their standard implementations, achieving up to 11.8 points in AlpacaEval 2 and overall improvements across downstream tasks. These results underscore the need to mitigate the reward-generation gap in DAAs by better aligning training objectives with autoregressive decoding dynamics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。