Teacher-Guided Policy Optimization for On-Policy Reasoning Distillation under Large Policy Divergence
On the map: LLM reasoning → Prompting and chain-of-thought reasoning
also in 1 more
Abstract (arXiv)
On-policy distillation (OPD) has become a promising paradigm for reasoning-oriented post-training of large language models (LLMs), especially when combined with reinforcement learning from verifiable rewards (RLVR). Existing OPD methods rely on reverse KL (RKL)-based teacher supervision over trajectories sampled from the student policy. However, we identify a critical limitation: under large teacher--student policy divergence, RL-driven exploration often produces trajectories outside the teacher distribution, resulting in uninformative negative feedback. To address this, we propose Teacher-Guided Policy Optimization (TGPO), an on-policy reasoning distillation method that remains effective under large policy divergence settings. Rather than relying solely on evaluative supervision, TGPO uses teacher to directly guide token level generation conditioning on student-generated contexts; together with RLVR-style trajectory level rewards, TGPO steers exploration toward improved continuations. Experiments on reasoning benchmarks show that TGPO consistently outperforms existing RKL-based OPD methods and remains robust across different teacher models.
Key facts. Cited by 13 papers in the map (data through 14 Oct 2026). Placed in Prompting and chain-of-thought reasoning, one of 17 directions in LLM reasoning. Builds on 17 of its 29 references that are in the map. First posted on arXiv 13 May 2026.
Builds on
17 of 29 references.
Closest ancestors · 1
Same direction, newest first
Other references · 16
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2025Qwen3 Technical Reportcited by 8,820
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
- 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,998
- 2025Kimi k1.5: Scaling Reinforcement Learning with LLMscited by 1,210
- 2024Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvementcited by 1,090
- 2025Process Reinforcement through Implicit Rewardscited by 504
- 2026Reinforcement Learning via Self-Distillationcited by 332
- 2026MiMo-V2-Flash Technical Reportcited by 191
- 2025Skywork Open Reasoner 1 Technical Reportcited by 165
4 more
- 2023MiniLLM: On-Policy Distillation of Large Language Modelscited by 139
- 2025On-Policy RL Meets Off-Policy Experts: Harmonizing Supervised Fine-Tuning and Reinforcement Learning via Dynamic Weightingcited by 91
- 2025No Prompt Left Behind: Exploiting Zero-Variance Prompts in LLM Reinforcement Learning via Entropy-Guided Advantage Shapingcited by 50
- 2025KDRL: Post-Training Reasoning LLMs via Unified Knowledge Distillation and Reinforcement Learningcited by 41
Built on by
Citing papers per year: 2026: 13.
13 papers.
Sort:
By area:
- 2026Mismatch Matters: On-Policy Distillation Beyond Token Agreementcited by 10
- 2026PowerOPD: Stabilizing On-Policy Distillation with Bounded Power Transformationcited by 9
- 2026A Formula-Driven Survey and Research Agenda for On-Policy Distillationcited by 6
- 2026OPDSearch+: On-Policy Distillation with RL Refinement for Search-Augmented Reasoningcited by 3
- 2026TISD: On-Policy Self-Distillation with Trajectory Interventioncited by 1
- 2026Expected Reasoning-Step Return Unifies On-Policy Learning from Rewards and Teacherscited by 0
- 2026TIDE: Teacher-Student Transition via Informative Distillation and Exploration for Agentic RLcited by 0
- 2026Improving OCR Faithfulness via Gated and Attenuated On-Policy Distillationcited by 0
- 2026From Given to Gathered Evidence: Agentic Learning for Longitudinal Medical Reasoningcited by 0
- 2026DiffGate: Difficulty-Gated Teacher Guidance for On-Policy Distillationcited by 0
- 2026PIVOT: Perplexity-Informed KD-to-RL Transition Scheduling for Vertical-Domain Few-Shot Distillationcited by 0
- 2026Demystifying On-Policy Distillation: Where Does the Teacher's Advantage Live?cited by 0
- 2026DTDD: Divergence-Triggered Dynamic Distillation for Reliable On-Policy Supervision13 Oct 2026
- 2026Demystifying On-Policy Distillation: Where Does the Teacher's Advantage Live?13 Oct 2026
- 2026PIVOT: Perplexity-Informed KD-to-RL Transition Scheduling for Vertical-Domain Few-Shot Distillation8 Oct 2026
- 2026DiffGate: Difficulty-Gated Teacher Guidance for On-Policy Distillation3 Oct 2026
- 2026From Given to Gathered Evidence: Agentic Learning for Longitudinal Medical Reasoning30 Sep 2026
- 2026Improving OCR Faithfulness via Gated and Attenuated On-Policy Distillation29 Sep 2026
- 2026TIDE: Teacher-Student Transition via Informative Distillation and Exploration for Agentic RL28 Sep 2026
- 2026Expected Reasoning-Step Return Unifies On-Policy Learning from Rewards and Teachers26 Sep 2026
- 2026TISD: On-Policy Self-Distillation with Trajectory Intervention25 Sep 2026
- 2026OPDSearch+: On-Policy Distillation with RL Refinement for Search-Augmented Reasoning25 Aug 2026
- 2026Mismatch Matters: On-Policy Distillation Beyond Token Agreement10 Aug 2026
- 2026A Formula-Driven Survey and Research Agenda for On-Policy Distillation22 Jun 2026
1 more
- 2026Mismatch Matters: On-Policy Distillation Beyond Token Agreementcited by 10
- 2026PowerOPD: Stabilizing On-Policy Distillation with Bounded Power Transformationcited by 9
- 2026A Formula-Driven Survey and Research Agenda for On-Policy Distillationcited by 6
- 2026TISD: On-Policy Self-Distillation with Trajectory Interventioncited by 1
- 2026Demystifying On-Policy Distillation: Where Does the Teacher's Advantage Live?cited by 0
- 2026DTDD: Divergence-Triggered Dynamic Distillation for Reliable On-Policy Supervisioncited by 0
- 2026DTDD: Divergence-Triggered Dynamic Distillation for Reliable On-Policy Supervision13 Oct 2026
- 2026Demystifying On-Policy Distillation: Where Does the Teacher's Advantage Live?13 Oct 2026
- 2026TISD: On-Policy Self-Distillation with Trajectory Intervention25 Sep 2026
- 2026Mismatch Matters: On-Policy Distillation Beyond Token Agreement10 Aug 2026
- 2026A Formula-Driven Survey and Research Agenda for On-Policy Distillation22 Jun 2026
- 2026PowerOPD: Stabilizing On-Policy Distillation with Bounded Power Transformation15 Jun 2026
- 2026OPDSearch+: On-Policy Distillation with RL Refinement for Search-Augmented Reasoningcited by 3
- 2026Expected Reasoning-Step Return Unifies On-Policy Learning from Rewards and Teacherscited by 0
- 2026From Given to Gathered Evidence: Agentic Learning for Longitudinal Medical Reasoningcited by 0
- 2026DiffGate: Difficulty-Gated Teacher Guidance for On-Policy Distillationcited by 0
- 2026PIVOT: Perplexity-Informed KD-to-RL Transition Scheduling for Vertical-Domain Few-Shot Distillationcited by 0
- 2026PIVOT: Perplexity-Informed KD-to-RL Transition Scheduling for Vertical-Domain Few-Shot Distillation8 Oct 2026
- 2026DiffGate: Difficulty-Gated Teacher Guidance for On-Policy Distillation3 Oct 2026
- 2026From Given to Gathered Evidence: Agentic Learning for Longitudinal Medical Reasoning30 Sep 2026
- 2026Expected Reasoning-Step Return Unifies On-Policy Learning from Rewards and Teachers26 Sep 2026
- 2026OPDSearch+: On-Policy Distillation with RL Refinement for Search-Augmented Reasoning25 Aug 2026
Similar, not cited
Close in wording, no citation either way
- 2026Beyond Imitation: Filtering On-Policy Distillation by Reasoning Progress
- 2026Bridging Reasoning Trajectories in On-Policy Distillation via Near-Future Guidance
- 2026Reward-Gated On-Policy Distillation
- 2026Transfer-Stratified On-Policy Distillation for RL-Improved Reasoning Teachers
- 2026An RL View of OPD: Least Square Policy Distillation for Sample-Efficient LLM Reasoning
- 2026On-Policy Distillation with Best-of-N Teacher Rollout Selection
- 2026EasyOPD: An Easy-to-use On-Policy Distillation Framework for Large Language Models
- 2026CataOPD: Catalytic On-Policy Distillation for Large Language Model Reasoning