小模型数学推理中,过程奖励比结果奖励更有效
Reward Granularity in RLVR: Comparing Process and Outcome Reward Structures for Mathematical Reasoning in Small Language Models

- 用步骤级奖励替代仅给最终答案奖励,提升小模型推理能力
- 过程奖励使准确率提升至63.73%,比结果奖励高近10个百分点
- 适合关注推理过程可信度的研究者与教育类应用
强化学习结合可验证奖励(RLVR)已成为提升语言模型数学推理能力的有前景范式。然而多数研究仅对最终答案进行奖励(结果导向),忽视了步骤级过程监督(过程奖励)的影响,尤其在缺乏自我修正能力的小模型上。本文在GSM8K数据集上,对Qwen2.5-0.5B模型使用组相对策略优化(GRPO)微调,系统比较五种奖励设置:无强化学习基线、仅过程奖励、仅结果奖励,以及三种混合权重(λ∈{0.9, 0.5, 0.1})。结果显示,仅过程奖励达63.73%测试准确率,显著优于仅结果奖励的53.75%;且推理链更具结构一致性,偏离真实链长更少。混合奖励总体随过程权重增加而提升,但低过程/高结果配置(λ=0.1)表现反而劣于纯结果奖励,暗示优化信号冲突。通过GPT-4o分析错误模式发现:过程模型生成结构不一致但计算正确的链条,结果模型则链路简短但含推导错误。结果表明,奖励粒度是RLVR设计的关键决策,过程级监督能显著提升小模型的准确率与推理轨迹保真度。
原文摘要 · Abstract (English)
Reinforcement Learning with Verifiable Rewards (RLVR) has emerged as a promising paradigm for improving mathematical reasoning in language models. Yet most RLVR work rewards only the final answer (outcome-based rewards), leaving the impact of step-level process supervision (process rewards) underexplored especially for small models that lack the capacity to self-correct under sparse feedback. We systematically compare five reward conditions applied to Qwen2.5-0.5B fine-tuned with Group Relative Policy Optimization (GRPO) on GSM8K: a no-RL baseline, process-only, outcome-only, and three hybrid weightings ($λ\in \{0.9, 0.5, 0.1\}$ process weight). Process-only supervision achieves 63.73% test accuracy versus 53.75% for outcome-only, a nearly 10-percentage point gap while yielding reasoning traces with higher step validity and lower deviation from ground-truth chain length. Hybrid rewards generally correlate positively with process weight, with one notable anomaly: the low-process / high-outcome configuration ($λ=0.1$) underperforms pure outcome supervision, suggesting conflicting optimization signals. Error analysis using GPT-4o as a judge reveals distinct failure mode distributions: process models generate structurally inconsistent but arithmetically grounded traces, while outcome models produce concise but derivation-error-prone chains. Our results demonstrate that reward granularity is a first-order design decision for RLVR, with process-level supervision substantially improving both accuracy and trace fidelity in small language models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。