Stop Summation: Min-Form Credit Assignment Is All Process Reward Model Needs for Reasoning

Jie Cheng, Gang Xiong, Ruixi Qiao, Lijun Li, Chao Guo +3 more
4/21/2025
cs.AIcs.LG

Official code: CJReinforce/PURE (174 stars, last commit 2025-10-23).

Abstract

Process reward models (PRMs) have proven effective for test-time scaling of Large Language Models (LLMs) on challenging reasoning tasks. However, reward hacking issues with PRMs limit their successful application in reinforcement fine-tuning. In this paper, we identify the main cause of PRM-induced reward hacking: the canonical summation-form credit assignment in reinforcement learning (RL), which defines the value as cumulative gamma-decayed future rewards, easily induces LLMs to hack steps with high rewards. To address this, we propose PURE: Process sUpervised Reinforcement lEarning. The key innovation of PURE is a min-form credit assignment that formulates the value function as the minimum of future rewards. This method significantly alleviates reward hacking by limiting the value function range and distributing advantages more reasonably. Through extensive experiments on 3 base models, we show that PRM-based approaches enabling min-form credit assignment achieve comparable reasoning performance to verifiable reward-based methods within only 30% steps. In contrast, the canonical sum-form credit assignment collapses training even at the beginning! Additionally, when we supplement PRM-based fine-tuning with just 10% verifiable rewards, we further alleviate reward hacking and produce the best fine-tuned model based on Qwen2.5-Math-7B in our experiments, achieving 82.5% accuracy on AMC23 and 53.3% average accuracy across 5 benchmarks. Moreover, we summarize the observed reward hacking cases and analyze the causes of training collapse. We release our code and model weights at https://github.com/CJReinforce/PURE.

View on arXivView PDF

Code Implementations(1)

CJReinforce/PUREOfficial100%

Official code for the paper, "Stop Summation: Min-Form Credit Assignment Is All Process Reward Model Needs for Reasoning"

1747ANTLR, ShellFeb 9, 202512 months ago
llmmathematicso1r1reasoning+3 more

Similar Papers

Cite this paper

@article{cheng2025stop,
  title  = {Stop Summation: Min-Form Credit Assignment Is All Process Reward Model Needs for Reasoning},
  author = {Jie Cheng and Gang Xiong and Ruixi Qiao and Lijun Li and Chao Guo and Junle Wang and Yisheng Lv and Fei-Yue Wang},
  year   = {2025},
  eprint = {2504.15275},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2504.15275v3}
}

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

Discussion