Agentic Reinforcement Learning with Self-Distilled Reward Shaping
On the map: LLM agents and tool use → Reinforcement learning for LLM agents
Abstract (arXiv)
Agentic reinforcement learning enables LLM agents to learn through interaction, but sparse trajectory-level rewards reveal success without identifying which intermediate decisions deserve credit. Training-only privileged skills can provide denser supervision by allowing the same frozen policy snapshot to rescore fixed tokens from skill-free trajectories while conditioned on task-matched procedural skills. Existing methods, however, do not jointly calibrate teacher scores across interaction steps, relate teacher confidence to realized returns, and integrate the resulting signal into native reward-to-advantage construction. We introduce Agentic Reinforcement Learning with Self-Distilled Reward Shaping (ADRS), a framework for constructing return-associated token-level credit for multi-turn language agents. ADRS centers and normalizes privileged token scores within each step, modulates them with a return-associated Teacher Value Advantage (TVA) gate based on within-group confidence--return association, and incorporates the gated token signal into native RL credit construction. Together, these components determine what the teacher prefers, when that preference is return-relevant, and how it enters the native reinforcement-learning credit path, while keeping rollouts and inference skill-free. Finally, experiments across three interactive benchmarks show that ADRS consistently improves performance on long-horizon tasks, with gains persisting across RL backbones, reduced-data settings, unseen tasks, and extended training. For anonymous review, our code is available at the following the link: https://github.com/gitrxh/ADRS-arxiv
Key facts. Cited by 3 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 18 of its 39 references that are in the map. First posted on arXiv 4 Aug 2026.
Builds on
18 of 39 references.
Closest ancestors · 4
Same direction, newest first
- 2026CRAFT: Counterfactual Credit Assignment from Free Sibling Rollouts for Self-Distilled Agentic Reinforcement Learning28 Jun
- 2025Agent Lightning: Train ANY AI Agents with Reinforcement Learning5 Aug
- 2025Group-in-Group Policy Optimization for LLM Agent Training16 May
- 2025Reinforcement Learning for Long-Horizon Interactive LLM Agents3 Feb
Other references · 14
- 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
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,754
- 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,955
- 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,444
- 2020Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Stepscited by 1,282
- 2020ALFWorld: Aligning Text and Embodied Environments for Interactive Learningcited by 1,205
- 2022When Not to Trust Language Models: Investigating Effectiveness of Parametric and Non-Parametric Memoriescited by 1,128
- 2022Measuring and Narrowing the Compositionality Gap in Language Modelscited by 1,075
- 2021MuSiQue: Multihop Questions via Single-hop Question Compositioncited by 916
- 2026Reinforcement Learning via Self-Distillationcited by 332
2 more
- 2026SkillRL: Evolving Agents via Recursive Skill-Augmented Reinforcement Learningcited by 278
- 2026TIP: Token Importance in On-Policy Distillationcited by 104
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026UniOPSD: Unifying Outcome and Hindsight Feedback for Agentic Reinforcement Learningcited by 0
- 2026From Given to Gathered Evidence: Agentic Learning for Longitudinal Medical Reasoningcited by 0
- 2026More Methods, Better Agents? A Reality Check on Self-Distillation in Agentic Reinforcement Learningcited by 0
- 2026More Methods, Better Agents? A Reality Check on Self-Distillation in Agentic Reinforcement Learning14 Oct 2026
- 2026From Given to Gathered Evidence: Agentic Learning for Longitudinal Medical Reasoning30 Sep 2026
- 2026UniOPSD: Unifying Outcome and Hindsight Feedback for Agentic Reinforcement Learning28 Sep 2026
Similar, not cited
Close in wording, no citation either way
- 2026AgentOPSD: Recursive Self-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
- 2026Self-Distilled Agentic Reinforcement Learning
- 2026Skill-SD: Skill-Conditioned Self-Distillation for Multi-turn LLM Agents
- 2026RetireOPD: Self-Retiring On-Policy Distillation for Agentic Reinforcement Learning
- 2026Dependency-Aware Reward Shaping for Agentic Reinforcement Learning
- 2026UCOB: Learning to Utilize and Evolve Agentic Skills via Credit-Aware On-Policy Bidirectional Self-Distillation