无需人工标注,用语言模型自身概率识别推理错误步骤。
Unsupervised Process Reward Models
- 基于大模型下一个词概率设计评分函数,批量判断推理中的首个错误节点。
- 在ProcessBench上比LLM-as-a-Judge准确率提升15%,验证效果接近有监督模型。
- 适用于推理纠错、测试时扩展和强化学习奖励信号,可规模化部署。
过程奖励模型(PRMs)通过细粒度的步骤级监督,有效引导大语言模型的推理过程,但其高成本源于对每一步推理都需要专家标注。本文提出无监督过程奖励模型(uPRM),完全无需人类标注,既不依赖步骤级注释,也不需要最终答案的真值验证。核心思路是利用大模型的下一个词概率构造一个评分函数,联合评估一批推理轨迹中首个错误步骤的候选位置。实验表明:(i) uPRM在ProcessBench数据集上对首个错误步骤的识别准确率比LLM-as-a-Judge高出最多15%;(ii) 作为测试时扩展的验证器,uPRM表现与有监督PRMs相当,并优于多数投票基线最多6.9%;(iii) 在强化学习中作为奖励信号,uPRM使策略优化更稳定,优于使用真值标签训练的有监督PRM。结果证明了复杂推理任务中可扩展奖励建模的新路径。
原文摘要 · Abstract (English)
Process Reward Models (PRMs) are a powerful mechanism for steering large language model reasoning by providing fine-grained, step-level supervision. However, this effectiveness comes at a significant cost: PRMs require expert annotations for every reasoning step, making them costly and difficult to scale. Here, we propose a method for training unsupervised PRMs (uPRM) that requires no human supervision, neither at the level of step-by-step annotations nor through ground-truth verification of final answers. The key idea behind our approach is to define a scoring function, derived from LLM next-token probabilities, that jointly assesses candidate positions of first erroneous steps across a batch of reasoning trajectories. We demonstrate the effectiveness of uPRM across diverse scenarios: (i) uPRM achieves up to 15% absolute accuracy improvements over the LLM-as-a-Judge in identifying first erroneous steps on the ProcessBench dataset; (ii) as a verifier for test-time scaling, uPRM performs comparably to supervised PRMs and outperforms the majority voting baseline by up to 6.9%, and (iii) when used as a reward signal in reinforcement learning, uPRM enables more robust policy optimization throughout training compared to a supervised PRM trained using ground-truth labels. Overall, our results open a path toward scalable reward modeling for complex reasoning tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。