Knowledge Distillation(知识蒸馏)
Knowledge Distillation把大模型的知识迁移到小模型
亦作、亦称:知识蒸馏 · KD · 模型蒸馏 · Model Distillation
Geoffrey Hinton 等人于 2015 年提出的模型压缩技术,通过让小型学生模型学习大型教师模型的知识表示(输出分布或特征),在保持较低参数量的同时获得接近教师模型的性能。维基百科将其定义为「将知识从大模型转移到小模型的机器学习方法」。
发展历程
知识蒸馏的发展可追溯至 2006 年 Rich Caruana 等人的工作,他们证明了可以通过训练单个模型模仿集成模型的 log 概率来转移知识。
2015 年,Geoffrey Hinton 等人在论文「Distilling the Knowledge in a Neural Network」中正式提出「蒸馏」概念,引入 softmax 温度参数 t 来控制输出分布的平滑度。
2020-2026 年,知识蒸馏在大语言模型时代获得新生:DistilBERT(2019)将 BERT 压缩到 66M 参数,保留 97% 性能;TinyLlama(2023)通过蒸馏将 Llama 2 70B 的能力迁移到 1.1B 模型;DeepSeek-R1(2025)通过蒸馏将推理能力迁移到 7B 模型。
2026 年,知识蒸馏已成为端侧 AI 的标准技术路径。
三大蒸馏范式
知识蒸馏按转移的知识类型分为三大范式。响应蒸馏(Response Distillation):学生模型学习教师模型的输出分布(软标签),这是 Hinton 原始论文的方法,使用交叉熵损失和高温 softmax。
特征蒸馏(Feature Distillation):学生模型学习教师模型中间层的特征表示,如 FitNets(2014)让学生模型模仿教师模型的隐藏层激活。
关系蒸馏(Relation Distillation):学生模型学习样本之间的关系结构,如 RKD(2019)让学生模型保持教师模型中样本间的距离和角度关系。
2026 年的趋势是自蒸馏(Self-Distillation)和多教师蒸馏(Multi-Teacher Distillation)——前者让模型在训练过程中蒸馏自己的知识,后者从多个教师模型集成知识。
2026 年产业实践
2026 年,知识蒸馏已成为端侧 AI 部署的核心技术。DeepSeek-R1 通过蒸馏将推理能力迁移到 7B 模型,在 SWE-bench 上达到 60% 解决率。TinyLlama 通过蒸馏将 Llama 2 70B 的能力迁移到 1.1B 模型,可在手机本地运行。
端侧部署:iPhone 17 Pro 的统一内存为 12GB,高通骁龙 8 Gen 4 的 NPU 算力约 75 TOPS,通过蒸馏的 7B 模型可在这类设备上实时运行。能耗优化:一个万亿参数模型的单次推理能耗约是 7B 模型的 100 倍,通过蒸馏可将能耗降低一个数量级。
开源生态:Hugging Face 的 Distillation Cookbook 提供了标准化的蒸馏流程,支持 PyTorch、TensorFlow、JAX 等主流框架。
常见误解
日常交流中容易听到的简化说法,未必准确,但能帮助理解误解从何而来。
- 「把大模型的知识迁移到小模型」
- 「Hinton 的暗知识理论」
相关术语
和本术语关联紧密的其他词条,便于串联理解。
🎯 考点练习
含该术语的高频面试题,含标准答案与追问。
- 中级概念查看详解 →
TensorFlow 中的计算图是什么?
TensorFlow 计算图将运算表示为有向无环图中的节点,边表示张量依赖;1.x 需先建图再用 Session 执行,2.x 默认即时执行但可用 tf.function 编译为图。
- 中级概念查看详解 →
TensorFlow Lite 是什么?适用于哪些场景?
TensorFlow Lite 是 TensorFlow 的移动端与嵌入式推理引擎,通过转换与量化压缩模型,在 Android、iOS、树莓派等边缘设备上低延迟运行深度学习模型。
- 中级概念查看详解 →
TensorFlow 中的 Variable 是什么?有何重要性?
TensorFlow 的 tf.Variable 是可训练的可变张量,用于存储模型权重和偏置;优化器通过 apply_gradients 更新 Variable,是训练中被学习的参数载体。
- 中级概念查看详解 →
什么是深度学习?与传统机器学习方法有何区别?
深度学习是机器学习子集,用多层神经网络自动学习层次特征表示;区别于传统 ML 需手工特征,深度学习可端到端从原始数据学习,但依赖更多数据与算力。
延伸阅读
从知识库精选 1 篇文章,帮助深入理解该术语。
外部参考
维基百科:查看「Knowledge Distillation」词条本页内容为本站原创撰写;维基百科链接仅作延伸参考。
