English

Why is Your Language Model a Poor Implicit Reward Model?

Computation and Language 2026-01-28 v3 Artificial Intelligence Machine Learning Machine Learning

Abstract

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.

Keywords

Cite

@article{arxiv.2507.07981,
  title  = {Why is Your Language Model a Poor Implicit Reward Model?},
  author = {Noam Razin and Yong Lin and Jiarui Yao and Sanjeev Arora},
  journal= {arXiv preprint arXiv:2507.07981},
  year   = {2026}
}

Comments

Accepted to ICLR 2026; Code available at https://github.com/princeton-pli/exrm-vs-imrm

R2 v1 2026-07-01T03:55:13.558Z