arXiv:2505.12843cs.LGcs.AI2025-05ACL被引 8

提出新方法消除奖励模型对长回复的偏见,让AI更精准回应而非堆字数。

Bias Fitting to Mitigate Length Bias of Reward Model in RLHF

  • 用轻量模型拟合长度与奖励的非线性关系,精准捕捉偏见模式。
  • 在DPO和BoN中应用后,长回复占比下降40%以上,胜率不降反而提升。
  • 适合关注生成质量、避免冗余输出的研究者与开发者。

强化学习从人类反馈(RLHF)依赖奖励模型将大语言模型对齐人类偏好。然而,RLHF常出现奖励劫持现象:政策学习利用训练好的奖励模型缺陷来最大化奖励分数,却未真正符合人类偏好。一个典型例子是长度偏见——奖励模型通常更青睐长回复,无论实际质量如何。此前工作存在明显局限:或仅缓解偏见而不分析其形式,或简单假设长度与奖励呈线性关系。为准确建模长度偏见的复杂特性并实现更有效的偏见缓解,我们提出FiMi-RM(Bias Fitting to Mitigate Length Bias of Reward Model),一种自主学习并校正底层偏见模式的框架。该方法包含三个阶段:首先,通过标准奖励模型预热,其天然包含长度偏见;其次,部署轻量拟合模型以捕捉长度与奖励间的非线性关系;最后,将学习到的关系融入奖励模型,有效解耦长度与奖励,同时保留偏好建模能力。实验表明,FiMi-RM实现了更均衡的长度-奖励分布。进一步应用于直接偏好优化(DPO)和最优- N(BoN)等对齐算法时,其去偏后的奖励模型显著提升了长度控制下的胜率,减少了冗余输出,且性能未受损。

原文摘要 · Abstract (English)

Reinforcement Learning from Human Feedback (RLHF) relies on reward models to align large language models with human preferences. However, RLHF often suffers from reward hacking, wherein policy learning exploits flaws in the trained reward model to maximize reward scores without genuinely aligning with human preferences. A significant example of such reward hacking is length bias, where reward models usually favor longer responses irrespective of actual response quality. Previous works on tackling length bias have notable limitations, these approaches either mitigate bias without characterizing the bias form, or simply assume a linear length-reward relation. To accurately model the intricate nature of length bias and facilitate more effective bias mitigation, we propose FiMi-RM (Bias Fitting to Mitigate Length Bias of Reward Model), a framework that autonomously learns and corrects underlying bias patterns. Our approach consists of three stages: First, we warm up by training a standard reward model which inherently contains length bias. Next, we deploy a lightweight fitting model to capture the non-linear relation between length and reward. Finally, we incorporate this learned relation into the reward model, effectively decoupling length from reward while preserving preference modeling capabilities. Experimental results demonstrate that FiMi-RM achieves a more balanced length-reward distribution. Furthermore, when applied to alignment algorithms such as Direct Preference Optimization (DPO) and Best-of-N (BoN), our debiased reward model improves length-controlled win rate and reduces verbosity without compromising its performance.

RLHF奖励模型长度偏见偏见修正

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