arXiv:2605.06755cs.LGcs.AI2026-05

用三次反向传播实现更优策略更新,加速大模型推理强化学习

Gradient Extrapolation-Based Policy Optimization

论文配图:Gradient Extrapolation-Based Policy Optimization
图 1 · 摘自论文原文
  • 仅用三次反向传播近似多步前瞻,复用原有数据不需新采样
  • 在数学推理任务上提升1.65至5.00分,速度提升达4倍
  • 自动切换回标准方法应对信号不稳定,适合高效训练大模型

强化学习广泛用于提升大语言模型的推理能力,尤其在答案可自动验证时。标准GRPO训练仅使用当前步骤更新,而完整多步前瞻虽方向更优但代价过高,需多次反向传播。本文提出梯度外推策略优化(GXPO),一种与GRPO兼容的策略更新规则。GXPO在活跃阶段仅需三次反向传播,通过复用同一组轨迹、奖励、优势和GRPO损失,无需新增采样或奖励计算,即可近似更长的局部前瞻。它执行两次快速优化步,测量梯度变化,预测虚拟K步前瞻点,将策略部分推向该点,并在新位置用真实梯度进行修正。当前瞻信号不稳定时,自动切换回单步GRPO。我们还给出基于纯梯度下降的代理分析,解释外推何时精确及局部误差来源。在Qwen2.5与Llama数学推理实验中,GXPO相较GRPO提升平均采样pass@1得分1.65至5.00点,较最强SFPO设置提升0.14至1.28点,且活跃阶段成本固定为三次反向传播。同时实现最高4.00倍步数加速、2.33倍实际运行时间加速和1.33倍反向传播加速,以达到GRPO峰值精度。

原文摘要 · Abstract (English)

Reinforcement learning is widely used to improve the reasoning ability of large language models, especially when answers can be automatically checked. Standard GRPO-style training updates the model using only the current step, while full multi-step lookahead can give a better update direction but is too expensive because it needs many backward passes. We propose Gradient Extrapolation-Based Policy Optimization (GXPO), a plug-compatible policy-update rule for GRPO-style reasoning RL. GXPO approximates a longer local lookahead using only three backward passes during an active phase. It reuses the same batch of rollouts, rewards, advantages, and GRPO loss, so it does not require new rollouts or reward computation at the lookahead points. GXPO takes two fast optimizer steps, measures how the gradients change, predicts a virtual K-step lookahead point, moves the policy partway toward that point, and then applies a corrective update using the true gradient at the new position. When the lookahead signal becomes unstable, GXPO automatically switches back to standard single-pass GRPO. We also give a plain-gradient-descent surrogate analysis that explains when the extrapolation is exact and where its local errors come from. Across Qwen2.5 and Llama math-reasoning experiments, GXPO improves the average sampled pass@1 by +1.65 to +5.00 points over GRPO and by +0.14 to +1.28 points over the strongest SFPO setting, while keeping the active-phase cost fixed at three backward passes. It also achieves up to 4.00x step speedup, 2.33x wall-clock speedup, and 1.33x backward-pass speedup in reaching GRPO's peak accuracy.

强化学习大模型推理策略优化加速训练

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