微软亚研提出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输出层的特征。当iTransformer中间层的特征。

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蒸馏,特征图蒸馏都更好的效果,在各种尺寸的模型上都是如此。

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


本文地址: /article/ebbcc7ea04cd2b5a314d.html
全局中部横幅
全局中部横幅
百合网_免费沟通_免费交友_免费征婚_相亲_专业婚恋服务

百合网--实名婚恋。免费通信,谈情说爱不再为钱所困。1.2亿会员为爱而来,给你1.2亿的爱情火种。心理学专业恋爱类型测试帮你找到对的TA。

湖南生活服务网站 - 湖南生活服务网站排名 - 网站排行榜

湖南生活服务网站排名,根据网站的综合值按照不同的湖南生活服务网站进行筛选排名结果,通过筛选湖南生活服务网站可以看到每个湖南生活服务网站里面的网站排名优质的网站是哪些

湖州培训网_湖州教育培训学校_湖州培训辅导机构信息网 - 厚学网

厚学网-湖州领先的培训学校、培训课程信息网.覆盖400个以上城市,超过10万家教育培训机构,100万个教育培训课程辅导班及相关优惠的培训课程费用信息,您要的优质培训这里都有.找培训,上厚学网!

高端齿轮箱,齿轮-常州市南方驱动技术有限公司

常州市南方驱动技术有限公司是一家专业从事高端齿轮箱制造的企业。产品应用于石化、冶金、煤矿、造船、电力、交通、工程机械等领域,公司从德国、美国、奥地利等国家引进了世界一流齿轮、机加工、热处理及检测设备,旨在为客户提供齿轮科技领域高附加值的产品及优良的品质;公司具有优良的加工制造能力及一流的技术、管理团队,公司本着“专业、诚信、协作、开拓”的精神,为客户提供系统的解决方案及服务。公司倡导尊重、理解、沟通的工作氛围,鼓励开拓与创新,以实现员工、客户、企业三方共赢。

网站建设_网站建设制作_网站设计_企业建站_网站开发_建网站-星辉互联

星辉互联创于2006年,一站式网站建设和网站建设系统。满足各类,网站建设、网站建设制作、建网站、网站搭建、网站建设订制等需求。是500,0000家用户建站首选!提供7x24小时网站建设,企业零基础在线建站,让网站建设不再困难。

保定驾校收费标准-保定学车哪家好-竞秀区莲池区高新区驾校-保定市驰安机动车驾驶员培训有限公司

保定市驰安机动车驾驶员培训有限公司,驾校主要负责:保定驾校收费标准、保定学车哪家好、竞秀区莲池区高新区驾校、保定大学生信赖的驾校、保定正规驾校、保定驾校价格、保定正规驾校哪家好等,是我省交通厅验收合格的驾驶员培训学校,具有交通部颁布的《机动车驾驶员培训学校》的办学资格。驰安驾校以严格的管理,规范的教学,热情的服务,赢得了市场的好评。驰安驾校基础设施建设齐全,训练场地设施全部按照123号令建设与考场同步。

浙江亨司迈智能机械有限公司

顶峰机械致力于金属带锯床的设计研发、生产制造和销售服务为一体,现共有各种智能数控,全自动,半自动,卧立式,双柱龙门式金属带锯床。

IPSA官网 - IPSA流金水自律乳-IPSA茵芙莎中国官网

欢迎访问IPSA茵芙莎中国官网!IPSA官网有IPSA流金水、IPSA自律乳等热门护肤产品,以及IPSA遮瑕、IPSA防晒等美妆产品。IPSA茵芙莎致力于通过先端的技术背景和人性化的服务方式,挖掘出每位女性与生俱来的美丽潜能。IPSA官网确保IPSA流金水,IPSA自律乳等产品正品发售。IPSA茵芙莎中国官网期待您的到来!

厦金机械集装箱旋转机,翻转机,石材叉装机,矿山叉装车,夹木机,装载机

厦金机械成立于2000年,有十七年的大吨位叉装机研发、生产、销售、售后服务经验,是五项叉装机国家标准起草单位

朗镜科技-实体零售货架商品图像识别|渠道终端门店稽查|神秘访客|数据化陈列|众包渠道检查-朗镜科技

朗镜科技是领先的新零售AI和大数据服务平台,为您提供货架图像识别,实体零售图像识别,商品图像识别,消费品图像识别,零售终端货架识别,众包渠道检查,数据化陈列,货架商品识别,神秘访客,渠道终端稽查,门店稽查服务。

全局底部横幅