English

Reducing Credit Assignment Variance via Counterfactual Reasoning Paths

Machine Learning 2026-05-26 v2 Artificial Intelligence Computation and Language

Abstract

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.

Keywords

Cite

@article{arxiv.2605.16302,
  title  = {Reducing Credit Assignment Variance via Counterfactual Reasoning Paths},
  author = {Fei Ding and Yongkang Zhang and Youwei Wang and Zijian Zeng},
  journal= {arXiv preprint arXiv:2605.16302},
  year   = {2026}
}
R2 v1 2026-07-22T07:15:12.714Z