RLKD: Distilling LLMs' Reasoning via Reinforcement Learning
On the map: LLM reasoning → Prompting and chain-of-thought reasoning
also in 1 more
Abstract (arXiv)
Distilling reasoning paths from teacher to student models via supervised fine-tuning (SFT) provides a shortcut for improving the reasoning ability of smaller Large Language Models (LLMs). However, the reasoning paths generated by teacher models often reflect only surface-level traces of their underlying authentic reasoning. Insights from cognitive neuroscience suggest that authentic reasoning involves a complex interweaving between meta-reasoning (which selects appropriate sub-problems from multiple candidates) and solving (which addresses the sub-problem). This implies authentic reasoning has an implicit multi-branch structure. Supervised fine-tuning collapses this rich structure into a flat sequence of token prediction in the teacher's reasoning path, preventing effective distillation of this structure to students. To address this limitation, we propose RLKD, a reinforcement learning (RL)-based distillation framework guided by a novel Generative Structure Reward Model (GSRM). Our GSRM converts reasoning paths into multiple meta-reasoning-solving steps and computes rewards to measure structural alignment between student and teacher reasoning. RLKD combines this reward with RL, enabling student LLMs to internalize the teacher's implicit multi-branch reasoning structure rather than merely mimicking fixed output paths. Experiments show RLKD surpasses standard SFT-RL pipelines even when trained on 0.1% of data under an RL-only regime, unlocking greater student reasoning potential than SFT-based distillation. Code is available at https://github.com/xsc1234/RLKD.
Key facts. Cited by 11 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 32 of its 41 references that are in the map. First posted on arXiv 22 May 2025.
Builds on
32 of 41 references.
Closest ancestors · 6
Same direction, newest first
- 2025LLMs Can Easily Learn to Reason from Demonstrations Structure, not content, is what matters!11 Feb
- 2024SuperCorrect: Advancing Small LLM Reasoning with Thought Template Distillation and Self-Correction11 Oct
- 2024Beyond Imitation: Learning Key Reasoning Steps from Dual Chain-of-Thoughts in Reasoning Distillation30 May
- 2023Graph of Thoughts: Solving Elaborate Problems with Large Language Models18 Aug
- 2022Least-to-Most Prompting Enables Complex Reasoning in Large Language Models21 May
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Models28 Jan
Other references · 26
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2021Evaluating Large Language Models Trained on Codecited by 8,320
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,211
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,879
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
- 2016Concrete Problems in AI Safetycited by 1,642
- 2024Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvementcited by 1,090
- 2022Measuring and Narrowing the Compositionality Gap in Language Modelscited by 1,075
- 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 982
14 more
- 2025SFT Memorizes, RL Generalizes: A Comparative Study of Foundation Model Post-trainingcited by 672
- 2025Towards Large Reasoning Models: A Survey of Reinforced Reasoning with Large Language Modelscited by 293
- 2025SFT or RL? An Early Investigation into Training R1-Like Reasoning Large Vision-Language Modelscited by 212
- 2025Light-R1: Curriculum SFT, DPO and RL for Long COT from Scratch and Beyondcited by 175
- 2024Teaching Large Language Models to Reason with Reinforcement Learningcited by 160
- 2024Generative Reward Modelscited by 148
- 2021Goal Misgeneralization in Deep Reinforcement Learningcited by 123
- 2023Search-in-the-Chain: Interactively Enhancing Large Language Models with Search for Knowledge-intensive Taskscited by 73
- 2025100 Days After DeepSeek-R1: A Survey on Replication Studies and More Directions for Reasoning Language Modelscited by 30
- 2024Unsupervised Information Refinement Training of Large Language Models for Retrieval-Augmented Generationcited by 19
- 2024Everything is Editable: Extend Knowledge Editing to Unstructured Data in Large Language Modelscited by 19
- 2025A Survey of Slow Thinking-based Reasoning LLMs using Reinforced Learning and Inference-time Scaling Lawcited by 19
- 2025Following the Autoregressive Nature of LLM Embeddings via Compression and Alignmentcited by 14
- 2024A Theory for Token-Level Harmonization in Retrieval-Augmented Generationcited by 7
Built on by
Citing papers per year: 2025: 2; 2026: 9.
11 papers.
- 2026A Survey of On-Policy Distillation for Large Language Modelscited by 163
- 2025LLM Latent Reasoning as Chain of Superpositioncited by 39
- 2026SOD: Step-wise On-policy Distillation for Small Language Model Agentscited by 21
- 2026Multi-Rollout On-Policy Distillation via Peer Successes and Failurescited by 10
- 2025Bridging Reasoning to Learning: Unmasking Illusions using Complexity Out of Distribution Generalizationcited by 1
- 2026Structural Rationale Distillation via Reasoning Space Compressioncited by 0
- 2026GateKD: Confidence-Gated Closed-Loop Distillation for Robust Reasoningcited by 0
- 2026ARKD: Adaptive Reinforcement Learning-Guided Bidirectional KL Divergence Distillation for Text Generationcited by 0
- 2026AdaStop: Cost-Aware Early Stopping for DNN Test Selectioncited by 0
- 2026STAR-OPD: Structured Aspect-Cascade-Aware On-Policy Reward Distillation for ABSA Quadruple Extractioncited by 0
- 2026Look Before You Select: Rethinking Vocabulary Sparsification in On-Policy Distillationcited by 0
- 2026Look Before You Select: Rethinking Vocabulary Sparsification in On-Policy Distillation28 Sep 2026
- 2026STAR-OPD: Structured Aspect-Cascade-Aware On-Policy Reward Distillation for ABSA Quadruple Extraction21 Aug 2026
- 2026AdaStop: Cost-Aware Early Stopping for DNN Test Selection6 Jul 2026
- 2026ARKD: Adaptive Reinforcement Learning-Guided Bidirectional KL Divergence Distillation for Text Generation29 Jun 2026
- 2026GateKD: Confidence-Gated Closed-Loop Distillation for Robust Reasoning13 May 2026
- 2026Multi-Rollout On-Policy Distillation via Peer Successes and Failures12 May 2026
- 2026SOD: Step-wise On-policy Distillation for Small Language Model Agents8 May 2026
- 2026Structural Rationale Distillation via Reasoning Space Compression8 May 2026
- 2026A Survey of On-Policy Distillation for Large Language Models1 Apr 2026
- 2025LLM Latent Reasoning as Chain of Superposition17 Oct 2025
- 2025Bridging Reasoning to Learning: Unmasking Illusions using Complexity Out of Distribution Generalization6 Oct 2025
Similar, not cited
Close in wording, no citation either way
- 2025Harnessing Negative Signals: Reinforcement Distillation from Teacher Data for LLM Reasoning
- 2026Reinforcement-aware Knowledge Distillation for LLM Reasoning
- 2025Beyond Scaling Law: A Data-Efficient Distillation Framework for Reasoning
- 2025Revisiting Reinforcement Learning for LLM Reasoning from A Cross-Domain Perspective
- 2025RL Squeezes, SFT Expands: A Comparative Study of Reasoning LLMs
- 2025Reshaping Reasoning in LLMs: A Theoretical Analysis of RL Training Dynamics through Pattern Selection
- 2026Reward-Gated On-Policy Distillation
- 2025Beyond Two-Stage Training: Cooperative SFT and RL for LLM Reasoning