无需额外奖励模型,用自洽机制提升大模型推理的强化学习方法。
Self-Guided Process Reward Optimization with Redefined Step-wise Advantage for Process Reinforcement Learning
- 从策略模型自身推导过程奖励,避免外加奖励模型。
- 通过重新定义步骤优势,实现更精准的动作优化,训练效率提升3.4倍。
- 无额外计算开销,适合工业部署,防止奖励作弊且保持探索性。
过程强化学习(PRL)在提升大语言模型(LLM)推理能力方面展现出巨大潜力。然而,引入额外的过程奖励模型会带来显著的计算开销,且缺乏统一的理论框架用于过程级优势估计。为解决这一问题,我们提出一种新型框架Self-Guided Process Reward Optimization(SPRO),其核心创新包括:(1) 证明过程奖励可从策略模型自身内生推导;(2) 通过引入累积过程奖励(CPR)与掩码步骤优势(MSA),重新定义步骤优势,实现在共享提示采样组内的严格步骤动作优势估计。实验表明,SPRO相较于原始GRPO,训练效率提升3.4倍,测试准确率提高12.9%。同时,SPRO在整个训练过程中保持稳定且较高的策略熵,响应长度显著减少,体现出充分探索并有效防止奖励欺骗。值得注意的是,SPRO相比如GRPO等结果监督的强化学习方法,不增加任何计算开销,有利于工业落地。
原文摘要 · Abstract (English)
Process Reinforcement Learning~(PRL) has demonstrated considerable potential in enhancing the reasoning capabilities of Large Language Models~(LLMs). However, introducing additional process reward models incurs substantial computational overhead, and there is no unified theoretical framework for process-level advantage estimation. To bridge this gap, we propose \textbf{S}elf-Guided \textbf{P}rocess \textbf{R}eward \textbf{O}ptimization~(\textbf{SPRO}), a novel framework that enables process-aware RL through two key innovations: (1) we show that process rewards can be derived intrinsically from the policy model itself, and (2) we redefine step-wise advantage by introducing well-defined Cumulative Process Rewards~(\textbf{CPR}) and \textbf{M}asked \textbf{S}tep \textbf{A}dvantage~(\textbf{MSA}), which facilitates rigorous step-wise action advantage estimation within shared-prompt sampling groups. Our experimental results show that SPRO outperforms vanilla GRPO with 3.4x higher training efficiency and a 12.9\% test accuracy improvement. Furthermore, SPRO maintains a stable and elevated policy entropy throughout training while achieving a considerable reduction in response length, evidencing sufficient exploration and prevention of reward hacking. Notably, SPRO incurs no additional computational overhead compared to outcome-supervised RL methods such as GRPO, which benefit industrial implementation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。