arXiv:2607.27631cs.AIcs.CL2026-07

用参考答案改进大模型数学推理中的价值估计和奖励分配。

ReDiPPO: Reference-Guided Value Calibration and Discrepancy-Aware Token Reweighting for Mathematical Reasoning

论文配图:ReDiPPO: Reference-Guided Value Calibration and Discrepancy-Aware Token Reweighting for Mathematical Reasoning
图 1 · 摘自论文原文
  • 引入参考答案指导价值网络,提升中间状态评估精度。
  • 通过标准与参考值差异识别难推理节点,动态重加权奖励信号。
  • 在多个数学推理数据集上优于PPO、DAPO等主流方法。

强化学习已成为提升大语言模型数学推理能力的有效范式。尽管近端策略优化(PPO)因其可提供逐标记信用分配而备受青睐,但在长推理路径和稀疏最终奖励的数学推理任务中,可靠的逐标记信用分配仍具挑战性。标准价值网络常无法准确评估中间推理状态,导致优势估计噪声大,策略更新效果差。本文提出ReDiPPO,一种基于参考答案引导与差异感知的PPO框架。ReDiPPO引入参考答案作为训练时的特权信号,构建参考引导价值网络以提升价值估计准确性;同时保留标准价值网络,并量化两者在标记级的差异,该差异作为困难推理状态的指示器,用于在PPO优化中重新加权对应标记的优势值。在多个数学推理基准上的实验证明,ReDiPPO显著提升了价值估计精度,且在最终推理性能上持续优于PPO、DAPO和GSPO等强基线。代码已开源:https://github.com/cii030/ReDiPPO。

原文摘要 · Abstract (English)

Reinforcement learning has emerged as an effective paradigm for enhancing the mathematical reasoning capabilities of large language models. Among existing policy optimization methods, Proximal Policy Optimization (PPO) remains particularly appealing because its learned critic can, in principle, provide token-level credit assignment. However, in mathematical reasoning tasks characterized by long reasoning horizons and sparse outcome rewards, reliable token-level credit assignment remains challenging. The standard critic often fails to accurately evaluate intermediate reasoning states, resulting in noisy advantage estimates and suboptimal policy updates. In this paper, we propose ReDiPPO, a Reference-guided and Discrepancy-aware PPO framework for mathematical reasoning. ReDiPPO introduces a reference-guided critic that uses reference answers as training-time privileged signals to provide more accurate value estimation. Meanwhile, it retains a standard critic and quantifies the token-level reference-standard discrepancy between the standard value estimate and the reference-guided value estimate. This discrepancy serves as an indicator of difficult reasoning states and is used to reweight the corresponding token-level advantages during PPO optimization. Extensive experiments on diverse mathematical reasoning benchmarks demonstrate that ReDiPPO improves value-estimation accuracy and consistently outperforms strong policy optimization baselines, including PPO, DAPO, and GSPO, in final reasoning performance. Our code is available on https://github.com/cii030/ReDiPPO.

数学推理强化学习价值估计PPO

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