用思维过程奖励提升大模型数学推理效率,训练更省样本。
Good Learners Think Their Thinking: Generative PRM Makes Large Reasoning Model More Efficient Math Learner
- 基于内在信号生成思维级奖励,判断每步正确性并整合为完整思考单元。
- 在1.5B和7B模型上,用更少样本达到更高解题准确率,比仅靠结果奖励快数倍。
- 适合研究大模型高效训练、数学推理优化的开发者与研究人员。
大型推理模型(LRMs)在强化学习(RL)优化后可解决复杂数学问题,但传统方法依赖结果奖励,反馈稀疏,导致训练效率低下。本文研究过程奖励模型(PRM)的作用,提出一种新型内在信号驱动的生成式过程评估机制,直接在思维层面进行判断。该方法不需预先知道如何解题,而是利用解题过程中的内在信号识别步骤正确性,并将连续正确或错误步骤聚合为结构化的‘思维’单元,从而减少步骤划分歧义,缓解奖励欺骗问题。进一步设计能力自适应奖励机制,根据模型当前能力动态平衡探索与利用,避免抑制创造性试错。上述创新集成于新提出的离策略强化学习算法TP-GRPO,该算法在分组近端优化基础上引入基于过程的奖励,显著提升训练效率。在1.5B和7B参数的LRMs上实验表明,本方法以更少训练样本实现更高解题准确率,验证了结构化过程奖励能大幅加速数学推理任务的模型优化。代码已开源:https://github.com/cs-holder/tp_grpo。
原文摘要 · Abstract (English)
Large reasoning models (LRMs) have recently shown promise in solving complex math problems when optimized with Reinforcement Learning (RL). But conventional approaches rely on outcome-only rewards that provide sparse feedback, resulting in inefficient optimization process. In this work, we investigate the function of process reward models (PRMs) to accelerate the RL training for LRMs. We propose a novel intrinsic signal-driven generative process evaluation mechanism operating at the thought level to address major bottlenecks in RL-based training. Specifically, instead of requiring PRMs to know how to solve problems, our method uses intrinsic signals in solutions to judge stepwise correctness and aggregate contiguous correct/incorrect steps into coherent 'thought' units. This structured, thought-level rewards enable more reliable credit assignment by reducing ambiguity in step segmentation and alleviating reward hacking. We further introduce a capability-adaptive reward mechanism that dynamically balances exploration and exploitation based on the LRM's current proficiency, guiding learning without stifling creative trial-and-error. These innovations are integrated into a new off-policy RL algorithm, TP-GRPO, which extends grouped proximal optimization with process-based rewards and improves training efficiency. Experiments on 1.5B and 7B parameter LRMs demonstrate that our method achieves higher problem-solving accuracy with significantly fewer training samples than outcome-only reward baselines. The results validate that well-structured process rewards can substantially accelerate LRM optimization in math reasoning tasks. Code is available at https://github.com/cs-holder/tp_grpo.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。