用强化学习让小模型学会大模型的多分支推理结构。
RLKD: Distilling LLMs' Reasoning via Reinforcement Learning
- 通过强化学习捕捉大模型隐含的多分支推理路径。
- 仅用0.1%数据训练,效果超越传统监督微调方法。
- 适合想提升小模型深层推理能力的研究者。
通过监督微调(SFT)将教师模型的推理路径蒸馏到学生模型,可快速提升小型大语言模型的推理能力。然而,教师模型生成的推理路径往往只反映其底层真实推理的表层痕迹。认知神经科学表明,真实推理包含元推理(从多个候选子问题中选择合适问题)与求解之间的复杂交织,具有隐含的多分支结构。而监督微调将这一丰富结构压缩为教师推理路径中扁平的词元序列,阻碍了该结构的有效蒸馏。为此,我们提出基于强化学习的蒸馏框架RLKD,引入一种新型生成结构奖励模型(GSRM),将推理路径转化为多个元推理-求解步骤,并计算奖励以衡量学生与教师推理结构的一致性。RLKD结合该奖励与强化学习,使学生模型能内化教师的隐含多分支推理结构,而非仅模仿固定输出路径。实验表明,即使在仅使用0.1%数据的强化学习独有训练设置下,RLKD仍优于标准的SFT-RL流水线,解锁了远超基于SFT蒸馏的学生推理潜力。代码已公开于https://github.com/xsc1234/RLKD。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。