LongRM: Revealing and Unlocking the Context Boundary of Reward Modeling

Zecheng Tang, Baibei Ji, Quantong Qiu, Haitian Wang, Xiaobo Liang +2 more
10/8/2025
cs.CLcs.AI

No official code found. Best implementation: LCM-Lab/LongRM (21 stars, last commit 2026-01-11).

Abstract

Reward model (RM) plays a pivotal role in aligning large language model (LLM) with human preferences. As real-world applications increasingly involve long history trajectories, e.g., LLM agent, it becomes indispensable to evaluate whether a model's responses are not only high-quality but also grounded in and consistent with the provided context. Yet, current RMs remain confined to short-context settings and primarily focus on response-level attributes (e.g., safety or helpfulness), while largely neglecting the critical dimension of long context-response consistency. In this work, we introduce Long-RewardBench, a benchmark specifically designed for long-context RM evaluation, featuring both Pairwise Comparison and Best-of-N tasks. Our preliminary study reveals that even state-of-the-art generative RMs exhibit significant fragility in long-context scenarios, failing to maintain context-aware preference judgments. Motivated by the analysis of failure patterns observed in model outputs, we propose a general multi-stage training strategy that effectively scales arbitrary models into robust Long-context RMs (LongRMs). Experiments show that our approach not only substantially improves performance on long-context evaluation but also preserves strong short-context capability. Notably, our 8B LongRM outperforms much larger 70B-scale baselines and matches the performance of the proprietary Gemini 2.5 Pro model.

View on arXivView PDF

Code Implementations(1)

LCM-Lab/LongRMCommunity73%

Revealing and unlocking the context boundary of reward models

211Shell, PythonOct 7, 20259 months ago
benchmarklong-context-evaluationreward-modeling

Similar Papers

Cite this paper

@article{tang2025longrm,
  title  = {LongRM: Revealing and Unlocking the Context Boundary of Reward Modeling},
  author = {Zecheng Tang and Baibei Ji and Quantong Qiu and Haitian Wang and Xiaobo Liang and Juntao Li and Min Zhang},
  year   = {2025},
  eprint = {2510.06915},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2510.06915v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2510.06915.svg)](https://codeofpaper.com/paper/2510.06915)

Discussion