知识蒸馏的简单理解
知识蒸馏是通过构建一个轻量化的小模型,利用性能更好的大模型的监督信息,来训练这个小模型大模型: Teacher model小模型:Student modelTeacher model输出的知识(knowledge)Student model 学习来自Teacher的监督的信息的过程–> Distillation(蒸馏)
知识蒸馏
1.概念介绍
知识蒸馏是通过构建一个轻量化的小模型,利用性能更好的大模型的监督信息,来训练这个小模型
大模型: Teacher model 小模型:Student model
Teacher model 输出的知识(knowledge) Student model 学习来自Teacher的监督的信息的过程–> Distillation(蒸馏)
2.种类
1.离线蒸馏:
Teacher model has pre-trained,Teacher模型的参数在蒸馏训练过程中保持不变,达到训练student模型的目的。
蒸馏的损失函数:distillation loss 计算 Teacher and student loss and student loss—>Loss
2.半监督蒸馏
利用Teacher模型的预测信息作为标签,来对student网络进行监督学习。
使用更少标注量的数据集,达到提升模型精度的目的
3.自监督蒸馏:
先开始训练student模型,在整个训练过程的最后几个epoch的时候,利用前面训练的student作为监督模型,在剩下的epoch中,对模型进行蒸馏。
3.知识蒸馏的功能
1.提升模型精度
2.压缩网络参数
3.图片标签之间的域迁移
4.降低标注量(半监督的方式)
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐



所有评论(0)