mlsanityData to 14 Oct 2026

KDRL: Post-Training Reasoning LLMs via Unified Knowledge Distillation and Reinforcement Learning

arXiv 2506.02208 ↗First posted 2 Jun 2025cs.LG cs.AI cs.CLcited by 41

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

  1. 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?18 Apr
  2. 2025Understanding R1-Zero-Like Training: A Critical Perspective26 Mar
  3. 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scale18 Mar
  4. 2025Cognitive Behaviors that Enable Self-Improving Reasoners, or, Four Habits of Highly Effective STaRs3 Mar
  5. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning22 Jan
  6. 2025Kimi k1.5: Scaling Reinforcement Learning with LLMs22 Jan
  7. 2017Proximal Policy Optimization Algorithms20 Jul

Other references · 12

  1. 2024The Llama 3 Herd of Modelscited by 14,715
  2. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  3. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  4. 2024Qwen2.5 Technical Reportcited by 4,942
  5. 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
  6. 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 2,057
  7. 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,998
  8. 2024Phi-3 Technical Report: A Highly Capable Language Model Locally on Your Phonecited by 1,917
  9. 2024OpenAI o1 System Cardcited by 1,244
  10. 2025SFT Memorizes, RL Generalizes: A Comparative Study of Foundation Model Post-trainingcited by 672
  11. 2023Scaling Relationship on Learning Mathematical Reasoning with Large Language Modelscited by 427
  12. 2024A Survey on Knowledge Distillation of Large Language Modelscited by 284

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 6; 2026: 35.

41 papers.

Sort:

By area:

  1. 2026A Survey of On-Policy Distillation for Large Language Modelscited by 163
  2. 2026Skill-SD: Skill-Conditioned Self-Distillation for Multi-turn LLM Agentscited by 90
  3. 2026Scaling Reasoning Efficiently via Relaxed On-Policy Distillationcited by 82
  4. 2025ExGRPO: Learning to Reason from Experiencecited by 53
  5. 2026SCOPE: Signal-Calibrated On-Policy Distillation Enhancement with Dual-Path Adaptive Weightingcited by 43
  6. 2026HDPO: Hybrid Distillation Policy Optimization via Privileged Self-Distillationcited by 42
  7. 2026Reinforcement-aware Knowledge Distillation for LLM Reasoningcited by 35
  8. 2025VOLD: Reasoning Transfer from LLMs to Vision-Language Models via On-Policy Distillationcited by 31
  9. 2025Quagmires in SFT-RL Post-Training: When High SFT Scores Mislead and What to Use Insteadcited by 28
  10. 2026Teacher-Guided Policy Optimization for On-Policy Reasoning Distillation under Large Policy Divergencecited by 13
  11. 2026Distill Where You Fail: Recovering Learning Signals of Negative RL-Groups from Adaptive Teacher Guidancecited by 11
  12. 2026Multi-Rollout On-Policy Distillation via Peer Successes and Failurescited by 10
29 more
  1. 2026Where Hindsight Credit Can Reside: A Signed-Capacity View of Token Updates in RLVRcited by 8
  2. 2026Distilled Reinforcement Learning for LLM Post-trainingcited by 8
  3. 2025Masking Teacher and Reinforcing Student for Distilling Vision-Language Modelscited by 6
  4. 2026A Formula-Driven Survey and Research Agenda for On-Policy Distillationcited by 6
  5. 2026Tail-Aware Top-k On-Policy Distillationcited by 5
  6. 2026RISE: Recursive Improvement via Self-Extrapolating Policy Distillationcited by 5
  7. 2026HeaPA: Difficulty-Aware Heap Sampling and On-Policy Query Augmentation for LLM Reinforcement Learningcited by 4
  8. 2025Toward Better EHR Reasoning in LLMs: Reinforcement Learning with Expert Attention Guidancecited by 3
  9. 2026Surgical Post-Training: Proximal On-Policy Distillation for Reasoning with Knowledge Retentioncited by 3
  10. 2026CoDistill-GRPO: A Co-Distillation Recipe for Efficient Group Relative Policy Optimizationcited by 3
  11. 2026Zone of Proximal Policy Optimization: Teacher in Prompts, Not Gradientscited by 3
  12. 2026OPDSearch+: On-Policy Distillation with RL Refinement for Search-Augmented Reasoningcited by 3
  13. 2026Offline Exploration-Aware Fine-Tuning for Long-Chain Mathematical Reasoningcited by 2
  14. 2026Structured Role-Aware Policy Optimization for Multimodal Reasoningcited by 2
  15. 2026One-Way Policy Optimization for Self-Evolving LLMscited by 2
  16. 2026Eliciting Weak-to-Strong Generalization with On-Policy Reverse Distillationcited by 2
  17. 2025Learning Reasoning Rewards from Expert Demonstrations with Inverse Reinforcement Learningcited by 1
  18. 2026A First-Principles Derivation of LLM Policy Optimization: From Expected Reward to GRPO and Its Structural Extensionscited by 1
  19. 2026Signal Reshaping for GRPO in Weak-Feedback Agentic Code Repaircited by 0
  20. 2026AIPO: Learning to Reason from Active Interactioncited by 0
  21. 2026PathRouter: Aligning Rewards with Retrieval Quality in Agentic Graph Retrieval-Augmented Generationcited by 0
  22. 2026When and Where to Trust the Teacher: Unifying On-Policy Distillation and GRPO through Entropy-Calibrated Credit Assignmentcited by 0
  23. 2026UOPD: Uncertainty-Aware Intervention for On-Policy Distillation of Multi-Turn Agentscited by 0
  24. 2026Improving OCR Faithfulness via Gated and Attenuated On-Policy Distillationcited by 0
  25. 2026DiffGate: Difficulty-Gated Teacher Guidance for On-Policy Distillationcited by 0
  26. 2026PIVOT: Perplexity-Informed KD-to-RL Transition Scheduling for Vertical-Domain Few-Shot Distillationcited by 0
  27. 2026CAST: Credit Alignment for Cross-Tokenizer On-Policy Distillationcited by 0
  28. 2026Reward-Aligned Distillationcited by 0
  29. 2026Teacher-Guided Tool Learning for Small Language Models via Bidirectional KL Constraintscited by 0

Similar, not cited

Close in wording, no citation either way

  1. 2025Beyond Scaling Law: A Data-Efficient Distillation Framework for Reasoning
  2. 2026Reinforcement Learning-based Semi-supervised Knowledge Distillation with LLM-as-a-Judge
  3. 2026KEPO: Knowledge-Enhanced Preference Optimization for Multimodal Reasoning with Applications to Medical VQA
  4. 2025Revisiting Reinforcement Learning for LLM Reasoning from A Cross-Domain Perspective
  5. 2026Decoupling KL and Trajectories: A Unified Perspective for SFT, DAgger, Offline RL, and OPD in LLM Distillation
  6. 2025Reshaping Reasoning in LLMs: A Theoretical Analysis of RL Training Dynamics through Pattern Selection
  7. 2025Scaling Up RL: Unlocking Diverse Reasoning in LLMs via Prolonged Training
  8. 2025Accelerating RL for LLM Reasoning with Optimal Advantage Regression