Fixing the Broken Compass: Diagnosing and Improving Inference-Time Reward Modeling

Jiachun Li, Pengfei Cao, Zhuoran Jin, Yubo Chen, Jiexin Xu +4 more
3/7/2025
cs.CLcs.AI

No confident code implementation found for this paper.

Abstract

Inference-time scaling techniques have shown promise in enhancing the reasoning capabilities of large language models (LLMs). While recent research has primarily focused on training-time optimization, our work highlights inference-time reward model (RM)-based reasoning as a critical yet overlooked avenue. In this paper, we conduct a systematic analysis of RM behavior across downstream reasoning tasks, revealing three key limitations: (1) RM can impair performance on simple questions, (2) its discriminative ability declines with increased sampling, and (3) high search diversity undermines RM performance. To address these issues, we propose CRISP (Clustered Reward Integration with Stepwise Prefixing), a novel inference-time algorithm that clusters generated reasoning paths by final answers, aggregates reward signals at the cluster level, and adaptively updates prefix prompts to guide generation. Experimental results demonstrate that CRISP significantly enhances LLM reasoning performance, achieving up to 5% accuracy improvement over other RM-based inference methods and an average of 10% gain over advanced reasoning models.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Cite this paper

@article{li2025fixing,
  title  = {Fixing the Broken Compass: Diagnosing and Improving Inference-Time Reward Modeling},
  author = {Jiachun Li and Pengfei Cao and Zhuoran Jin and Yubo Chen and Jiexin Xu and Huaijun Li and Xiaojian Jiang and Kang Liu and Jun Zhao},
  year   = {2025},
  eprint = {2503.05188},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2503.05188v2}
}

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/2503.05188.svg)](https://codeofpaper.com/paper/2503.05188)

Discussion