Design Conditions for Intra-Group Learning of Sequence-Level Rewards: Token Gradient Cancellation
Organizations: Alibaba Group · Tsinghua University
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.
Figures & tables
| Method | Qwen3-32B Acc avg@32 (%) | Qwen3-Next Acc avg@32 (%) | GPT-OSS-120B Acc avg@32 (%) | ||||||
|---|---|---|---|---|---|---|---|---|---|
| AIME25 | LiveCodeBench | HMMT25 | AIME25 | LiveCodeBench | HMMT25 | AIME25 | LiveCodeBench | HMMT25 | |
| DFPO (Min-Replace) | 82.5 1.1 | 71.6 0.7 | 61.4 1.5 | 93.2 0.9 | 75.1 0.8 | 80.1 1.1 | 98.3 0.5 | 84.7 0.6 | 91.9 0.9 |
| DFPO (Orth-Proj) | 82.6 | 71.6 | 61.3 | 93.1 | 75.2 | 80.2 | 98.2 | 84.8 | 91.7 |
| GSPO | 76.9 1.3 | 64.7 1.5 | 55.8 0.9 | 89.8 1.7 | 71.0 1.4 | 75.8 1.2 | 96.5 0.5 | 80.3 1.2 | 87.6 1.0 |
| GRPO | 76.9 1.2 | 64.5 0.8 | 55.5 1.5 | 89.7 1.2 | 70.9 0.9 | 75.5 1.8 | 96.3 0.7 | 80.1 0.8 | 87.4 1.5 |
| GRPO-fix | 80.6 1.4 | 69.1 1.2 | 59.6 0.9 | 91.9 1.5 | 73.9 1.1 | 79.4 0.7 | 98.0 0.6 | 83.6 0.9 | 90.8 0.6 |
| Model variant | AIME25 (%) | LiveCodeBench (%) | HMMT25 (%) | |
|---|---|---|---|---|
| DFPO(Min-Replace) | 82.5 | 71.6 | 61.4 | |
| GSPO (baseline) | 76.9 | 64.7 | 55.8 | |
| DFPO (no stop-grad) | 75.6 | 63.8 | 54.9 | |
| DFPO (scale by 0.5) | 75.1 | 63.2 | 54.1 |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| G | 2 | 4 | 8 | 16 |
|---|---|---|---|---|
| avg@32 | 3.1 | 4.2 | 5.6 | 5.7 |
RL for Language Model Reasoning
Reinforcement learning for language model reasoning uses reward-driven training to improve multi-step problem solving, tool use, and reliable answers.
See all 1,087 papers