arXiv:2601.18734cs.LGcs.CL2026-01被引 259

让大模型自己教自己推理,提升效率与准确率

Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Models

  • 单个模型分角色扮演教师与学生,教师看完整推理路径
  • 训练时使学生输出与教师在相同轨迹上的分布一致
  • 无需额外教师模型,适合数学推理任务优化

知识蒸馏通过将教师大模型的知识压缩到较小模型中来提升大语言模型的推理能力。在策略蒸馏中,学生模型在自我采样轨迹的同时,由教师模型提供密集的逐标记监督,解决了离策略方法中训练与推理分布不匹配的问题。然而,现有方法通常需要一个独立且更大的教师模型,并未显式利用推理数据集中存在的真实答案。受启发于‘足够强大的大模型可解释外部权威推理过程并指导自身’这一直觉,我们提出一种名为自策蒸馏(OPSD)的学习算法:单个大模型在不同上下文中同时担任教师与学生角色。教师策略依赖于特权信息(如验证过的推理路径),而学生策略仅能看到问题本身;训练目标是最小化学生自身轨迹上两者的逐标记分布差异。我们在多个数学推理基准上验证了该方法的有效性,相比强化学习方法具备更优的标记效率,且性能优于离策略蒸馏方法。

原文摘要 · Abstract (English)

Knowledge distillation improves large language model (LLM) reasoning by compressing the knowledge of a teacher LLM to train smaller LLMs. On-policy distillation advances this approach by having the student sample its own trajectories while a teacher LLM provides dense token-level supervision, addressing the distribution mismatch between training and inference in off-policy distillation methods. However, on-policy distillation typically requires a separate, often larger, teacher LLM and does not explicitly leverage ground-truth solutions available in reasoning datasets. Inspired by the intuition that a sufficiently capable LLM can rationalize external privileged reasoning traces and teach its weaker self, we introduce On-Policy Self-Distillation (OPSD), a learning algorithm where a single LLM acts as both teacher and student with different contexts. The teacher policy conditions on privileged information (e.g., verified reasoning traces) while the student policy sees only the question; training minimizes the per-token divergence between these distributions over the student's own rollouts. We demonstrate the efficacy of our method on multiple mathematical reasoning benchmarks, achieving superior token efficiency compared to reinforcement learning methods and better performance over off-policy distillation methods. Code repo: https://github.com/siyan-zhao/OPSD.

大模型推理知识蒸馏自训练

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。