arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2509.08814cs.LGcs.AIcs.CL

思维合并蒸馏

Merge-of-Thought Distillation

Zhejiang University(浙江大学);Inclusion AI, Ant Group(蚂蚁集团 inclusion AI)

Zhanming Shen, Zeyu Qin, Zenan Huang, Hao Chen, Jiaqi Hu, Yihong Zhuang, Guoshan Lu, Gang Chen, Junbo Zhao

AI总结:

提出思维合并蒸馏(MoT),通过交替进行教师特定微调与权重合并,将多教师推理能力统一至学生模型,在竞赛数学上超越强模型并提升通用推理。

摘要:

高效的长思维链(CoT)模型推理蒸馏日益受到单一“教师”模型假设的约束,尽管在实际中往往存在多个候选教师模型以及不断增长的CoT语料库。我们重新审视了教师选择问题,并观察到不同的学生模型有不同的“最佳教师”,即使对于同一个学生模型,其最佳教师也可能因数据集而异。因此,为了将多个教师的推理能力统一到一个学生模型中,以克服不同教师监督之间的冲突,我们提出了思维合并蒸馏(Merge-of-Thought Distillation, MoT),这是一种轻量级框架,它在针对特定教师的监督微调分支与对所得学生变体进行权重空间合并之间交替进行。在竞赛数学基准上,仅使用约200个CoT样本,将MoT应用于Qwen3-14B学生模型,其性能超越了包括Deepseek-R1、Qwen3-32B和OpenAI-O1在内的强模型,展示了显著的提升。此外,MoT始终优于最佳单教师蒸馏,在数学之外还提升了通用推理能力,同时减少了灾难性遗忘,并且对分布偏移和同级教师表现出鲁棒性。最后,我们证明了MoT通过消除特定教师的归纳偏差和教师间冲突,同时反复强化共识推理特征的学习,从而具备了共识CoT。这些结果使MoT成为从多样化教师高效蒸馏长CoT能力到紧凑学生模型的简单有效途径。

英文摘要:

Efficient reasoning distillation for long chain-of-thought (CoT) models is increasingly constrained by the assumption of a single oracle teacher, despite the practical availability of multiple candidate teachers and growing CoT corpora. We revisit teacher selection and observe that different students have different "best teachers," and even for the same student, the best teacher can vary across datasets. Therefore, to unify multiple teachers' reasoning abilities into a student to overcome conflicts among various teachers' supervision, we propose Merge-of-Thought Distillation (MoT), a lightweight framework that alternates between teacher-specific supervised fine-tuning branches and weight-space merging of the resulting student variants. On competition math benchmarks, using only about 200 CoT samples, applying MoT to a Qwen3-14B student surpasses strong models including Deepseek-R1, Qwen3-32B, and OpenAI-O1, demonstrating substantial gains. Besides, MoT consistently outperforms the best single-teacher distillation, improves general reasoning beyond mathematics while reducing catastrophic forgetting, and shows robustness to distribution-shifted and peer-level teachers. Finally, we have demonstrated MoT possesses consensus CoT by eliminating teacher-specific inductive biases and inter-teacher conflicts while repeatedly reinforcing the learning of consensus reasoning features. These results position MoT as a simple, effective route to efficiently distilling long CoT capabilities from diverse teachers into compact students.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑