arXiv:2512.06343cs.LGcs.AI2025-12

BT损失因表示距离偏差导致训练信号失真,提出NormBT改进更新机制。

When Distance Distracts: Representation Distance Bias in BT-Loss for Reward Models

  • 发现BT损失梯度受表示距离影响,大距离对对更新过强
  • 小距离对即使错排也更新微弱,影响精细区分能力
  • 提出NormBT自适应归一化,轻量改造提升模型性能

奖励模型在基于强化学习的人类反馈(RLHF)框架中对大语言模型对齐至关重要。标准奖励建模目标为布拉德利-特里(BT)损失,利用包含优选与次优响应的成对数据进行学习。本文分析了BT损失的样本级梯度,揭示了由表示距离引发的虚假学习信号。具体而言,BT梯度范数由两部分构成:(1) 预测误差,反映优选与次优响应间预测奖励的差异;(2) 成对样本在最终层输出空间中的表示距离。第一项为预期训练信号,第二项则显著影响更新幅度并造成学习错位。尤其当表示距离较小时,即使排序错误,更新仍趋近于零;而大距离对则获得过强更新。这导致大距离对的梯度主导小距离对,后者恰恰需要精细区分。为此,我们提出NormBT,一种自适应成对归一化方案,通过重缩放更新以平衡表示驱动效应,聚焦于预测误差。NormBT是轻量级、可直接替换的改进,开销极低。在多种大语言模型架构和数据集上,其性能均持续提升,在RewardBench推理类别中提升超5%——该类别包含大量细粒度成对样本。

原文摘要 · Abstract (English)

Reward models are central to Large Language Model (LLM) alignment within the framework of RLHF. The standard objective used in reward modeling is the Bradley-Terry (BT) loss, which learns from pairwise data consisting of chosen and rejected responses. In this work, we analyze the per-sample gradient of BT-loss and show spurious learning signals due to representation distance. In particular, BT gradient norm scales with two distinct components: (1) prediction error, reflected by the difference in predicted rewards between chosen and rejected responses, and critically, (2) representation distance between the pair measured in the output space of the final layer. While the first term captures the intended training signal, the second term can significantly impact the update magnitude and misalign learning. Specifically, pairs with small representation distance often receive vanishingly weak updates, even when misranked, while pairs with large distance receive disproportionately strong updates. This leads to gradients from large-distance pairs to overshadow those from small-distance pairs, where fine-grained distinctions are especially important. To overcome this limitation, we propose NormBT, an adaptive pair-wise normalization scheme that rescales updates to balance representation-driven effects and focuses learning signals on prediction error. NormBT is a lightweight, drop-in modification to BT loss with negligible overhead. Across various LLM backbones and datasets, NormBT improves reward model performance consistently, with notable gains of over 5% on the Reasoning category of RewardBench, which contains numerous fine-grained pairs.

奖励模型对比学习训练优化语言模型对齐

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