用极少标注训练出高效推理验证模型,提升大模型解题准确率。
Process Reward Models That Think
- 基于长思维链生成验证过程,仅需1%标注数据训练
- 在多个挑战性评测中超越基线模型,最高提升8%
- 适合需要低标注成本、高推理可信度的场景
逐步验证器(又称过程奖励模型,PRMs)是测试时扩展的关键组件,但其训练需要大量步骤级标注,成本高昂。本文提出ThinkPRM,一种通过生成式思维链(CoT)进行逐步验证的语音化过程奖励模型,仅需传统判别式PRMs约1%的标注数据即可训练。该方法充分利用长思维链模型的内在推理能力,在ProcessBench、MATH-500和AIME '24等多任务评测中,通过最佳选择与奖励引导搜索策略均优于基线。在跨领域评估中,对GPQA-Diamond子集和LiveCodeBench的表现分别超过全量标注训练的判别式验证器8%和4.5%。在相同计算预算下,ThinkPRM的验证算力扩展效率高于LLM-as-a-Judge,于ProcessBench子集上领先7.2%。结果表明,生成式长思维链PRM可在极低监督下实现高效验证,有效拓展测试时计算资源。代码、数据与模型已开源。
原文摘要 · Abstract (English)
Step-by-step verifiers -- also known as process reward models (PRMs) -- are a key ingredient for test-time scaling. PRMs require step-level supervision, making them expensive to train. This work aims to build data-efficient PRMs as verbalized step-wise reward models that verify every step in the solution by generating a verification chain-of-thought (CoT). We propose ThinkPRM, a long CoT verifier fine-tuned on orders of magnitude fewer process labels than those required by discriminative PRMs. Our approach capitalizes on the inherent reasoning abilities of long CoT models, and outperforms LLM-as-a-Judge and discriminative verifiers -- using only 1% of the process labels in PRM800K -- across several challenging benchmarks. Specifically, ThinkPRM beats the baselines on ProcessBench, MATH-500, and AIME '24 under best-of-N selection and reward-guided search. In an out-of-domain evaluation on a subset of GPQA-Diamond and LiveCodeBench, our PRM surpasses discriminative verifiers trained on the full PRM800K by 8% and 4.5%, respectively. Lastly, under the same token budget, ThinkPRM scales up verification compute more effectively compared to LLM-as-a-Judge, outperforming it by 7.2% on a subset of ProcessBench. Our work highlights the value of generative, long CoT PRMs that can scale test-time compute for verification while requiring minimal supervision for training. Our code, data, and models are released at https://github.com/mukhal/thinkprm.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。