Swin Transformer 一种旨在设计通用的backbone,却在应用了屡屡刷屏
自从2017年Google Brain 提出了 Transformer之后,便掀起了一股研究transformer的热潮,随着BERT在NLP领域的频繁刷屏,Transformer也开始在CV领域大展身手,从ViTa到DETR,卷积和Transformer之间的结合应用变得越来越紧密。有关以Transformer为backbone的研究也在如火如潮的进行当中。
Swin Transformer提出了Transformer应用在CV领域中的最需要克服的两点区别:
- 1: scale,图像中的内容不像NLP中是一个一个的词向量,里面的内容是有不同的大小的。目前大部分的序列转换都是以某个特定大小的图像块或者图像特征块作为一个序列的基本单元,这容易导致Transformer不能够很好的适应各种大小不一的物体。
- 2 higher resolution 和计算成本的扩张一个256*256的照片维度就高达65536维度,Transformer的原版处理的序列也就是几百几千维,即使通过卷积的方式进行了降维,由于Transformer的二次方计算复杂度那也是一个特别需要时间和内存的任务。
Swin Transformer用下列的方式解决了上述的问题:
下面我会用程序运行流程中各种输入维度和操作的详细信息来讲解Transformer的执行过程和各种维度变化
SwinTransformer整体的框架:
i n : ( B , C , H , W ) o u t : ( B , e m b e d _ d i m ∗ 2 s t a g e − 1 ) in:(B,C,H,W) \quad out:(B,embed\_dim*2^{stage-1}) in:(B,C,H,W)out:(B,embed_dim∗2stage−1)
- 1.首先输入一个批量的图片数据: ( B , C , H , W ) (B,C,H,W) (B,C,H,W)
- 2.进行块嵌入维度变为 ( B , H W p _ s 2 , e m b e d _ d i m ) 其 中 p _ s 表 示 块 大 小 , 块 是 这 个 图 像 处 理 中 的 最 小 单 位 (B,\cfrac{HW}{p\_s^2},embed\_dim)其中p\_s表示块大小,块是这个图像处理中的最小单位 (B,p_s2HW,embed_dim)其中p_s表示块大小,块是这个图像处理中的最小单位
- 3.进行ape(绝对位置嵌入)+Dropout
- 4.进行basic_layer的循环也就是上图中的四个stage阶段,注意每一个stage的输入为 ( B , H W p _ s 2 , e m b e d _ d i m ) (B,\cfrac{HW}{p\_s^2},embed\_dim) (B,p_s2HW,embed_dim)输出为 ( B , H W p _ s 2 / 4 , e m b e d _ d i m ∗ 2 ) (B,\cfrac{HW}{p\_s^2}/4,embed\_dim*2) (B,p_s2HW/4,embed_dim∗2),这里为了统一处理,假设每一个stage的输入都是 ( B , L , C ) (B,L,C) (B,L,C)
- 5. ( B , L , C ) − > ( B , C , L ) − > A d a p t i v e P o o l 1 d ( B , C , 1 ) − > ( B , C ) (B,L,C)-> (B,C,L)-> AdaptivePool1d(B,C,1) -> (B,C) (B,L,C)−>(B,C,L)−>AdaptivePool1d(B,C,1)−>(B,C)主要是通过池化消除掉了L维度,形成最终的C个特征,这C个特征就是最终提出出来的特征。 C = e m b e d _ d i m ∗ 2 s t a g e − 1 C = embed\_dim*2^{stage-1} C=embed_dim∗2stage−1
Basic_Layer(也就是图中的state的框架):
i n : ( B , L , C ) o u t : ( B , L / 4 , 2 C ) in:(B,L,C) \quad out:(B,L/4,2C) in:(B,L,C)out:(B,L/4,2C)
- 循环depth次SwinTransformerBlock,这个过程保持维度
- 进行一次patch_merging(即down_sample) 。 ( − > ( B , L / 4 , 2 ∗ C ) (-> (B,L/4,2*C) (−>(B,L/4,2∗C)
SwinTransformerBlock:
i n : ( B , L , C ) o u t : ( B , L , C ) 每 一 个 B l o c k 包 括 两 个 连 续 的 B l o c k , 一 个 带 S W , 一 个 不 带 in:(B,L,C)\quad out:(B,L,C)每一个Block包括两个连续的Block,一个带SW,一个不带 in:(B,L,C)out:(B,L,C)每一个Block包括两个连续的Block,一个带SW,一个不带
-
- 第一个带滑动窗口的SwinTransformerBlock,包含超参数窗口尺寸window_size=7
- 记录点: o u t 1 out_1 out1=x
- ( B , L , C ) − > n o r m − > ( B , H , W , C ) (B,L,C) -> norm->(B,H,W,C) (B,L,C)−>norm−>(B,H,W,C)
- 进行斜上方循环 R o l l ( s h i f t s i z e ) Roll(shift_size) Roll(shiftsize)操作, 其 中 s h i f t 、 s i z e = w i n _ s i z e / / 2 其中shift、_size=win\_size//2 其中shift、size=win_size//2
- window_partition操作 − > ( B H W w _ s 2 , w _ s , w _ s , C ) − > ( B ∗ n W , w _ s 2 , C ) -> (\cfrac {BHW}{w\_s^2},w\_s,w\_s,C)->(B*nW,w\_s^2,C) −>(w_s2BHW,w_s,w_s,C)−>(B∗nW,w_s2,C)
- 进行一次WindowAttention操作
- ( − > B , H , W , C ) − > 反 向 R o l l ( s h i f t s i z e ) (-> B,H,W,C)->反向Roll(shift_size) (−>B,H,W,C)−>反向Roll(shiftsize)
- 记录点: o u t 2 = x = x + o u t 1 out_2=x=x+out1 out2=x=x+out1
- − > L a y e r N o r m ( o u t 2 ) − > M L P ->LayerNorm(out_2)->MLP −>LayerNorm(out2)−>MLP
- o u t = x + o u t 2 out = x+out2 out=x+out2
-
- 第二个不带滑动窗口的SwinTransformerBlock
- 待续
WindowAttention:
i n : ( B ∗ n W , w _ s 2 , C ) o u t : ( B ∗ n W , w _ s 2 , C ) in:(B*nW,w\_s^2,C)\quad out:(B*nW,w\_s^2,C) in:(B∗nW,w_s2,C)out:(B∗nW,w_s2,C)
- − > Q K V 操 作 L i n e a r ( B ∗ n W , w _ s 2 , 3 C ) − > ( 3 , B ∗ n W , n h , w _ s 2 , C / n _ h ) ->QKV操作Linear(B*nW,w\_s^2,3C)->(3,B*nW,n_h,w\_s^2,C/n\_h) −>QKV操作Linear(B∗nW,w_s2,3C)−>(3,B∗nW,nh,w_s2,C/n_h)
- Q , K , V 瓜 分 第 一 个 维 度 即 d i m Q = d i m K = d i m V : ( B ∗ n W , n _ h , w _ s 2 , C / n _ h ) Q,K,V瓜分第一个维度即dimQ=dimK=dimV:(B*nW,n\_h,w\_s^2,C/n\_h) Q,K,V瓜分第一个维度即dimQ=dimK=dimV:(B∗nW,n_h,w_s2,C/n_h)
- Q**scale*,scale是一个放缩系数
- a t t e n = Q @ K a t t e n : ( B ∗ n W , n _ h , w _ s 2 , w _ h 2 atten = Q @ K atten:(B*nW,n\_h,w\_s^2,w\_h^2 atten=Q@Katten:(B∗nW,n_h,w_s2,w_h2
- a t t e n = a t t e n + m a s k + r p s ( r e l a t i v e p o s i t i o n b i a s ) atten = atten+mask+rps(relative position bias) atten=atten+mask+rps(relativepositionbias)
- x = s o f t m a x ( a t t e n , − 1 ) @ V x : ( B ∗ n W , n h , w _ 2 , C / n _ h ) x = softmax(atten,-1) @ V x:(B*nW,n_h,w\_^2,C/n\_h) x=softmax(atten,−1)@Vx:(B∗nW,nh,w_2,C/n_h)
- − > ( B ∗ n W , w _ s 2 , C ) -> (B*nW,w\_s^2,C) −>(B∗nW,w_s2,C)
- d r o p ( p r o j e c t ( x ) ) 维 度 不 变 drop(project(x))维度不变 drop(project(x))维度不变
PatchEmbedding:
i n : 标 准 图 像 输 入 ( B , C , H , W ) o u t : ( B , H W / p _ s 2 , e m b e d _ d i m ) = ( B , n P , e m b e d _ d i m ) in:标准图像输入(B,C,H,W)\quad out:(B,HW/p\_s^2,embed\_dim)=(B,nP,embed\_dim) in:标准图像输入(B,C,H,W)out:(B,HW/p_s2,embed_dim)=(B,nP,embed_dim)
- 首先 − > C o n v 2 d ( C , e m b e d d i m , k = p a t c h _ s i z e , s t r i d e = p a t c h s i z e ) : ( B , e m b e d d i m , H / p _ s , w / p _ s ) ->Conv2d(C,embed_dim,k=patch\_size,stride=patch_size):(B,embed_dim,H/p\_s,w/p\_s) −>Conv2d(C,embeddim,k=patch_size,stride=patchsize):(B,embeddim,H/p_s,w/p_s)
- − > ( B , H W / p _ s 2 , e m b e d _ d i m ) − > D r o u o u t : ( B , n _ P , e m b e d _ d i m ) ->(B,HW/p\_s^2,embed\_dim)->Drouout:(B,n\_P,embed\_dim) −>(B,HW/p_s2,embed_dim)−>Drouout:(B,n_P,embed_dim)
window_partion
i n : ( B , H , W , C ) o u t : ( B H W / w _ s 2 , w s , w s , C ) = ( B ∗ n W , w _ s 2 , w _ s 2 , C ) in:(B,H,W,C)\quad out:(BHW/w\_s^2,w_s,w_s,C)=(B*nW,w\_s^2,w\_s^2,C) in:(B,H,W,C)out:(BHW/w_s2,ws,ws,C)=(B∗nW,w_s2,w_s2,C)
aps (absolute position embedding)
用来和PatchEmbedding相加,P_E后的维度为 ( B , n _ P , C ) (B,n\_P,C) (B,n_P,C),aps的维度为 ( 1 , n _ P , C ) (1,n\_P,C) (1,n_P,C)是一个可学习的Param
relative_pos_bias
用来和注意力矩阵相加,注意力矩阵的维度是 ( n W ∗ B , n _ h , w _ s 2 , w _ s 2 ) (nW*B,n\_h,w\_s^2,w\_s^2) (nW∗B,n_h,w_s2,w_s2),rpb的维度是 ( n _ h e a d , w _ s 2 , w _ s 2 ) (n\_head,w\_s^2,w\_s^2) (n_head,w_s2,w_s2)。
- 作者注册了一个维度为 ( ( 2 ∗ w H − 1 ) ( 2 ∗ w W − 1 ) , n _ h e a d s ) ((2*wH-1)(2*wW-1),n\_heads) ((2∗wH−1)(2∗wW−1),n_heads)的可学习张量rpb,还注册了一个buff:rpi(relative_pos_index) ( W h ∗ W w , W h ∗ W w ) (Wh*Ww,Wh*Ww) (Wh∗Ww,Wh∗Ww)并且这个buff张量中的数字大小都在 0 − ( 2 ∗ w H − 1 ) ( 2 ∗ w W − 1 ) 0-(2*wH-1)(2*wW-1) 0−(2∗wH−1)(2∗wW−1)之间,然后调用的时候 r p b [ r p i . f l a t t e n ( ) , : ] − > ( n _ h e a d , w _ s 2 , w _ s 2 ) rpb[rpi.flatten(),:]->(n\_head,w\_s^2,w\_s^2) rpb[rpi.flatten(),:]−>(n_head,w_s2,w_s2)
mask
作者通过使用滑动窗口来实现了全局注意力,但是之前不是一个window中的后来和其它窗口混在一起,相互直接就别算注意力了,那么就应该注意力矩阵中把这些窗口中之前不在一个窗口的patch相互注意力填充的很小。
一般的对于注意力的计算流程
首先说明一下自注意力对于向量来说,这里计为(N,C)其输出应该是和输入是一样的。
- 首先需要计算一个q,k,v*,假设对于每一个向量的q,k,v维度都是d**,因此需要进行一次线性变换让(N,C)* (C,3d) ->3 个(N,d)。
- 然后得到了q:(N,d)和k(N,d),v(N,d)对一个需要
q
i
需
要
和
所
有
的
k
j
做
点
积
得
到
关
于
q_i需要 和所有的k_j做点积得到关于
qi需要和所有的kj做点积得到关于因此一共有N*N的式子:
o
i
,
j
=
∑
k
q
i
,
k
∗
k
j
,
k
o_{i,j}=\sum_{k}q_{i,k}*k_{j,k}
oi,j=∑kqi,k∗kj,k用 爱因斯坦表法就是
einsum("i k , j k -> i j",q,k)
- 刚刚得到的注意力矩阵**attn:(N,Ni)*表示了第i个向量关于所有的向量的注意力大小,可式子 o i = ∑ k v k ∗ a t t n i , k o_{i} =\sum_{k}v_k*attn_{i,k} oi=∑kvk∗attni,k.第二个位置不变即 o i , j = ∑ k v k , j ∗ a t t n i , k o_{i,j} =\sum_{k}v_{k,j}*attn_{i,k} oi,j=∑kvk,j∗attni,k,即einsum(k j , i k -> i,j)