mlsanityData to 7 Oct 2026

AMR-SD: Asymmetric Meta-Reflective Self-Distillation for Token-Level Credit Assignment

arXiv 2605.18529 ↗First posted 18 May 2026cs.AIcited by 1

On the map: LLM reasoning → Reinforcement learning for reasoning

also in 1 more
Abstract (arXiv)

The alignment of Large Language Models (LLMs) for complex reasoning heavily relies on Reinforcement Learning with Verifiable Rewards (RLVR). However, standard algorithms like GRPO apply sequence-level rewards uniformly to all tokens, creating a severe credit-assignment bottleneck. While on-policy self-distillation attempts to resolve this by conditioning a self-teacher on privileged contexts, direct exposure to raw oracle solutions often induces over-conditioned teacher distributions, implicit answer leakage, and late-stage training collapse. To overcome these limitations, we propose Asymmetric Meta-Reflective Self-Distillation (AMR-SD). Instead of conditioning directly on raw reference traces, AMR-SD inserts a reflection bottleneck: it compresses diagnostic signals -- from verifier outcomes, peer rollouts, or reference feedback -- into concise, self-generated Socratic hints and critiques. Furthermore, we introduce Causal Information Gain (CIG) with an asymmetric, ReLU-gated threshold to translate these reflections into sparse, highly precise token-level advantage modulations. Combined with temporal annealing, this mechanism preserves the base environmental reward while filtering out distributional noise. Experiments across scientific, mathematical, and tool-use benchmarks demonstrate that AMR-SD significantly outperforms existing baselines, achieving robust long-horizon stability and successfully preventing late-stage collapse.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 16 of its 26 references that are in the map. First posted on arXiv 18 May 2026.

Builds on

16 of 26 references.

Closest ancestors · 7

Same direction, newest first

  1. 2026ResRL: Boosting LLM Reasoning via Negative Sample Projection Residual Reinforcement Learning1 May
  2. 2026Unifying Group-Relative and Self-Distillation Policy Optimization via Sample Routing2 Apr
  3. 2026Contextual Rollout Bandits for Reinforcement Learning with Verifiable Rewards9 Feb
  4. 2026Outcome-Grounded Advantage Reshaping for Fine-Grained Credit Assignment in Mathematical Reasoning12 Jan
  5. 2025Unlocking Exploration in RLVR: Uncertainty-aware Advantage Shaping for Deeper Reasoning12 Oct
  6. 2025RLPR: Extrapolating RLVR to General Domains without Verifiers23 Jun
  7. 2017Proximal Policy Optimization Algorithms20 Jul

Other references · 9

  1. 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
  2. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 840
  3. 2023ToolAlpaca: Generalized Tool Learning for Language Models with 3000 Simulated Casescited by 383
  4. 2026Reinforcement Learning via Self-Distillationcited by 285
  5. 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipecited by 262
  6. 2025The Invisible Leash: Why RLVR May or May Not Escape Its Origincited by 92
  7. 2026Why Does Self-Distillation (Sometimes) Degrade the Reasoning Capability of LLMs?cited by 87
  8. 2026RubricHub: A Comprehensive and Highly Discriminative Rubric Dataset via Automated Coarse-to-Fine Generationcited by 42
  9. 2026Embarrassingly Simple Self-Distillation Improves Code Generationcited by 36

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2026Localizing Credit at the Divergence: Path-Conditioned Self-Distillation for LLM Reasoning
  2. 2026Learning from Own Solutions: Self-Conditioned Credit Assignment for Reinforcement Learning with Verifiable Rewards
  3. 2025Exploiting Tree Structure for Credit Assignment in RL Training of LLMs
  4. 2026Not All Tokens Deserve Equal Credit: Counterfactual Sensitivity Credit Reallocation for Long-CoT Reasoning
  5. 2026OPPO: Bayesian Value Recursion for Token-Level Credit Assignment in LLM Reasoning
  6. 2026SRPO: Self-Reflective Policy Optimization for Long-Horizon Reasoning
  7. 2026InT: Self-Proposed Interventions Enable Credit Assignment in LLM Reasoning
  8. 2026PACT: From Credit Assignment to Critic Alignment