arXiv:2605.16302cs.LGcs.AI2026-05

用反事实路径降低大模型推理中的信用分配方差,提升训练稳定性和性能上限。

Reducing Credit Assignment Variance via Counterfactual Reasoning Paths

论文配图:Reducing Credit Assignment Variance via Counterfactual Reasoning Paths
图 1 · 摘自论文原文
  • 通过采样多条推理路径,用差异模拟替代决策,生成细粒度学习信号。
  • 在数学与代码推理任务上,训练更稳定,性能显著优于基线方法。
  • 适合研究大模型推理优化与强化学习的开发者参考。

基于大语言模型(LLMs)的多步推理强化学习通常依赖稀疏的最终奖励,导致信用分配问题严重:最终反馈被均匀传播至所有中间决策。这引发高梯度方差、训练不稳定及大量无效更新,限制了模型持续改进。本文提出一种反事实比较框架,对每个输入采样多条推理轨迹,将其差异视为替代决策的隐式近似,从而生成过程级的优势估计量,将稀疏终端奖励转化为对每一步敏感的学习信号。在此基础上,提出隐式行为策略优化(IBPO),显著提升训练稳定性与数学及代码推理基准上的性能上限。结果表明该方向有望释放LLMs的推理潜力。

原文摘要 · Abstract (English)

Reinforcement learning for multi-step reasoning with large language models (LLMs) typically relies on sparse terminal rewards, which creates a poorly conditioned credit-assignment problem: the final feedback is propagated uniformly across all intermediate decisions. This leads to high gradient variance, unstable training, and many ineffective updates, ultimately limiting sustained model improvement. We propose a counterfactual-comparison framework for credit assignment. For each input, the framework samples multiple reasoning trajectories and treats their differences as implicit approximations to alternative decisions. This yields an implicit process-level advantage estimator that converts sparse terminal rewards into step-sensitive learning signals. Building on this framework, we introduce Implicit Behavior Policy Optimization (IBPO), which substantially improves training stability and the performance ceiling on mathematical and code-reasoning benchmarks. Our results point to a promising direction for unlocking the reasoning potential of LLMs.

强化学习大模型推理信用分配

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