Swin transformer 模型开创性的使用了窗口注意力机制的概念。 我们知道VIT模型虽然把图片分割成一个一个的patch ,但是其模型是在整个图片上来做注意力机制,其计算复杂度依然很高。为了解决计算复杂度的问题,Swin transform

人工智能研究所 人工智能研究所 墨衍会员 · AI 创作全网分发 墨衍智能分发 通过墨衍一键同步至各平台 可同步平台 CSDN微信公众号微博知乎掘金百家号博客园抖音小红书 你的文章也可以这样分发 墨衍会员 ¥399起/年,写一次发全网 我也要 + 加入社区
Swin transformer 模型开创性的使用了窗口注意力机制的概念。 我们知道VIT模型虽然把图片分割成一个一个的patch ,但是其模型是在整个图片上来做注意力机制,其计算复杂度依然很高。为了解决计算复杂度的问题,Swin transformer 模型提出了窗口注意力机制的概念。 在把图片分割成一个个patch后,模型把patch再分割成一个个的小窗口,每个小窗口中,自己计算注意力机制。虽然这样解决了计算复杂度的问题,但是图片毕竟是一个整体,窗口与窗口之间没有任何交互,肯定是不行的。 Swin transformer 在窗口注意力机制的基础上,又提出了移动窗口的概念,当计算完窗口注意力机制之后,需要把窗口进行移动,让窗口与窗口之间有重合的部分,这样再计算一遍注意力机制,就得到了窗口之间的关系。 从Swin transformer 模型的框架上可以看出,其模型的transformer 部分每次都是成对出现的,一个是窗口注意力机制,另外一个是移动窗口注意力机制。#动图详解Transformer
...全文

相关推荐

评论
收藏 已收藏
分享
返回顶部