2015年,Geoffrey Hinton、Oriol Vinyals 和 Jeff Dean等学者<ref>HINTON G, VINYALS O, DEAN J. Distilling the Knowledge in a Neural Network[J/OL]. 2015. DOI:10.48550/arxiv.1503.02531. https://arxiv.org/pdf/1503.02531</ref>提出了在AI领域的知识蒸馏技术,即将大型、复杂模型(教师模型)的知识转移到小型、简单模型(学生模型)的技术,旨在提升小模型的性能并降低计算资源需求。例如,在分类任务中,教师模型进行预测时,除了给出概率最大的类别作为硬标签,也会同时给出每个类别的概率分布作为软标签。学生模型根据教师模型的软标签和数据本身的硬标签进行训练,其损失函数包括学生模型输出与数据本身硬标签之间的交叉熵损失,学生模型输出与教师模型软标签的蒸馏损失。
+
2015年,Geoffrey Hinton、Oriol Vinyals 和 Jeff Dean等学者<ref>HINTON G, VINYALS O, DEAN J. Distilling the Knowledge in a Neural Network[J/OL]. 2015. DOI:10.48550/arxiv.1503.02531. https://arxiv.org/pdf/1503.02531</ref>提出了在AI领域的知识蒸馏技术,即将大型、复杂模型(教师模型)的'''知识'''转移到小型、简单模型(学生模型)的技术。DeepSeek团队发现,不仅仅是知识,推理能力也可以从大模型的推理能力蒸馏递到小模型。
DeepSeek团队用DeepSeek-R1第三阶段整理得到的80万样本对一些开源模型如Qwen2.5和Llama3.3的小模型版本进行了直接微调。它们发现,仅仅是如此简单的微调就能大大提升小模型的推理能力,而且这种幅度的提升是仅凭大规模强化学习所不能得到的。为此,它们使用Qwen-32B-Base做了一个实验,对其分别做了与Deepseek-R1-Zero相同的大规模强化学习<ref> 这里有点微妙。原文是“we conduct large-scale RL training on Qwen-32B-Base using math,code,and STEM data,training for over '''10K steps'''”。很难说1万步的训练到底够不够大规模。</ref>,以及从Deepseek-R1做蒸馏。实验结果是,大规模强化学习后得到的版本,DeepSeek-R1-Zero-Qwen-32B,与QwQ-32B-Preview型能相似,然而从Deepseek-R1蒸馏得到的DeepSeek-R1-Distill-Qwen-32B表现却远超前两者。