CRAFT: Counterfactual Credit Assignment from Free Sibling Rollouts for Self-Distilled Agentic Reinforcement Learning
On the map: LLM agents and tool use → Reinforcement learning for LLM agents
also in 1 more
Abstract (arXiv)
Self-distilled agentic reinforcement learning augments trajectory-level reward with a token-level distillation loss, using as its teacher the same policy conditioned on privileged context. The prevailing recipe gates this loss by a single scalar, the teacher-student log-probability gap. This signal is doubly limited: it is retrospective, scoring only the realised rollout and never the counterfactual ones, and it is sign-blind, never signalling when a teacher-preferred action would have harmed the trajectory. We introduce CRAFT, a three-pillar credit-assignment scheme that addresses both limitations. Pillar 1, Counterfactual Token Importance, reuses the G-1 sibling rollouts that GRPO already samples and importance-weights them by the log-probability gap to form a self-normalised estimate of the group-level counterfactual change in advantage from up-weighting teacher-preferred actions at each step; this yields a signed per-token credit at near-zero extra compute. Pillar 2 is an asymmetric controller that raises the distillation weight as it lowers the reference-KL weight along an exponential moving average of gate activity, and conversely. Pillar 3 polarises the KL penalty token by token, switching between a mode-seeking and a mode-covering update according to the sign of the credit. Each pillar has an independent switch that, when disabled, renders the loss and gradient byte-identical to the baseline in IEEE-754 arithmetic, so any measured gain is attributable to algorithmic change rather than implementation drift. We prove the estimator's consistency and a variance bound, give structural and bit-exact reproducibility guarantees, and evaluate CRAFT across three agentic environments, four model scales, and five end-to-end methods, plus two tabulated prior-work baselines. Among these is Adaptive-CRINGE, a comparator sharing Pillar 2 with CRAFT, isolating the counterfactual contribution.
Key facts. Cited by 8 papers in the map (data through 14 Oct 2026). Placed in Reinforcement learning for LLM agents, one of 16 directions in LLM agents and tool use. Builds on 21 of its 40 references that are in the map. First posted on arXiv 28 Jun 2026.
Builds on
21 of 40 references.
Closest ancestors · 1
Same direction, newest first
Other references · 20
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2025Qwen3 Technical Reportcited by 8,820
- 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,695
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2023Toolformer: Language Models Can Teach Themselves to Use Toolscited by 4,182
- 2023Let's Verify Step by Stepcited by 3,789
- 2022Constitutional AI: Harmlessness from AI Feedbackcited by 3,050
- 2015Trust Region Policy Optimizationcited by 2,580
- 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,444
8 more
- 2020ALFWorld: Aligning Text and Embodied Environments for Interactive Learningcited by 1,205
- 2023Math-Shepherd: Verify and Reinforce LLMs Step-by-step without Human Annotationscited by 928
- 2024Self-Play Fine-Tuning Converts Weak Language Models to Strong Language Modelscited by 557
- 2019Neural Text Generation with Unlikelihood Trainingcited by 503
- 2022ScienceWorld: Is your Agent Smarter than a 5th Grader?cited by 327
- 2024Quiet-STaR: Language Models Can Teach Themselves to Think Before Speakingcited by 301
- 2018BabyAI: A Platform to Study the Sample Efficiency of Grounded Language Learningcited by 244
- 2018Woulda, Coulda, Shoulda: Counterfactually-Guided Policy Searchcited by 82
Built on by
Citing papers per year: 2026: 8.
8 papers.
- 2026CROP: Task Relevance via Counterfactuals for Selective On-Policy Distillationcited by 5
- 2026Reconciling Process Supervision with Outcome-Based Credit in Agentic Policy Optimizationcited by 4
- 2026Agentic Reinforcement Learning with Self-Distilled Reward Shapingcited by 3
- 2026Trust Is Not Enough: Influence Calibration for On-Policy Self-Distillation in Agentic RLcited by 3
- 2026When Does Muon Help Agentic Reinforcement Learning?cited by 1
- 2026SkillGate: Training In-Policy Skill Selection in Long-Horizon Agentscited by 1
- 2026IGSD: Environment-Verified Hindsight Self-Distillation for Search Agentscited by 0
- 2026Of Errors and Echoes: Responsibility-Aware Credit Assignment for Tool-Use Agentscited by 0
- 2026Of Errors and Echoes: Responsibility-Aware Credit Assignment for Tool-Use Agents12 Oct 2026
- 2026IGSD: Environment-Verified Hindsight Self-Distillation for Search Agents26 Sep 2026
- 2026Reconciling Process Supervision with Outcome-Based Credit in Agentic Policy Optimization31 Aug 2026
- 2026SkillGate: Training In-Policy Skill Selection in Long-Horizon Agents19 Aug 2026
- 2026Trust Is Not Enough: Influence Calibration for On-Policy Self-Distillation in Agentic RL14 Aug 2026
- 2026CROP: Task Relevance via Counterfactuals for Selective On-Policy Distillation13 Aug 2026
- 2026Agentic Reinforcement Learning with Self-Distilled Reward Shaping4 Aug 2026
- 2026When Does Muon Help Agentic Reinforcement Learning?17 Jul 2026
Similar, not cited
Close in wording, no citation either way
- 2026AgentOPSD: Recursive Self-Distillation for Agentic Reinforcement Learning
- 2026Keep Policy Gradient in Charge: Sibling-Guided Credit Distillation for Long-Horizon Tool-Use Agents
- 2026StepOPSD: Step-Aware Online Preference Self-Distillation for Agent Reinforcement Learning
- 2026RetireOPD: Self-Retiring On-Policy Distillation for Agentic Reinforcement Learning
- 2026RSIPO: Recursive Self-Improving Policy Optimization for Agentic Reinforcement Learning via Skill-Policy Co-Evolution
- 2026Group-Reflective Self-Distillation for Agentic Reinforcement Learning
- 2026TRACE: Turn-level Reward Assignment via Credit Estimation for Long-Horizon Agents
- 2026How Much, Then Where: Credit-Conserving Action-to-Token Allocation for Multi-Turn Agent Reinforcement Learning