arXiv:2504.00891cs.CL2025-04AAAI被引 92

用生成式推理让奖励模型在测试时动态计算,显著提升数学推理能力

GenPRM: Scaling Test-Time Compute of Process Reward Models via Generative Reasoning

  • 通过代码验证的思维链推理生成过程评分
  • 仅用23K数据训练即超越现有模型,1.5B版超GPT-4o
  • 适合需要高精度推理与模型优化的研究者

大型语言模型(LLMs)的进展表明,利用过程奖励模型(PRMs)作为验证器可有效提升性能。然而现有PRMs存在三大挑战:(1) 过程监督和泛化能力有限,(2) 依赖标量预测,未发挥LLM的生成能力,(3) 难以扩展测试时计算。本文提出GenPRM,一种生成式过程奖励模型,通过代码验证的显式思维链(CoT)推理对每一步进行判断。为获取高质量过程标注与理由数据,提出相对进度估计(RPE)与融合代码验证的推理合成框架。在ProcessBench及多个数学推理任务上,仅使用23K来自MATH数据集的训练数据,GenPRM显著优于先前方法。通过测试时扩展,1.5B版本表现超过GPT-4o,7B版本超越Qwen2.5-Math-PRM-72B。此外,GenPRM展现出作为策略模型优化批评者的强大能力。本工作建立了一种新范式,弥合了PRMs与批评者模型之间的差距。代码、模型与数据将公开于https://ryanliu112.github.io/GenPRM。

原文摘要 · Abstract (English)

Recent advancements in Large Language Models (LLMs) have shown that it is promising to utilize Process Reward Models (PRMs) as verifiers to enhance the performance of LLMs. However, current PRMs face three key challenges: (1) limited process supervision and generalization capabilities, (2) dependence on scalar value prediction without leveraging the generative abilities of LLMs, and (3) inability to scale the test-time compute of PRMs. In this work, we introduce GenPRM, a generative process reward model that performs explicit Chain-of-Thought (CoT) reasoning with code verification before providing judgment for each reasoning step. To obtain high-quality process supervision labels and rationale data, we propose Relative Progress Estimation (RPE) and a rationale synthesis framework that incorporates code verification. Experimental results on ProcessBench and several mathematical reasoning tasks show that GenPRM significantly outperforms prior PRMs with only 23K training data from MATH dataset. Through test-time scaling, a 1.5B GenPRM outperforms GPT-4o, and a 7B GenPRM surpasses Qwen2.5-Math-PRM-72B on ProcessBench. Additionally, GenPRM demonstrates strong abilities to serve as a critic model for policy model refinement. This work establishes a new paradigm for process supervision that bridges the gap between PRMs and critic models in LLMs. Our code, model, and data will be available in https://ryanliu112.github.io/GenPRM.

过程奖励生成推理数学推理测试时计算

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。