语言模型的隐式奖励模型为何泛化能力差?
Why is Your Language Model a Poor Implicit Reward Model?
- 隐式奖励模型依赖表面词元线索,泛化能力弱于显式模型。
- 在词元分布偏移下,隐式模型性能下降更明显,即使同分布也更差。
- 设计差异虽小,但显著影响模型泛化,适合关注鲁棒性的研究者看。
奖励模型是语言模型后训练与推理流程的核心。近期研究发现,每个语言模型本身即构成一个隐式奖励模型(IM-RM),无需架构改动。然而,相比使用专用线性头的显式奖励模型(EX-RM),IM-RM泛化能力较差,尤其在分布外情形。尽管两者使用相同数据、损失函数和语言模型,仅奖励计算方式不同,这一差距仍令人困惑。本文通过理论与实验揭示根本原因:IM-RM过度依赖表面词元级线索。因此,在词元级分布偏移及同分布条件下,其泛化表现均劣于EX-RM。我们还排除了其他解释,如‘生成难度高于验证’的假设——因模型可同时作为生成器与验证器,并非主因。结果表明,微小的设计选择会显著影响奖励模型的泛化行为。
原文摘要 · Abstract (English)
Reward models are key to language model post-training and inference pipelines. Conveniently, recent work showed that every language model defines an implicit reward model (IM-RM), without requiring any architectural changes. However, such IM-RMs tend to generalize worse, especially out-of-distribution, compared to explicit reward models (EX-RMs) that apply a dedicated linear head over the hidden representations of a language model. The existence of a generalization gap is puzzling, as EX-RMs and IM-RMs are nearly identical. They can be trained using the same data, loss function, and language model, and differ only in how the reward is computed. Toward a fundamental understanding of the implicit biases underlying different reward model types, we investigate the root cause of this gap. Our main finding, backed by theory and experiments, is that IM-RMs rely more heavily on superficial token-level cues. Consequently, they often generalize worse than EX-RMs under token-level distribution shifts, as well as in-distribution. Furthermore, we provide evidence against alternative hypotheses for the generalization gap. Most notably, we challenge the claim that IM-RMs struggle in tasks where generation is harder than verification because they can operate both as a verifier and a generator. Overall, our results highlight that seemingly minor design choices can substantially impact the generalization behavior of reward models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。