arXiv:2409.17407cs.AIcs.CL2024-09ICLR被引 31

修复大模型奖励模型的长度偏好,不需额外训练即可提升对齐效果。

Post-hoc Reward Calibration: A Case Study on Length Bias

  • 事后校准奖励模型,通过局部加权回归估计并消除长度偏差。
  • 在33个模型上平均性能提升3.11,与GPT-4和人类偏好更一致。
  • 方法轻量高效,可推广至其他类型偏差,适合实际部署场景。

基于人类反馈的强化学习(RLHF)通过奖励模型(RM)将大语言模型输出对齐人类价值观。但奖励模型可能因训练数据中的虚假相关性产生偏差,例如偏好较长或特定风格的输出而非真实质量,导致排名错误、评估失准,并放大不良行为。本文提出无需额外数据和训练的“事后奖励校准”方法,先估计偏差项并消除,再用局部加权回归扩展为通用鲁棒方案。聚焦常见的长度偏差,在三个实验设置中验证:(1) 在RewardBench数据集上33个奖励模型平均性能提升3.11;(2) 与GPT-4评估及人类偏好更一致,体现在AlpacaEval基准上;(3) 多种大模型-奖励模型组合中,长度控制下的胜率显著提高。该方法计算高效,可推广至其他偏差类型与奖励模型,提供一种可扩展、鲁棒的偏差缓解方案。代码与结果见https://github.com/ZeroYuHuang/Reward-Calibration。

原文摘要 · Abstract (English)

Reinforcement Learning from Human Feedback aligns the outputs of Large Language Models with human values and preferences. Central to this process is the reward model (RM), which translates human feedback into training signals for optimising LLM behaviour. However, RMs can develop biases by exploiting spurious correlations in their training data, such as favouring outputs based on length or style rather than true quality. These biases can lead to incorrect output rankings, sub-optimal model evaluations, and the amplification of undesirable behaviours in LLMs alignment. This paper addresses the challenge of correcting such biases without additional data and training, introducing the concept of Post-hoc Reward Calibration. We first propose an intuitive approach to estimate the bias term and, thus, remove it to approximate the underlying true reward. We then extend the approach to a more general and robust form with the Locally Weighted Regression. Focusing on the prevalent length bias, we validate our proposed approaches across three experimental settings, demonstrating consistent improvements: (1) a 3.11 average performance gain across 33 reward models on the RewardBench dataset; (2) enhanced alignment of RM rankings with GPT-4 evaluations and human preferences based on the AlpacaEval benchmark; and (3) improved Length-Controlled win rate of the RLHF process in multiple LLM--RM combinations. Our method is computationally efficient and generalisable to other types of bias and RMs, offering a scalable and robust solution for mitigating biases in LLM alignment. Our code and results are available at https://github.com/ZeroYuHuang/Reward-Calibration.

奖励模型偏差校准长度偏好RLHF

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