mlsanityData to 14 Oct 2026

RLKD: Distilling LLMs' Reasoning via Reinforcement Learning

arXiv 2505.16142 ↗First posted 22 May 2025cs.CLcited by 11

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

  1. 2025LLMs Can Easily Learn to Reason from Demonstrations Structure, not content, is what matters!11 Feb
  2. 2024SuperCorrect: Advancing Small LLM Reasoning with Thought Template Distillation and Self-Correction11 Oct
  3. 2024Beyond Imitation: Learning Key Reasoning Steps from Dual Chain-of-Thoughts in Reasoning Distillation30 May
  4. 2023Graph of Thoughts: Solving Elaborate Problems with Large Language Models18 Aug
  5. 2022Least-to-Most Prompting Enables Complex Reasoning in Large Language Models21 May
  6. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Models28 Jan

Other references · 26

  1. 2022Training language models to follow instructions with human feedbackcited by 15,971
  2. 2017Proximal Policy Optimization Algorithmscited by 15,215
  3. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  4. 2021Evaluating Large Language Models Trained on Codecited by 8,320
  5. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
  6. 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,211
  7. 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,879
  8. 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
  9. 2016Concrete Problems in AI Safetycited by 1,642
  10. 2024Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvementcited by 1,090
  11. 2022Measuring and Narrowing the Compositionality Gap in Language Modelscited by 1,075
  12. 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 982
14 more
  1. 2025SFT Memorizes, RL Generalizes: A Comparative Study of Foundation Model Post-trainingcited by 672
  2. 2025Towards Large Reasoning Models: A Survey of Reinforced Reasoning with Large Language Modelscited by 293
  3. 2025SFT or RL? An Early Investigation into Training R1-Like Reasoning Large Vision-Language Modelscited by 212
  4. 2025Light-R1: Curriculum SFT, DPO and RL for Long COT from Scratch and Beyondcited by 175
  5. 2024Teaching Large Language Models to Reason with Reinforcement Learningcited by 160
  6. 2024Generative Reward Modelscited by 148
  7. 2021Goal Misgeneralization in Deep Reinforcement Learningcited by 123
  8. 2023Search-in-the-Chain: Interactively Enhancing Large Language Models with Search for Knowledge-intensive Taskscited by 73
  9. 2025100 Days After DeepSeek-R1: A Survey on Replication Studies and More Directions for Reasoning Language Modelscited by 30
  10. 2024Unsupervised Information Refinement Training of Large Language Models for Retrieval-Augmented Generationcited by 19
  11. 2024Everything is Editable: Extend Knowledge Editing to Unstructured Data in Large Language Modelscited by 19
  12. 2025A Survey of Slow Thinking-based Reasoning LLMs using Reinforced Learning and Inference-time Scaling Lawcited by 19
  13. 2025Following the Autoregressive Nature of LLM Embeddings via Compression and Alignmentcited by 14
  14. 2024A Theory for Token-Level Harmonization in Retrieval-Augmented Generationcited by 7

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 2; 2026: 9.

11 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2025Harnessing Negative Signals: Reinforcement Distillation from Teacher Data for LLM Reasoning
  2. 2026Reinforcement-aware Knowledge Distillation for LLM Reasoning
  3. 2025Beyond Scaling Law: A Data-Efficient Distillation Framework for Reasoning
  4. 2025Revisiting Reinforcement Learning for LLM Reasoning from A Cross-Domain Perspective
  5. 2025RL Squeezes, SFT Expands: A Comparative Study of Reasoning LLMs
  6. 2025Reshaping Reasoning in LLMs: A Theoretical Analysis of RL Training Dynamics through Pattern Selection
  7. 2026Reward-Gated On-Policy Distillation
  8. 2025Beyond Two-Stage Training: Cooperative SFT and RL for LLM Reasoning