继vit之后,进一步证明了Transformer可以在视觉领域广泛应用,并且可以应用到半监督以及自监督中。
Swin Transformer:Hierarchical Vision Transformer Using Shifted Windows
Swin Transformer:用了移动窗口的层级式的Transformer,Swin来自Shifted Windows。
他可以像CNN一样做到层级式提取,使提取到的特征有多尺度的概念
摘要
Vit出现后虽然让大家看到了Transformer在视觉领域的潜力,但并不确定Transformer可以做掉所有视觉任务。
Swin Transformer可以作为一个通用的骨干网络。
面对的挑战:1、多尺度。 2、高像素。
移动窗口提高效率,并通过Shifted操作变相达到全局建模能力。
层次结构:灵活,可以提供各个尺度特征信息,容易使用到下游任务中。
最后分层设计和移位窗口方法也被证明对所有mlp体系结构都是有益的。
引言
Vit虽然可以全局建模,但单一尺寸和低分辨率减弱了对多尺寸特征的把握,然而对密集预测型任务多尺寸特征非常重要。同时Vit是全局建模,复杂度与图像大小为平方关系。
由此,借鉴CNN的先验知识和设计理念,提出了Swin Transformer。
1、复杂度
在小窗口内计算自注意力从而使复杂度降到线性,这利用了CNN中局部性的先验知识:同一个物体的不同部分和语意相近部分大概率相邻。
2、多尺寸特征
卷积神经网络为什么有多尺度特征?因为池化操作,池化能增大每一个卷积核的感受野。因此这里也提出一个类似池化的patch merging,将相邻的小patch合成大patch,增大感受野和获取多尺寸特征信息。有了这些多尺寸特征图就可以扔给分割,检测,这使它可以作为一个骨干网络。
shift操作
假设窗口大小为M,shift操作就是向右向下分别平移M//2个patch,然后在新的特征图重新进行自注意力,这让相邻窗口有了交互的能力。估计看到这里会一头雾水,先不用搞清,具体原理稍后就说。
先说说为什么会能够具有交互能力,比如第四行第四列的patch,最初只能和最左上角的窗口内的其他patch进行自注意力,但shift之后就可以和第四行第五列的patch,第五行第四列的patch,第五行第五列的patch进行自注意力,而这三个patch分别来自其它三个窗口,这就使得不同窗口间有了交互能力。再加上patch merging就可以做到全局自注意力操作了。
网络结构
整体结构
patch partition
与Vit类似,使用一个卷积核大小为4*4,步长为4,通道数为48的卷积神经网络后维度为(H/4,W/4,48)
linear Embedding
将维度调整到C,并且它还包含一个layer norm
Swin Transformer Block
输入输出维度不变。首先一个Swin Transformer Blocks包含两个block,分别将原有的Multi-head attention(MSA)替换成窗口注意力(Window-MSA)和滑动窗口注意力(Shifted Window-MSA)
Patch Merging
类似于池化,对特征图进行下采样,同时通道数变为2倍,后面细讲。
具体实现
Patch Merging
1 | 2 | 1 | 2 |
3 | 4 | 3 | 4 |
1 | 2 | 1 | 2 |
3 | 4 | 3 | 4 |
1 | 1 |
1 | 1 |
2 | 2 |
2 | 2 |
3 | 3 |
3 | 3 |
4 | 4 |
4 | 4 |
这里的1,2,3,4代表的是序号,也就是说先将序号相同的patch汇聚到一起,这时候有四个(H/2,W/2,C)的张量,再在C的维度依次拼接这四个张量得到(H/2,W/2,4C)的张量,对他在4C维度上进行layer norm。因为卷积神经网络每次下采样后通道数会变成两倍,因此我们通过一个Linear将其维度转变为(H/2,W/2,2C) 。
W-MSA
目的:减少计算量
缺点:窗口之间无法进行信息交互。
做法:只对window之内的patch做self-attention。
以下是复杂度对比,推导见Swin-Transformer网络结构详解_swin transformer_太阳花的小绿豆的博客-CSDN博客
SW-MSA
再来仔细看这幅图:
将窗口向右向下分别平移M//2个patch,这时会出现9个窗口,并且每个窗口内的patch数不一样。
有两种解决方法,一种是对每个窗口进行补0,但这样就从一开始要计算的4个窗口变成了计算9个窗口,计算量大大增加。一种是通过masking方法来减少计算。
先将AC移到下方,再讲BA移到右方,被移动的和它现在相邻的在原本是没有关系的。比如原来C是天,最下面的是地,现在C被移到最下方,也不能说天在地下面。因此这些原本无关的patch就不能进行自注意力计算,因此我们需要使用masking。
现在我们已经得到了4*4的window,怎么进行mask呢?(如果看不懂强烈建议去学习朱导的精读,时间45:00。)
首先一个窗口有7*7个patch,那么向下向右平移7//2=3个单位。假设各部分按此编号。我们将窗口2展开,各个patch的来源应该是
3,3,3,3,3,3,3,
3,3,3,3,3,3,3,
3,3,3,3,3,3,3,
3,3,3,3,3,3,3,
6,6,6,6,6,6,6,
6,6,6,6,6,6,6,
6,6,6,6,6,6,6,
前28个来自3,后21个来自6。自注意力后得到
attention矩阵维度是49*49,左上角28*28的块是3号区域内的attention,右下角的21*21是6号区域内的attention,我们需要mask掉3号区域与6号区域的attention就可以。作者提到因为attention都是一个零点几的小数,所以我们直接给要mask掉的地方-100就可以做到softmax后其值为0.
类似的我们看看编号为12的窗口,它展开应该是这样的
1,1,1,1,2,2,2,
1,1,1,1,2,2,2,
1,1,1,1,2,2,2,
1,1,1,1,2,2,2
与刚才的方法类似,于是我们得到各个窗口的mask如下
最终我们还要把刚刚移动的移回去就大功告成。
相对位置偏移
因为本文仅供自己复习用,如果看不懂我写的强烈建议学习12.1 Swin-Transformer网络结构详解_哔哩哔哩_bilibili
时间37:00
这里的B就是相对位置偏移 。首先,在Swin Transformer中使用绝对位置编码表现并不好,其次相对位置编码表现得很好,因此我们用相对位置编码。。。
1、相对位置索引
假如特征图如下
a | b |
c | d |
这时候a,b,c,d的绝对位置索引是
0,0 | 0,1 |
1,0 | 1,1 |
a,b,c,d与其他的相对位置矩阵分别为(自己的索引减去别人的索引)
0,0 | 0,-1 |
-1,0 | -1,-1 |
0,1 | 0,0 |
-1,1 | -1,0 |
1,0 | 1,-1 |
0,0 | 0,-1 |
1,1 | 1,0 |
0,1 | 0,0 |
将四个矩阵展平,拼接
0,0 | 0,-1 | -1,0 | -1,-1 |
0,1 | 0,0 | -1,1 | -1,0 |
1,0 | 1,-1 | 0,0 | 0,-1 |
1,1 | 1,0 | 0,1 | 0,0 |
然后再将二维坐标转换为一维坐标,如果直接将行列坐标相加,会发生之前索引不同而之后索引相同的情况,如(-1,0)和(0,-1)
于是我们先给行和列都加上M-1,在这个例子中M=2。
说白了其实就是消除负数
1,1 | 1,0 | 0,1 | 0,0 |
1,2 | 1,1 | 0,2 | 0,1 |
2,1 | 2,0 | 1,1 | 1,0 |
2,2 | 2,1 | 1,2 | 1,1 |
然后我们给行标乘2M-1再和列标相加
4 | 3 | 1 | 0 |
5 | 4 | 2 | 1 |
7 | 6 | 4 | 3 |
8 | 7 | 5 | 4 |
通过上述方法我们让刚刚索引相同的值仍然相同,索引不同但行标列标相加后相同的情况索引不再相同,比如刚刚的(0,-1)和(-1,0)
2、相对位置偏置
我们根据相对位置索引去相对位置偏置表中取需要的值,相对位置偏置表的元素个数为(2M-1)*(2M-1),他是一个可学习的参数,相对位置索引是固定的
最终得到相对位置偏置
0.1 | 0.8 | 0.2 | 0.1 |
0.6 | 0.1 | 0.3 | 0.2 |
0.4 | 0.4 | 0.1 | 3 |
0.7 | 0.4 | 0.6 | 0.1 |
模型配置参数
消融实验
结论
复杂度同图像大小是线性关系。基于移动窗口的自注意力在密集预测型任务中非常有效。
最后,本文章仅供学习