微软亚研提出TinyMIM,用知识蒸馏改进小型ViT|实验|vit|预训练|科学家|财务报表|财务会计|tinymim

用微信扫码二维码

微软亚研提出TinyMIM用知识蒸馏改进小

分享至好友和朋友圈

掩码建模(MIM,MAE)被证明是非常有效的自监督训练方法。然而,如图1所示,MIM对于更大的模型效果相对更好。当模型很小的时候(比如ViT-T5M参数,这样的模型对于现实世界非常重要),MIM甚至可能一定程度上降低模型的效果。比如用MAE训练的ViT-L比普通监督训练的模型在ImageNet上的分类效果提升3.3%,但是用MAE训练的ViT-T比普通监督训练的模型在ImageNet上的分类效果降低了0.6%。

在这篇工作中我们提出了TinyMIM,其在保持ViT结构不变并且不修改结构引入其他归纳偏置(inductIvebias)的基础上、用蒸馏的方法迁移大模型上的知识到小模型。

我们系统性的研究了蒸馏目标、数据增强、正则化、辅助损失函数等对于蒸馏的影响。在严格的只用ImageNet-1K作为训练数据的情况下(包括Teachermodel也只用ImageNet-1K训练)和ViT-B作为模型,我们的方法实现了当前最好的性能。如图所示:

把我们的方法(TinyMIM)和基于掩码重建的方法MAE,以及监督式学习的方法从头开始训练的DeiT作比较。MAE在模型比较大的时候有显著的性能提升,但是在模型比较小的时候提升幅度有限甚至会伤害模型的最终效果。我们的方法TinyMIM在不同模型的大小上都有大幅提升。

1.蒸馏的目标(Distillationtargets):1)蒸馏token之间的关系比单独蒸馏classtoken或者特征图(featuremap)更有效;2)用中间层作为蒸馏的目标更有效。

2.数据增强和模型正则化(Dataandnetworkregularization):1)用带掩码的图片效果更差;2)学生模型需要一点droppath,但是teacher模型不需要。

3.辅助损失函数(auxiliarylosses):MIM作为辅助损失函数没有意义。

4.宏观蒸馏策略(Macrodistillationstrategy):我们发现序列化的蒸馏(ViT-B->ViT-S->ViT-T)效果最好。

我们系统性的调研了蒸馏的目标,输入的图片,蒸馏目标模块。

2.1影响蒸馏效果的因素

a.中间block特征和输出特征

当i=L时,指的是Transformer输出层的特征。当i

b.注意力(Attention)特征和前馈层(FFN)层特征

Transformer每一个block有Attention层和FFN层,蒸馏不同的层会带来不同的影响。

在Attention层内会有Q,K,V特征,这些特征用于计算注意力机制,我们也调研了直接蒸馏这些特征。

Q,K,V用于计算注意力图,这些特征之间的关系也可以作为知识蒸馏的目标。

传统的知识蒸馏是直接输入完整的图片。我们的方法为了探索蒸馏掩码建模模型,所以我们也探索了带掩码的图片是否适合作为知识蒸馏时候的输入。

2.2知识蒸馏方法对比

最简单的方法就是类似DeiT直接蒸馏MAE预训练模型的classtoken:

2)特征蒸馏:我们直接参考了featuredistillation[1]作为对比

我们的方法在ImageNet-1K上预训练,而且教师模型也是在ImageNet-1K预训练。然后我们将我们预训练的模型在下游任务(分类、语义分割)上进行了微调。模型表现如图:

我们的方法显著超过之前基于MAE的方法,尤其是小模型。具体来讲,对于超小的模型ViT-T,我们的方法实现了75.8%的分类准确性,相比MAE基线模型实现了4.2的提升。对于小模型ViT-S,我们实现了83.0%的分类准确性,比之前最好的方法提升了1.4。对于Base尺寸的模型,我们的方法分别超过MAE基线模型和以前最好的模型CAE4.1和2.0。

同时我们也测试了模型的鲁棒性,如图所示:

TinyMIM-B对比MAE-B,在ImageNet-A和ImageNet-R分别提升了6.4和4.6。

同时蒸馏QK,VV关系而且在计算关系的时候有Softmax实现了最好的效果。

TinyMIM这种蒸馏关系的方法实现了比MAE基线模型,classtoken蒸馏,特征图蒸馏都更好的效果,在各种尺寸的模型上都是如此。

我们发现蒸馏第十八层实现了最好的效果。


本文地址: https://www.gpxz.com/article/ebbcc7ea04cd2b5a314d.html
全局中部横幅
全局中部横幅
mapbar.com

站长之家Alexa排名查询工具提供mapbar.com网站alexa排名查询,特别是能够对网站预估收入以及网站价值进行评估的功能,帮助用户更好的提升alexa排名。

中源钛业

钛及钛合金材料专业供应商

量子出行司机

量子出行司机,量子出行加盟,量子出行下载,叮叮出行加盟,叮叮出行司机,叮叮出行下载,量子一嗨,量子一嗨出行,量子一嗨下载,量子一嗨司机量子

康煌蚕丝被

康煌蚕丝被是高档桑蚕丝被品牌、高档欧式奢华真丝床品真丝四件套十大品牌之一.只采用100%纯天然特级桑蚕丝,重塑丝绸皇家尊贵,缔造高端优雅品质生活.

成都鱼缸

锦江区千彩阳光水族用品店多年来一直从事成都鱼缸定制,成都海鲜池定制,成都实木鱼缸等水族产品,咨询热线:18982061599

景区网红墙绘

金华尚拙墙绘有限公司_从事景区网红墙绘,美丽乡村墙绘,地面墙绘彩绘壁画,幼儿园墙绘,酒店壁画涂鸦,墙体手绘,古建彩绘,墙面彩绘公司拥有独立的设计团队,专业的墙绘设计方案,高品质的绘画服务,赢得超千家合作单位及个人信赖,目前已打造了超300000平方米墙绘绘画案例。

电机出厂测试系统

杭州银浩电子科技有限公司专业从事测试仪器、电源和控制测试系统的研究、开发、生产、销售和技术服务的厂家,主营电力测功机、磁滞测功机、磁粉测功机、电涡流测功机、电动车测功机、电机出厂测试系统、电机定子测试系统、发电机测试系统、水泵测试系统、电机测试系统等,产品畅销上海、江苏、浙江、山东、广东、安徽、四川等全国各地。

自由图书

自由图书,www.book5678.com提供pdf,epub,mobi,txt等格式文件资源,共享大量的图书资料,资料涵盖文学,小说,天文地理,计算机,英语等,免费为大家提供电子书下载平台,让大家早日实现读书自由!

电动观光车

五菱连续7年中国汽车工业30强企业,60年品牌积淀.专业生产电动观光车、高尔夫球车、燃油观光车、观光小火车等专用设备厂家,3.0阴极电泳技术,整车保用10年,一年节省80%售后费用.咨询热线:400-150-9850.

全局底部横幅