mlsanityData to 7 Oct 2026

Outcome-Grounded Advantage Reshaping for Fine-Grained Credit Assignment in Mathematical Reasoning

arXiv 2601.07408 ↗First posted 12 Jan 2026cs.CL cs.LGcited by 18

On the map: LLM reasoning → Reinforcement learning for reasoning

Abstract (arXiv)

Group Relative Policy Optimization (GRPO) has emerged as a promising critic-free reinforcement learning paradigm for reasoning tasks. However, standard GRPO employs a coarse-grained credit assignment mechanism that propagates group-level rewards uniformly to to every token in a sequence, neglecting the varying contribution of individual reasoning steps. We address this limitation by introducing Outcome-grounded Advantage Reshaping (OAR), a fine-grained credit assignment mechanism that redistributes advantages based on how much each token influences the model's final answer. We instantiate OAR via two complementary strategies: (1) OAR-P, which estimates outcome sensitivity through counterfactual token perturbations, serving as a high-fidelity attribution signal; (2) OAR-G, which uses an input-gradient sensitivity proxy to approximate the influence signal with a single backward pass. These importance signals are integrated with a conservative Bi-Level advantage reshaping scheme that suppresses low-impact tokens and boosts pivotal ones while preserving the overall advantage mass. Empirical results on extensive mathematical reasoning benchmarks demonstrate that while OAR-P sets the performance upper bound, OAR-G achieves comparable gains with negligible computational overhead, both significantly outperforming a strong GRPO baseline, pushing the boundaries of critic-free LLM reasoning.

Key facts. Cited by 18 papers in the map (data through 7 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 23 of its 34 references that are in the map. First posted on arXiv 12 Jan 2026.

Builds on

23 of 34 references.

Closest ancestors · 10

Same direction, newest first

  1. 2025Beyond High-Entropy Exploration: Correctness-Aware Low-Entropy Segment-Based Advantage Shaping for Reasoning LLMs30 Nov
  2. 2025Unlocking Exploration in RLVR: Uncertainty-aware Advantage Shaping for Deeper Reasoning12 Oct
  3. 2025RLPR: Extrapolating RLVR to General Domains without Verifiers23 Jun
  4. 2025Reasoning with Exploration: An Entropy Perspective17 Jun
  5. 2025KTAE: A Model-Free Algorithm to Key-Tokens Advantage Estimation in Mathematical Reasoning22 May
  6. 2025General-Reasoner: Advancing LLM Reasoning Across All Domains20 May
  7. 2025Understanding R1-Zero-Like Training: A Critical Perspective26 Mar
  8. 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scale18 Mar
  9. 2025Process Reinforcement through Implicit Rewards3 Feb
  10. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning22 Jan

Other references · 13

  1. 2020Language Models are Few-Shot Learnerscited by 32,199
  2. 2023GPT-4 Technical Reportcited by 18,132
  3. 2021Training Verifiers to Solve Math Word Problemscited by 9,743
  4. 2021Evaluating Large Language Models Trained on Codecited by 8,053
  5. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
  6. 2017Axiomatic Attribution for Deep Networkscited by 3,350
  7. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
  8. 2019Attention is not Explanationcited by 916
  9. 2025Qwen2.5-1M Technical Reportcited by 158
  10. 2024Generative Reward Modelscited by 148
  11. 2024Preserving Diversity in Supervised Fine-Tuning of Large Language Modelscited by 68
  12. 2024TIS-DPO: Token-level Importance Sampling for Direct Preference Optimization With Estimated Weightscited by 42
1 more
  1. 2025Enhancing Large Language Model Reasoning via Selective Critical Token Fine-Tuningcited by 20

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 18.

18 papers.

Sort:

By area:

  1. 2026Self-Distilled RLVRcited by 109
  2. 2026What and When to Distill: Selective Hindsight Distillation for Multi-Turn Agentscited by 9
  3. 2026Reinforcement Learning from Rich Feedback with Distributional DAggercited by 9
  4. 2026Distill Where You Fail: Recovering Learning Signals of Negative RL-Groups from Adaptive Teacher Guidancecited by 8
  5. 2026Look Ahead Before You Distill: Future Trajectory Validation of Teacher Guidance for Agentic On-Policy Distillationcited by 8
  6. 2026Self-Distilled Policy Gradientcited by 6
  7. 2026GRAIL: Gradient-Reweighted Advantages for Reinforcement Learning with Verifiable Rewardscited by 2
  8. 2026AMR-SD: Asymmetric Meta-Reflective Self-Distillation for Token-Level Credit Assignmentcited by 1
  9. 2026Trait-Aware Policy Optimization for Autoregressive Multi-Trait Essay Scoringcited by 1
  10. 2026RREDCoT: Segment-Level Reward Redistribution for Reasoning Modelscited by 1
  11. 2026When Implausible Tokens Get Reinforced: Tail-Aware Credit Calibration for LLM Reinforcement Learningcited by 1
  12. 2026IAPO: Influence-Aware Policy Optimization for Credit Assignment in Multi-Turn Service Agentscited by 1
6 more
  1. 2026Hidden States Know Where Reasoning Diverges: Credit Assignment via Span-Level Wasserstein Distancecited by 0
  2. 2026CRPO: Character-centric Group Relative Policy Optimization for Role-aware Reasoning in Role-playing Agentscited by 0
  3. 2026Let Credit Follow Computation: Architecture-Aware Credit Transport for Large Language Model Reinforcement Learningcited by 0
  4. 2026SGA-Flow-GRPO: Spatial Gradient-Guided Credit Assignment for Flow-GRPOcited by 0
  5. 2026Surprising Success, Repeated Failure: Entropy-Guided Credit Assignment for Exploration in LLM Reasoningcited by 0
  6. 2026Semifactual Credit-Augmented Policy Optimizationcited by 0

Similar, not cited

Close in wording, no citation either way

  1. 2026On the Policy Gradient Foundations of Group Relative Policy Optimization: Credit Assignment, Gradient Sparsity, and Rank Collapse
  2. 2025Mitigating Think-Answer Mismatch in LLM Reasoning Through Noise-Aware Advantage Reweighting
  3. 2026Advantage Collapse in Group Relative Policy Optimization: Diagnosis and Mitigation
  4. 2026InT: Self-Proposed Interventions Enable Credit Assignment in LLM Reasoning
  5. 2026PrAg-PO: Prompt Augmented Policy Optimization for Robust and Diverse Mathematical Reasoning
  6. 2026Targeting Pivotal Decisions for Credit Assignment in Agentic Reinforcement Learning
  7. 2026DGPO: Distribution Guided Policy Optimization for Fine Grained Credit Assignment
  8. 2026OPPO: Bayesian Value Recursion for Token-Level Credit Assignment in LLM Reasoning