当前位置: 首页 > news >正文

模型蒸馏(Knowledge Distillation)

        知识蒸馏是 用一个大型、性能较强的教师模型(Teacher Model)来指导一个小型、轻量的学生模型(Student Model)学习 的方法。核心思想是:

  • 教师模型通常很大、精度高,但部署成本高。

  • 学生模型通常较小、推理快,但精度较低。

  • 通过让学生“模仿”教师的输出分布,学生可以学习到比直接训练数据更丰富的知识。

核心公式:

  1. 普通交叉熵训练学生模型:

          
  2. 知识蒸馏引入教师模型输出(soft target):

              

3.总损失:

细节和技巧

        还有个小技巧是可以增加标记位置token的损失惩罚。

高温度(T>1):使 Softmax 输出更加平滑,类别之间的差异变小。
低温度(T=1):得到标准的 Softmax 输出。
极低温度(T→0):使 Softmax 输出接近独热编码。

        Logits 是模型在最后一层的原始输出,表示模型对每个类别的预测置信度。它们通常需要经过 Softmax 函数处理,以转换为概率分布,用于分类决策和损失计算。在模型蒸馏中,Logits 通过温度参数调整,生成软标签,帮助学生模型学习教师模型的隐含知识。

词表对齐

词表对齐指的是 确保两个模型(教师和学生)在处理文本输入时,使用相同的词或子词(token)表示,即 tokenizer 输出的一致性

  • 主要目标是:让学生能够正确理解教师输出的概率分布或者中间表示

  • 如果教师和学生词表不对齐,直接用软标签蒸馏可能会出现“词不对应”的问题。

词表对齐的方式

        也可以使用伪标签微调,利用大模型教师生成大量高质量回答(文本级别伪标签);构建学生模型训练集:prompt + 教师生成回答;对学生模型(使用自身tokenizer)进行微调,学习教师的生成分布。

硬标签和软标签有什么区别?为什么要用软标签?

  • 硬标签:真实标签 one-hot,信息有限。

  • 软标签:教师模型输出的概率分布,包含类别关系。

  • 软标签的作用:帮助学生模型学习类别间的关系,提高泛化能力。

为什么蒸馏时要用温度 T?

  • 高温度 softmax 会使概率分布更平滑,凸显教师对其他类别的相似性。

知识蒸馏有哪些类型?

Logits 蒸馏(Classic Hinton KD):直接让学生模型 模仿教师模型输出的概率分布(softmax/logits)

特征蒸馏(Feature Distillation):不仅模仿输出,还让学生 对齐教师模型的中间层特征。通过 L2 或 MSE 损失匹配教师和学生的特征表示。

注意力蒸馏(Attention Distillation):学生模型不仅对齐特征,还 对齐教师的注意力模式(Attention Map)。在 Transformer 中常用。

关系蒸馏(Relational Distillation): 蒸馏的不再是单个样本的输出或特征,而是 样本之间的关系。样本间的相似性或距离被用作蒸馏目标。

类型匹配对象主要作用适用场景
Logits 蒸馏输出概率传递类别关系分类、模型压缩
特征蒸馏中间层特征学生学习教师内部表示CNN、Transformer
注意力蒸馏Attention Map学生学习关注模式NLP、ViT
关系蒸馏样本间关系学生理解数据分布关系图像识别、低资源任务
http://www.dtcms.com/a/602828.html

相关文章:

  • 电商网站合作网页设计图片主流尺寸
  • 投资网站建设及推广北京建设网服务大厅
  • 郑州建设网站的公司阳城seo排名
  • 宜春市城市建设网站苍南县住房和城乡规划建设局网站
  • 浅谈网站规划建设与管理维护企业站模板大全
  • 汕头网站建设维护网站建设的需要的工具
  • 网站提交了被收录后改怎么做php网站后台模板下载不了
  • Q3: create 和 create2 有什么区别?
  • 研发管理知识库(6)什么是CI/CD
  • 数据库知识整理——SQL数据更新
  • win7 iis架设网站思途旅游网站建设系统
  • 反编译易语言 | 探讨易语言的反编译方法与安全性分析
  • 无锡网站营销公司简介郑州加盟网站建设
  • 单位网站建设费用什么会计科目广告优化是做什么的
  • 自动化测试工具Katalon 全面介绍与实际体验
  • 游戏盾的流量清洗
  • 楚雄做网站敬请期待上一句
  • 山东鲁中公路建设有限公司网站站点传统的推广方式主要有
  • 电子商务网站建设读书笔记秒收网站
  • 互联网网站 数据库网站优化建设宁夏
  • 网站扫码怎么做的团智慧团建登录入口
  • 哪个网站可以做免费商业推广建设个人网站的要求
  • 4.1.0 EL9410 用于 E-bus 的电源端子模块介绍
  • 网站开发脚本语言农产品网站的品牌建设
  • 从0到1学习Qt -- 常见控件QWidget(二)
  • 网站建设程序文件一台云服务器可以做多个网站
  • Oracle:查询当前正在等待执行的SQL语句
  • 交通信息华建设网站鱼爪网商城网站如何建设
  • 炫酷做网站背景图环保网站建设的目的
  • iOS的runtime的理解以及常用的使用场景有哪些