KDRL: Post-Training Reasoning LLMs via Unified Knowledge Distillation and Reinforcement Learning
On the map: LLM reasoning → Reinforcement learning for reasoning
also in 1 more
Abstract (arXiv)
Recent advances in large language model (LLM) post-training have leveraged two distinct paradigms to enhance reasoning capabilities: reinforcement learning (RL) and knowledge distillation (KD). While RL enables the emergence of complex reasoning behaviors, it often suffers from low sample efficiency when the initial policy struggles to explore high-reward trajectories. Conversely, KD improves learning efficiency via mimicking the teacher model but tends to generalize poorly to out-of-domain scenarios. In this work, we present KDRL, a textitunified post-training framework that jointly optimizes a reasoning model through teacher supervision (KD) and self-exploration (RL). Specifically, KDRL leverages policy gradient optimization to simultaneously minimize the reverse Kullback-Leibler divergence (RKL) between the student and teacher distributions while maximizing the expected rule-based rewards. We first formulate a unified objective that integrates GRPO and KD, and systematically explore how different KL approximations, KL coefficients, and reward-guided KD strategies affect the overall post-training dynamics and performance. Empirical results on multiple reasoning benchmarks demonstrate that KDRL outperforms GRPO and various KD baselines while achieving a favorable balance between performance and reasoning token efficiency. These findings indicate that integrating KD and RL serves as an effective and efficient strategy to train reasoning LLMs.
Key facts. Cited by 41 papers in the map (data through 14 Oct 2026). Placed in Reinforcement learning for reasoning, one of 17 directions in LLM reasoning. Builds on 19 of its 56 references that are in the map. First posted on arXiv 2 Jun 2025.
Builds on
19 of 56 references.
Closest ancestors · 7
Same direction, newest first
- 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?18 Apr
- 2025Understanding R1-Zero-Like Training: A Critical Perspective26 Mar
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scale18 Mar
- 2025Cognitive Behaviors that Enable Self-Improving Reasoners, or, Four Habits of Highly Effective STaRs3 Mar
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning22 Jan
- 2025Kimi k1.5: Scaling Reinforcement Learning with LLMs22 Jan
- 2017Proximal Policy Optimization Algorithms20 Jul
Other references · 12
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2024Qwen2.5 Technical Reportcited by 4,942
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 2,057
- 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,998
- 2024Phi-3 Technical Report: A Highly Capable Language Model Locally on Your Phonecited by 1,917
- 2024OpenAI o1 System Cardcited by 1,244
- 2025SFT Memorizes, RL Generalizes: A Comparative Study of Foundation Model Post-trainingcited by 672
- 2023Scaling Relationship on Learning Mathematical Reasoning with Large Language Modelscited by 427
- 2024A Survey on Knowledge Distillation of Large Language Modelscited by 284
Built on by
Citing papers per year: 2025: 6; 2026: 35.
41 papers.
Sort:
By area:
- 2026A Survey of On-Policy Distillation for Large Language Modelscited by 163
- 2026Skill-SD: Skill-Conditioned Self-Distillation for Multi-turn LLM Agentscited by 90
- 2026Scaling Reasoning Efficiently via Relaxed On-Policy Distillationcited by 82
- 2025ExGRPO: Learning to Reason from Experiencecited by 53
- 2026SCOPE: Signal-Calibrated On-Policy Distillation Enhancement with Dual-Path Adaptive Weightingcited by 43
- 2026HDPO: Hybrid Distillation Policy Optimization via Privileged Self-Distillationcited by 42
- 2026Reinforcement-aware Knowledge Distillation for LLM Reasoningcited by 35
- 2025VOLD: Reasoning Transfer from LLMs to Vision-Language Models via On-Policy Distillationcited by 31
- 2025Quagmires in SFT-RL Post-Training: When High SFT Scores Mislead and What to Use Insteadcited by 28
- 2026Teacher-Guided Policy Optimization for On-Policy Reasoning Distillation under Large Policy Divergencecited by 13
- 2026Distill Where You Fail: Recovering Learning Signals of Negative RL-Groups from Adaptive Teacher Guidancecited by 11
- 2026Multi-Rollout On-Policy Distillation via Peer Successes and Failurescited by 10
29 more
- 2026Where Hindsight Credit Can Reside: A Signed-Capacity View of Token Updates in RLVRcited by 8
- 2026Distilled Reinforcement Learning for LLM Post-trainingcited by 8
- 2025Masking Teacher and Reinforcing Student for Distilling Vision-Language Modelscited by 6
- 2026A Formula-Driven Survey and Research Agenda for On-Policy Distillationcited by 6
- 2026Tail-Aware Top-k On-Policy Distillationcited by 5
- 2026RISE: Recursive Improvement via Self-Extrapolating Policy Distillationcited by 5
- 2026HeaPA: Difficulty-Aware Heap Sampling and On-Policy Query Augmentation for LLM Reinforcement Learningcited by 4
- 2025Toward Better EHR Reasoning in LLMs: Reinforcement Learning with Expert Attention Guidancecited by 3
- 2026Surgical Post-Training: Proximal On-Policy Distillation for Reasoning with Knowledge Retentioncited by 3
- 2026CoDistill-GRPO: A Co-Distillation Recipe for Efficient Group Relative Policy Optimizationcited by 3
- 2026Zone of Proximal Policy Optimization: Teacher in Prompts, Not Gradientscited by 3
- 2026OPDSearch+: On-Policy Distillation with RL Refinement for Search-Augmented Reasoningcited by 3
- 2026Offline Exploration-Aware Fine-Tuning for Long-Chain Mathematical Reasoningcited by 2
- 2026Structured Role-Aware Policy Optimization for Multimodal Reasoningcited by 2
- 2026One-Way Policy Optimization for Self-Evolving LLMscited by 2
- 2026Eliciting Weak-to-Strong Generalization with On-Policy Reverse Distillationcited by 2
- 2025Learning Reasoning Rewards from Expert Demonstrations with Inverse Reinforcement Learningcited by 1
- 2026A First-Principles Derivation of LLM Policy Optimization: From Expected Reward to GRPO and Its Structural Extensionscited by 1
- 2026Signal Reshaping for GRPO in Weak-Feedback Agentic Code Repaircited by 0
- 2026AIPO: Learning to Reason from Active Interactioncited by 0
- 2026PathRouter: Aligning Rewards with Retrieval Quality in Agentic Graph Retrieval-Augmented Generationcited by 0
- 2026When and Where to Trust the Teacher: Unifying On-Policy Distillation and GRPO through Entropy-Calibrated Credit Assignmentcited by 0
- 2026UOPD: Uncertainty-Aware Intervention for On-Policy Distillation of Multi-Turn Agentscited by 0
- 2026Improving OCR Faithfulness via Gated and Attenuated On-Policy Distillationcited 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
- 2026CAST: Credit Alignment for Cross-Tokenizer On-Policy Distillationcited by 0
- 2026Reward-Aligned Distillationcited by 0
- 2026Teacher-Guided Tool Learning for Small Language Models via Bidirectional KL Constraintscited by 0
- 2026Teacher-Guided Tool Learning for Small Language Models via Bidirectional KL Constraints13 Oct 2026
- 2026Reward-Aligned Distillation10 Oct 2026
- 2026CAST: Credit Alignment for Cross-Tokenizer On-Policy Distillation9 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
- 2026Improving OCR Faithfulness via Gated and Attenuated On-Policy Distillation29 Sep 2026
- 2026UOPD: Uncertainty-Aware Intervention for On-Policy Distillation of Multi-Turn Agents27 Sep 2026
- 2026When and Where to Trust the Teacher: Unifying On-Policy Distillation and GRPO through Entropy-Calibrated Credit Assignment23 Sep 2026
- 2026Eliciting Weak-to-Strong Generalization with On-Policy Reverse Distillation8 Sep 2026
- 2026RISE: Recursive Improvement via Self-Extrapolating Policy Distillation4 Sep 2026
- 2026OPDSearch+: On-Policy Distillation with RL Refinement for Search-Augmented Reasoning25 Aug 2026
- 2026Tail-Aware Top-k On-Policy Distillation12 Aug 2026
29 more
- 2026Distill Where You Fail: Recovering Learning Signals of Negative RL-Groups from Adaptive Teacher Guidance1 Aug 2026
- 2026Distilled Reinforcement Learning for LLM Post-training19 Jul 2026
- 2026A Formula-Driven Survey and Research Agenda for On-Policy Distillation22 Jun 2026
- 2026Zone of Proximal Policy Optimization: Teacher in Prompts, Not Gradients16 Jun 2026
- 2026A First-Principles Derivation of LLM Policy Optimization: From Expected Reward to GRPO and Its Structural Extensions15 Jun 2026
- 2026PathRouter: Aligning Rewards with Retrieval Quality in Agentic Graph Retrieval-Augmented Generation15 Jun 2026
- 2026One-Way Policy Optimization for Self-Evolving LLMs21 May 2026
- 2026Teacher-Guided Policy Optimization for On-Policy Reasoning Distillation under Large Policy Divergence13 May 2026
- 2026Multi-Rollout On-Policy Distillation via Peer Successes and Failures12 May 2026
- 2026CoDistill-GRPO: A Co-Distillation Recipe for Efficient Group Relative Policy Optimization9 May 2026
- 2026AIPO: Learning to Reason from Active Interaction8 May 2026
- 2026Signal Reshaping for GRPO in Weak-Feedback Agentic Code Repair8 May 2026
- 2026Structured Role-Aware Policy Optimization for Multimodal Reasoning8 May 2026
- 2026Where Hindsight Credit Can Reside: A Signed-Capacity View of Token Updates in RLVR13 Apr 2026
- 2026SCOPE: Signal-Calibrated On-Policy Distillation Enhancement with Dual-Path Adaptive Weighting12 Apr 2026
- 2026Skill-SD: Skill-Conditioned Self-Distillation for Multi-turn LLM Agents12 Apr 2026
- 2026A Survey of On-Policy Distillation for Large Language Models1 Apr 2026
- 2026HDPO: Hybrid Distillation Policy Optimization via Privileged Self-Distillation25 Mar 2026
- 2026Offline Exploration-Aware Fine-Tuning for Long-Chain Mathematical Reasoning17 Mar 2026
- 2026Scaling Reasoning Efficiently via Relaxed On-Policy Distillation11 Mar 2026
- 2026Surgical Post-Training: Proximal On-Policy Distillation for Reasoning with Knowledge Retention2 Mar 2026
- 2026Reinforcement-aware Knowledge Distillation for LLM Reasoning26 Feb 2026
- 2026HeaPA: Difficulty-Aware Heap Sampling and On-Policy Query Augmentation for LLM Reinforcement Learning30 Jan 2026
- 2025Masking Teacher and Reinforcing Student for Distilling Vision-Language Models23 Dec 2025
- 2025VOLD: Reasoning Transfer from LLMs to Vision-Language Models via On-Policy Distillation27 Oct 2025
- 2025ExGRPO: Learning to Reason from Experience2 Oct 2025
- 2025Learning Reasoning Rewards from Expert Demonstrations with Inverse Reinforcement Learning2 Oct 2025
- 2025Quagmires in SFT-RL Post-Training: When High SFT Scores Mislead and What to Use Instead2 Oct 2025
- 2025Toward Better EHR Reasoning in LLMs: Reinforcement Learning with Expert Attention Guidance19 Aug 2025
- 2026Scaling Reasoning Efficiently via Relaxed On-Policy Distillationcited by 82
- 2025ExGRPO: Learning to Reason from Experiencecited by 53
- 2026SCOPE: Signal-Calibrated On-Policy Distillation Enhancement with Dual-Path Adaptive Weightingcited by 43
- 2026HDPO: Hybrid Distillation Policy Optimization via Privileged Self-Distillationcited by 42
- 2026Reinforcement-aware Knowledge Distillation for LLM Reasoningcited by 35
- 2025VOLD: Reasoning Transfer from LLMs to Vision-Language Models via On-Policy Distillationcited by 31
- 2025Quagmires in SFT-RL Post-Training: When High SFT Scores Mislead and What to Use Insteadcited by 28
- 2026Teacher-Guided Policy Optimization for On-Policy Reasoning Distillation under Large Policy Divergencecited by 13
- 2026Where Hindsight Credit Can Reside: A Signed-Capacity View of Token Updates in RLVRcited by 8
- 2026HeaPA: Difficulty-Aware Heap Sampling and On-Policy Query Augmentation for LLM Reinforcement Learningcited by 4
- 2025Toward Better EHR Reasoning in LLMs: Reinforcement Learning with Expert Attention Guidancecited by 3
- 2026Surgical Post-Training: Proximal On-Policy Distillation for Reasoning with Knowledge Retentioncited by 3
12 most cited of 25 All 25 (JSON)
- 2026Reward-Aligned Distillation10 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
- 2026When and Where to Trust the Teacher: Unifying On-Policy Distillation and GRPO through Entropy-Calibrated Credit Assignment23 Sep 2026
- 2026OPDSearch+: On-Policy Distillation with RL Refinement for Search-Augmented Reasoning25 Aug 2026
- 2026A First-Principles Derivation of LLM Policy Optimization: From Expected Reward to GRPO and Its Structural Extensions15 Jun 2026
- 2026One-Way Policy Optimization for Self-Evolving LLMs21 May 2026
- 2026Teacher-Guided Policy Optimization for On-Policy Reasoning Distillation under Large Policy Divergence13 May 2026
- 2026CoDistill-GRPO: A Co-Distillation Recipe for Efficient Group Relative Policy Optimization9 May 2026
- 2026AIPO: Learning to Reason from Active Interaction8 May 2026
- 2026Signal Reshaping for GRPO in Weak-Feedback Agentic Code Repair8 May 2026
- 2026Structured Role-Aware Policy Optimization for Multimodal Reasoning8 May 2026
12 newest of 25 All 25 (JSON)
- 2026A Survey of On-Policy Distillation for Large Language Modelscited by 163
- 2026Distill Where You Fail: Recovering Learning Signals of Negative RL-Groups from Adaptive Teacher Guidancecited by 11
- 2026Multi-Rollout On-Policy Distillation via Peer Successes and Failurescited by 10
- 2026Distilled Reinforcement Learning for LLM Post-trainingcited by 8
- 2026A Formula-Driven Survey and Research Agenda for On-Policy Distillationcited by 6
- 2026Tail-Aware Top-k On-Policy Distillationcited by 5
- 2026RISE: Recursive Improvement via Self-Extrapolating Policy Distillationcited by 5
- 2026Zone of Proximal Policy Optimization: Teacher in Prompts, Not Gradientscited by 3
- 2026Eliciting Weak-to-Strong Generalization with On-Policy Reverse Distillationcited by 2
- 2026UOPD: Uncertainty-Aware Intervention for On-Policy Distillation of Multi-Turn Agentscited by 0
- 2026CAST: Credit Alignment for Cross-Tokenizer On-Policy Distillationcited by 0
- 2026CAST: Credit Alignment for Cross-Tokenizer On-Policy Distillation9 Oct 2026
- 2026UOPD: Uncertainty-Aware Intervention for On-Policy Distillation of Multi-Turn Agents27 Sep 2026
- 2026Eliciting Weak-to-Strong Generalization with On-Policy Reverse Distillation8 Sep 2026
- 2026RISE: Recursive Improvement via Self-Extrapolating Policy Distillation4 Sep 2026
- 2026Tail-Aware Top-k On-Policy Distillation12 Aug 2026
- 2026Distill Where You Fail: Recovering Learning Signals of Negative RL-Groups from Adaptive Teacher Guidance1 Aug 2026
- 2026Distilled Reinforcement Learning for LLM Post-training19 Jul 2026
- 2026A Formula-Driven Survey and Research Agenda for On-Policy Distillation22 Jun 2026
- 2026Zone of Proximal Policy Optimization: Teacher in Prompts, Not Gradients16 Jun 2026
- 2026Multi-Rollout On-Policy Distillation via Peer Successes and Failures12 May 2026
- 2026A Survey of On-Policy Distillation for Large Language Models1 Apr 2026
Similar, not cited
Close in wording, no citation either way
- 2025Beyond Scaling Law: A Data-Efficient Distillation Framework for Reasoning
- 2026Reinforcement Learning-based Semi-supervised Knowledge Distillation with LLM-as-a-Judge
- 2026KEPO: Knowledge-Enhanced Preference Optimization for Multimodal Reasoning with Applications to Medical VQA
- 2025Revisiting Reinforcement Learning for LLM Reasoning from A Cross-Domain Perspective
- 2026Decoupling KL and Trajectories: A Unified Perspective for SFT, DAgger, Offline RL, and OPD in LLM Distillation
- 2025Reshaping Reasoning in LLMs: A Theoretical Analysis of RL Training Dynamics through Pattern Selection
- 2025Scaling Up RL: Unlocking Diverse Reasoning in LLMs via Prolonged Training
- 2025Accelerating RL for LLM Reasoning with Optimal Advantage Regression