How Do Vision Transformers Work

本文探讨了Vision Transformer(ViT)中的多头自注意力(MSA)机制,分析了MSA与卷积(Conv)的不同,以及如何协调两者以发挥优势。研究发现,MSA在优化网络时起到平滑损失平面的作用,但可能导致非凸优化问题。同时,MSA与Conv在特征融合和滤波特性上互补,提出的AlterNet结构结合了两者的优点,提高了模型性能。
摘要由CSDN通过智能技术生成

原文链接:How Do Vision Transformer Work, in ICLR 2022
代码链接:Github

引入

从CNN的角度来看,MSA可以看成是一种特殊的卷积变换,它作用在整个feature map上,具有大尺寸的核参数,因此MSA应该具有卷积操作的基础能力。但是MSA的长距离依赖建模能力对于模型预测性能的提升,本文作者提出了反对意见,作者认为给予模型适当的距离约束可以帮助模型学习更鲁棒的表示,例如一些局部建模MSA的工作,CSAN和SwinTransformer都仅在小窗口内计算自注意力,不仅在小数据集上表现优异,而且在大型数据集例如Imagenet-21K上达到了更好的性能。基于此,作者提出了针对MSA和ViT的三个核心问题:

  1. 我们需要MSA中的哪些属性来更好的优化网络,MSA的长期依赖建模到底对模型优化起到了怎样的作用?
  2. MSA的操作与Conv有何不同?
  3. 如何协调使用MSA和Conv才能更好的发挥它们各自的优势?

分析

问题一:我们需要MSA中的哪些属性来更好的优化网络,MSA的长期依赖建模到底对模型优化起到了怎样的作用?

结论:作者首先证明了MSA的本质是广义空间平滑操作,其本身的弱归纳偏置会扰乱神经网络的训练。MSA的一

  • 0
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值