arXiv:2603.13260cs.CLcs.AI2026-03中稿 · ed被引 13

让小模型自己说理,用智能反馈提升推理能力

Explain in Your Own Words: Improving Reasoning via Token-Selective Dual Knowledge Distillation

  • 按重要性选择关键词进行知识蒸馏,减轻小模型负担
  • 结合间接反馈与选择性分布匹配,提升推理准确率54.4%
  • 支持学生自主表达,适合资源有限的推理任务部署

知识蒸馏(KD)可将大模型的推理能力迁移至小模型,降低生成思维链的成本。传统方法要求学生模仿教师整个输出分布,但小模型容量有限,易因过度监督导致分布不匹配,尤其在复杂推理任务中。本文提出一种面向学生的双路选择性知识蒸馏(TSD-KD)框架:通过偏好排序提供弱反馈,教师仅对自生成候选答案重排序,不强制完整分布;直接蒸馏则基于师生置信度差异,选择性地蒸馏关键token;并引入熵正则化保持学生置信度。实验显示,TSD-KD在10个挑战性推理基准上达到最先进性能,准确率超越基线和亚军高达54.4%和40.3%。尤为突出的是,在4个案例中学生模型甚至超过其教师模型,最高提升20.3%。代码已开源。

原文摘要 · Abstract (English)

Knowledge Distillation (KD) can transfer the reasoning abilities of large models to smaller ones, which can reduce the costs to generate Chain-of-Thoughts for reasoning tasks. KD methods typically ask the student to mimic the teacher's distribution over the entire output. However, a student with limited capacity can be overwhelmed by such extensive supervision causing a distribution mismatch, especially in complex reasoning tasks. We propose Token-Selective Dual Knowledge Distillation (TSD-KD), a framework for student-centric distillation. TSD-KD focuses on distilling important tokens for reasoning and encourages the student to explain reasoning in its own words. TSD-KD combines indirect and direct distillation. Indirect distillation uses a weak form of feedback based on preference ranking. The student proposes candidate responses generated on its own; the teacher re-ranks those candidates as indirect feedback without enforcing its entire distribution. Direct distillation uses distribution matching; however, it selectively distills tokens based on the relative confidence between teacher and student. Finally, we add entropy regularization to maintain the student's confidence during distillation. Overall, our method provides the student with targeted and indirect feedback to support its own reasoning process and to facilitate self-improvement. The experiments show the state-of-the-art performance of TSD-KD on 10 challenging reasoning benchmarks, outperforming the baseline and runner-up in accuracy by up to 54.4\% and 40.3\%, respectively. Notably, a student trained by TSD-KD even outperformed its own teacher model in four cases by up to 20.3\%. The source code is available at https://github.com/kmswin1/TSD-KD.

知识蒸馏推理增强小模型优化

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