On the Policy Gradient Foundations of Group Relative Policy Optimization: Credit Assignment, Gradient Sparsity, and Rank Collapse
On the map: LLM reasoning → Reinforcement learning for reasoning
Abstract (arXiv)
Group Relative Policy Optimization (GRPO) eliminates the learned critic in PPO by using the mean reward of grouped rollouts as a baseline. We provide a rigorous derivation of GRPO from first principles of the policy gradient theorem, revealing a fundamental credit assignment failure: under output-only reward, every token in a rollout receives identical advantage, collapsing token-level credit to a single scalar. We prove this induces gradient sparsity that intensifies over training, and demonstrate empirically via SVD analysis of GRPO gradients on Nemotron-4B/GSM8K that the gradient matrix has effective rank ≈ 2 regardless of group size R ∈ 2, 4, 8. We formalize this as an intrinsic rank-2 structure arising from the zero-sum constraint on advantages and derive conditions under which GRPO's baseline is optimal. Our results characterize when GRPO's simplicity is theoretically justified and identify the credit assignment bottleneck as the key limitation for multi-step reasoning.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 9 of its 19 references that are in the map. First posted on arXiv 28 Jun 2026.
Builds on
9 of 19 references.
Closest ancestors · 3
Same direction, newest first
- 2023Let's Verify Step by Step31 May
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Model29 May
- 2017Proximal Policy Optimization Algorithms20 Jul
Other references · 6
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2015High-Dimensional Continuous Control Using Generalized Advantage Estimationcited by 2,026
- 2018Measuring the Intrinsic Dimension of Objective Landscapescited by 370
- 2013The Optimal Reward Baseline for Gradient-Based Reinforcement Learningcited by 85
- 2024Nemotron-4 15B Technical Reportcited by 36
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Advantage Scale Calibration Imbalance in Group-Relative Optimization under Low-Variance Rewards: Diagnosis and Bounded Recoverycited by 1
- 2026EvoSteer: Online Self-Evolving Graph Orchestration via Reference-Anchored Credit Assignmentcited by 0
- 2026DiffGate: Difficulty-Gated Teacher Guidance for On-Policy Distillationcited by 0
- 2026DiffGate: Difficulty-Gated Teacher Guidance for On-Policy Distillation3 Oct 2026
- 2026EvoSteer: Online Self-Evolving Graph Orchestration via Reference-Anchored Credit Assignment29 Sep 2026
- 2026Advantage Scale Calibration Imbalance in Group-Relative Optimization under Low-Variance Rewards: Diagnosis and Bounded Recovery27 Aug 2026
Similar, not cited
Close in wording, no citation either way
- 2026Constrained Group Relative Policy Optimization
- 2025Revisiting Group Relative Policy Optimization: Insights into On-Policy and Off-Policy Training
- 2026Demystifying Group Relative Policy Optimization: Its Policy Gradient is a U-Statistic
- 2026EP-GRPO: Entropy-Progress Aligned Group Relative Policy Optimization with Implicit Process Guidance
- 2025Learning Without Critics? Revisiting GRPO in Classical Reinforcement Learning Environments
- 2026Gradient Starvation in Binary-Reward GRPO: Why Group-Mean Centering Fails and Why the Simplest Fix Works
- 2026Advantage Collapse in Group Relative Policy Optimization: Diagnosis and Mitigation
- 2026On the Impossibility of Unbiased and Length-Invariant Policy Optimization with Outcome Rewards