Teacher-Guided Tool Learning for Small Language Models via Bidirectional KL Constraints
On the map: LLM agents and tool use → Reinforcement learning for LLM agents
also in 1 more
Abstract (arXiv)
External tools can extend the capabilities of small language models (SLMs), enabling their deployment in resource constrained environments. However, their limited capabilities make it difficult for SLMs to discover effective tool use behaviors through reinforcement learning alone, while direct teacher supervision suffers from teacher student behavioral distribution mismatch and may propagate erroneous preferences. To address these challenges, we propose Selective Teacher guided Policy Optimization (SeTPO), which coordinates student exploration with selective teacher guidance and persistent demonstration supervision. Specifically, SeTPO incorporates reverse KL derived token level teacher feedback into policy advantages on low reward trajectories, while auxiliary forward KL supervision preserves teacher demonstrated behaviors throughout training. Under an idealized bidirectional KL objective, we theoretically establish that the optimal policy achieves expected reward no lower than the teacher's while retaining a guaranteed fraction of the teacher's probability mass. Extensive experiments on BFCL v4 and API Bank demonstrate that SeTPO achieves superior overall performance over supervised fine tuning, reinforcement learning, and distillation baselines across all three student scales.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Reinforcement learning for LLM agents, one of 16 directions in LLM agents and tool use. Builds on 23 of its 34 references that are in the map. First posted on arXiv 13 Oct 2026.
Builds on
23 of 34 references.
Closest ancestors · 3
Same direction, newest first
- 2026Tool-R0: Self-Evolving LLM Agents for Tool-Learning from Zero Data24 Feb
- 2025Can Compact Language Models Search Like Agents? Distillation-Guided Policy Optimization for Preserving Agentic RAG Capabilities27 Aug
- 2025ToolRL: Reward is All Tool Learning Needs16 Apr
Other references · 20
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 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
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 939
- 2015Scheduled Sampling for Sequence Prediction with Recurrent Neural Networkscited by 803
- 2023API-Bank: A Comprehensive Benchmark for Tool-Augmented LLMscited by 591
- 2025Learning to Reason under Off-Policy Guidancecited by 281
- 2024APIGen: Automated Pipeline for Generating Verifiable and Diverse Function-Calling Datasetscited by 198
- 2025RL's Razor: Why Online Reinforcement Learning Forgets Lesscited by 176
- 2025APIGen-MT: Agentic Pipeline for Multi-Turn Data Generation via Simulated Agent-Human Interplaycited by 167
- 2025Learning What Reinforcement Learning Can't: Interleaved Online Fine-Tuning for Hardest Questionscited by 97
- 2026Unifying Group-Relative and Self-Distillation Policy Optimization via Sample Routingcited by 76
8 more
- 2025DeepAgent: A General Reasoning Agent with Scalable Toolsetscited by 71
- 2025Distilling LLM Agent into Small Models with Retrieval and Code Toolscited by 47
- 2025KDRL: Post-Training Reasoning LLMs via Unified Knowledge Distillation and Reinforcement Learningcited by 41
- 2025TAID: Temporally Adaptive Interpolated Distillation for Efficient Knowledge Transfer in Language Modelscited by 22
- 2025LoopTool: Closing the Data-Training Loop for Robust LLM Tool Callscited by 17
- 2026Distill Where You Fail: Recovering Learning Signals of Negative RL-Groups from Adaptive Teacher Guidancecited by 11
- 2026Teaching LLMs to Learn Tool Trialing and Execution through Environment Interactioncited by 3
- 2026STAR: Similarity-guided Teacher-Assisted Refinement for Super-Tiny Function Calling Modelscited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025MENTOR: Reinforcement Learning via Flexible Teacher-Optimized Rewards for Tool-Use Distillation
- 2026Teacher-Guided Policy Optimization for On-Policy Reasoning Distillation under Large Policy Divergence
- 2026DiffGate: Difficulty-Gated Teacher Guidance for On-Policy Distillation
- 2026SocraticPO: Policy Optimization via Interactive Guidance
- 2023TGRL: An Algorithm for Teacher Guided Reinforcement Learning
- 2026Learn from Whoever Is Right: Answer-Verified Multi-Teacher Distillation for Multi-Domain LLMs
- 2026EasyOPD: An Easy-to-use On-Policy Distillation Framework for Large Language Models
- 2026MOPD: Multi-Teacher On-Policy Distillation for Capability Integration in LLM Post-Training