arXiv:2505.07271cs.CLcs.AI2025-05ICML被引 13

提升语言模型对齐中奖励模型的鲁棒性,防止过优化导致性能下降。

On the Robustness of Reward Models for Language Model Alignment

  • 通过批处理归零正则化约束奖励值分布,缓解隐藏状态范数过度分散问题。
  • 在8B规模模型上,复杂偏好预测任务准确率提升超5%,生成长度减少40%。
  • 适用于高阶对齐训练,尤其适合追求稳定性和泛化能力的研究者。

Bradley-Terry(BT)模型广泛用于基于人类反馈的强化学习(RLHF)中的奖励建模。尽管有效,但使用BT损失训练的奖励模型(RMs)容易出现过优化,丧失对未见输入分布的泛化能力。本文研究了过优化的原因及其对RLHF的下游影响,强调了奖励模型在未知数据上的分布鲁棒性至关重要。首先,我们发现隐藏状态范数过度分散是过优化的主要原因。随后提出批处理归零正则化(BSR),强制每批次奖励和为零,从而限制极端奖励值。我们在四种过优化场景下评估了BSR,结果表明其显著提升了鲁棒性。进一步对比原始BT模型与加入BSR的模型在RLHF训练中的表现,实证显示鲁棒奖励模型更优地对齐策略与黄金偏好模型。最后,将BSR应用于高质量数据与模型,在8B规模下于复杂偏好预测任务中超越现有最优模型超过5%。通过使用8B RMs进行RLOO训练,AlpacaEval 2.0将生成长度减少40%,同时胜率提升7%,进一步证明奖励模型的鲁棒性能传导至整个RLHF过程。代码、数据与模型已开源:https://github.com/LinkedIn-XFACT/RM-Robustness。

原文摘要 · Abstract (English)

The Bradley-Terry (BT) model is widely practiced in reward modeling for reinforcement learning with human feedback (RLHF). Despite its effectiveness, reward models (RMs) trained with BT model loss are prone to over-optimization, losing generalizability to unseen input distributions. In this paper, we study the cause of over-optimization in RM training and its downstream effects on the RLHF procedure, accentuating the importance of distributional robustness of RMs in unseen data. First, we show that the excessive dispersion of hidden state norms is the main source of over-optimization. Then, we propose batch-wise sum-to-zero regularization (BSR) to enforce zero-centered reward sum per batch, constraining the rewards with extreme magnitudes. We assess the impact of BSR in improving robustness in RMs through four scenarios of over-optimization, where BSR consistently manifests better robustness. Subsequently, we compare the plain BT model and BSR on RLHF training and empirically show that robust RMs better align the policy to the gold preference model. Finally, we apply BSR to high-quality data and models, which surpasses state-of-the-art RMs in the 8B scale by adding more than 5% in complex preference prediction tasks. By conducting RLOO training with 8B RMs, AlpacaEval 2.0 reduces generation length by 40% while adding a 7% increase in win rate, further highlighting that robustness in RMs induces robustness in RLHF training. We release the code, data, and models: https://github.com/LinkedIn-XFACT/RM-Robustness.

奖励模型鲁棒性对齐训练RLHF

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