arXiv:2607.16240cs.LGcs.AI2026-07

通过正则化改善偏好优化中的过拟合问题,提升模型生成质量与泛化能力。

Normalized Rewards for Preference Optimization

论文配图:Normalized Rewards for Preference Optimization
图 1 · 摘自论文原文
  • 引入长度归一化的正则项,抑制偏好优化中对响应概率的过度调整。
  • 在Llama-3.1-8B上实现AlpacaEval2得分提升超20%,通用基准性能增9%以上。
  • 特别缓解了关键异常词元的分布偏移,适用于希望稳定生成质量的研究者。

直接对齐算法(DAAs)如DPO已成为对大语言模型进行后训练以匹配人类偏好的常用方法。然而,这些方法常导致其隐式奖励模型过度优化,降低优选响应的概率,进而减少偏好数据集中响应的总概率,可能引发不良行为。为缓解这一副作用,本文研究了在目标函数中加入正则项以保持所选与被拒响应的总长度归一化概率的方法。为理解过度优化机制,我们分析了有无正则化时响应概率变化在词元上的分布。发现大部分概率变化源于少数异常词元,这解释了为何尽管优选响应概率下降,生成质量仍能提升。将该正则化应用于基于参考的(DPO)和无参考的(SimPO)方法,均取得:(1) 生成质量与通用基准能力之间的更好权衡;(2) 跨数据集的奖励建模性能提升。例如,在Llama-3.1-8B-Instruct上,AlpacaEval2得分相对提升超20%,通用基准表现相对提高9%以上。此外,正则项有效减少了优选响应整体及异常词元处的分布偏移,尤其通过利用低概率词元实现。

原文摘要 · Abstract (English)

Direct Alignment Algorithms (DAAs) such as DPO have become a common way to post-train and align LLMs with human preferences. However, DAAs have been observed to over-optimize their implicit reward model and decrease the likelihood of preferred responses. This results in a decrease in the total likelihood assigned to responses seen in the preference dataset, potentially resulting in undesirable behavior. To counteract this undesired side-effect of DAAs, we examine the effect of using objectives that add a regularization term to maintain the total length-normalized probabilities of the chosen and rejected responses. To better understand over-optimization, we investigate how response likelihood changes are distributed over the tokens with and without regularization. We find that a significant portion of the likelihood changes are due to a small set of outlier tokens, which explains how DAAs improve generation quality despite decreasing the likelihoods of chosen responses. We apply the proposed regularization to reference-based (DPO) and reference-free (SimPO) methods and find (1) improved trade-offs between generation quality and general benchmark capability and (2) improvements in reward modeling across datasets. For example, on Llama-3.1-8B-Instruct, we see both a >20% relative increase in AlpacaEval2 scores and >9% relative performance gains on general benchmarks. Additionally, we find that the added regularization term effectively mitigates the amount of displacement within preferred responses overall, and for the outlier tokens specifically, by utilizing low-likelihood tokens.

偏好优化正则化大模型对齐生成质量

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