Processing math: 100%

Vision Transformer

2021/11/18 Neural_Network 共 2297 字,约 7 分钟

Vision Transformer:

接着来开个ViT的新坑,论文链接: An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale

因为Transformer的计算效率和可扩展性,该文采用了此结构(可参考之前的博文:Transformer),主要分为Embedding、Transformer Encoder、MLP Head三部分,下面就按这个顺序进行整理。

image-20211203101814028

1. Embedding:

Transformer Encoder 的输入要求为token,也就是一个形状为[numtoken,tokendim]的矩阵。

编码的具体流程参考下图中,对于形状为[H,W,C]的输入图像而言,被分成了很多的形状为[P,P,C]的块,可以算的最终块的数目为:H×WP×P,所以总体形状变为[H×WP×P,P×P×C],然后再经过一个线性映射总体形状变为[H×WP×P,D],此处的D是自定义的。

接下来,还需加入一个用于分类的cls,因此token形状变为[H×WP×P+1,P×P×C]

此外,为了保证块的位置信息,再在token中加上可以学习的位置编码参数,最终得到的token大小仍然是[H×WP×P+1,P×P×C]

image-20211203103822943

这里提一下,代码中切割patch可以用conv16×16 Stride=16的卷积操作来实现,而在position embedding后还需要经过一步dropout(emd_dropout)才进入了encoder block里面。

2. Transformer Encoder:

image-20211203112220105

Encoder这里就和Transformer的Encoder差不多。值得注意的是这里每一层前,都要进行层归一化(Layer Norm,可以参考之前的博文:Batch Normalization)。此外,每层之后包含一个残差链接。而MLP那里也很简单,就是Linear->GELU->Linear->Dropout,其中第一个Linear会把输入节点翻成4倍,形状变为[H×WP×P,P×P×C×4],第二个Linear再把它缩小4倍。

从Transformer Encoder出来之后,在进入MLP head之前,还需要进入一个Layer Norm。

因此原文中将上述过程简单的用公式描述为:

z0=[xclass;x1pE;x2pE;;xNpE;]+Epos,ER(P2C),EposR(N+1)×Dzl=MSA(LN(zl1))+zl1l=1Lzl=MLP(LN(zl))+zll=1Ly=LN(z0L)

2.1 归纳偏置:

相比于CNN,ViT含有更少的图像归纳偏置,这种偏置被称为偏好更恰当,是因为学习算法必须有某种偏好,才能产生它认为正确的模型,换而言之就是算法在其假设空间内,进行选择的启示录或价值观。

CNN中,局部性、二维相邻结构以及平移同变性这些归纳偏置,是嵌入到每一层贯穿整个模型的。而ViT中只有MLP部分有局部性和平移同变性,且自注意力层是全局性的,此外二维相邻结构仅在切割图像为块和在微调时为不同分辨率图像调整位置编码时会使用到。

2.1.1 平移不变性/同变性:

这里简单整理下,平移不变性(Translation Invariance)和平移同变性(Translation Equivariance),就是一张图片中的主要内容发生了平移后,分别在图像分类和目标检测两种任务中体现的不同结果的原因。

img

来看上面这张图,其中的X1经过平移后得到X2中,二者中的数字4的含义其实是一样的,而再经过相同的映射方法之后,得到的特征图F1F2,仍然可以通过之前的平移操作实现转换。针对于分类任务,得到的分类结果是相同的,而检测任务中,二者的候选框却是不同的。

更具体的讲解可以参考这两篇文章关于平移不变性与平移相等性的理解与讨论What is translation equivariance, and why do we use convolutions to get it?

2.2 GELU:

高斯误差线性单元(Gaussian Error Linerar Units),(这一段是看网上博客整理的材料,没有去翻原论文,不保证准确性)就是在ReLU中引入了随机正则(Dropout)的思想,根据自身输入的情况来,来决定抛弃还是保留当前神经元。

函数形式:GELU(x)=0.5x(1+tanh(2/π(x+0.044715x3)))

3. MLP Head:

然后这个MLP Head感觉没啥可以讲的,就是一个线性层。

最后来一个表简单的理解一下整个网络结构。之后再详细补上代码等内容

image-20211203114949262

文档信息

Related Issues not found

Please contact @mzlogin to initialize the comment

Search

    Table of Contents

    文章目录