Transformer ================== `Transformer `_ 是一个典型的 Encoder-Decoder 模型，采用完全依赖于注意力机制的架构。 .. image:: ./13_transformer.png :width: 400 px :align: center Encoder 和 Decoder 均由 6 个 Block 堆叠而成。 输入 ----------- 包括 Input Embedding 和 Positional Encoding（正弦/余弦编码）。 Encoder ------------ Encoder 由 3 个子模块构成： - Multi-Head Self-Attention - Layer Normalization - Feed Forward Network Decoder -------------- Decoder 由 4 个子模块构成： - Multi-Head Self-Attention - Multi-Head Encoder-Dencoder Attention 交互模块 - Layer Normalization - Feed Forward Network Encoder 端可以并行计算，一次性将输入序列全部 Encoding 出来，但 Decoder 端不是一次性把所有 Token 预测出来的，而是像 seq2seq 一样一个接着一个预测出来的。 Self-Attention -------------------------------- .. image:: ./13_attention.png :width: 600 px :align: center 在 Self-Attention 中，序列中的每个单词和该序列中其余单词进行 Attention 计算。Self-Attention 的特点在于无视 Token 之间的距离直接计算依赖关系，从而能够学习到序列的内部结构，实现起来也比较简单。引入 Self-Attention 后会更容易捕获句子中长距离的相互依赖的特征。 .. math:: \mathrm{SelfAttention}(Q,K,V) = \mathrm{softmax} \left( \frac{QK^T}{\sqrt{d}} \right)V 其中， :math:`Q=XW_Q,\ K=XW_K,\ V=XW_V \in \mathbb{R}^{n \times d}` 。由于这 3 个矩阵都是由输入 :math:`X` 得到，因此叫 Self-Attention。 使用 :math:`\boldsymbol{q}` 和 :math:`\boldsymbol{k}` 两组向量来计算权重（而不是一组），增加了权重的自由度，权重有可能是非对偶的（即 :math:`\boldsymbol{q}_i^{\top}\boldsymbol{k}_j \ne \boldsymbol{k}_i^{\top}\boldsymbol{q}_j` ），增强了模型的表达能力。 假设两个 :math:`d` 维向量的每个分量都是一个相互独立的服从标准正态分布的随机变量，那么他们的点乘方差就是 :math:`d` ，对每一个分量除以 :math:`\sqrt{d}` 可以让点乘的方差归一化成 1。 Multi-Head Attention 相当于 :math:`h` 个不同的 Self-Attention 的集成。将模型分为多个头，形成多个子空间，可以让模型去关注不同方面的信息，最后再将各个方面的信息综合起来。类比 CNN 中同时使用多个卷积核的作用，多头的注意力有助于网络捕捉到更丰富的特征/信息。 .. figure:: ./13_example.png :width: 600 px :align: center `Self-Attention 示例 `_ 计算复杂度 ^^^^^^^^^^^^^^ 假设序列长度为 :math:`n` ， :math:`Q,K,V \in \mathbb{R}^{n \times d}` 。 Self-Attention 三个步骤的复杂度： - 相似度计算： :math:`\mathcal{O}(n^2 d)` - Softmax： :math:`\mathcal{O}(n^2)` - 加权平均： :math:`\mathcal{O}(n^2 d)` 因此总的时间复杂度是 :math:`\mathcal{O}(n^2 d)` 。 Multi-Head Attention 的实现不是循环地计算每个头，而是通过 Transposes and Reshapes，把一个大矩阵相乘变成了多个小矩阵的相乘。 Multi-Head Attention时间复杂度也是 :math:`\mathcal{O}(n^2 d)` （每个头的对应的 :math:`\boldsymbol{q},\boldsymbol{k},\boldsymbol{v}` 向量维度为 :math:`\frac{d}{h}` ）。 .. note:: 如果考虑从 :math:`X` 到 :math:`Q,K,V` 的线性映射，时间复杂度还需要加上 :math:`\mathcal{O}(n d^2)` 。 Layer Normalization --------------------------- Layer Normalization 的作用是对神经网络中隐藏层输出的 Embedding 施加尺度约束，将其归一为标准正态分布，以起到加快训练速度、加速收敛的作用。 CV 领域的应用 ------------------------- `ViT `_ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ .. image:: ./13_vit.png :width: 600 px :align: center ViT将输入图片分为多个 Patch，再将每个 Patch 投影为固定长度的向量（Patch Embedding）送入 Transformer，后续 Encoder 的操作和原始 Transformer 中完全相同。 对于图像分类任务，在输入序列中加入一个特殊的 Token，它对应的输出即为最后的类别预测。 当训练数据集不够大的时候，ViT 的表现通常比同等大小的 ResNet 要差一些，因为 Transformer 和 CNN 相比缺少归纳偏置（Inductive Bias），即一种先验知识/提前做好的假设。 CNN 具有两种归纳偏置： - 局部性（Locality/Two-dimensional Neighborhood Structure）：图片上相邻的区域具有相似的特征； - 平移不变性（Translation Invariance）：当目标出现在图像的不同位置时，也能得出相同的分类结果（主要依靠 Global Pooling）。 当 CNN 具有以上两种归纳偏置，就有了很多先验信息，需要相对少的数据就可以学习一个比较好的模型。 Patch Embedding ++++++++++++++++++++ 假设输入图片大小为 :math:`224 \times 224 \times 3` ，将图片分为固定大小的 Patch，每个 Patch大小为 :math:`16 \times 16 \times 3` ，则每张图像会生成 :math:`\frac{224 \times 224}{16 \times 16} = 196` 个 Patch，即输入序列长度为 :math:`196` 。 每个 Patch 维度 :math:`16 \times 16 \times 3 = 768` ，线性投射层的维度为 :math:`768 \times 768` ，因此输入通过线性投射层之后的维度依然为 :math:`196 \times 768` ，即一共有 :math:`196` 个 Token，每个 Token Embedding 的维度是 :math:`768` 。 这里还需要加上一个特殊字符 cls，因此最终的维度是 :math:`197 \times 768` 。 经过 Transformer Encoder 的维度仍然是 :math:`197 \times 768` ，取特殊字符 cls 对应的输出第 0 个 :math:`768` 维的向量作为 Encoder 的最终输出 ，代表最终的 Image Presentation（另一种做法是不加 cls 字符，对所有的 Token 的输出向量取平均），再接 MLP 进行分类。 Positional Encoding +++++++++++++++++++++++ 有 1-D 位置编码和 2-D 位置编码。 位置编码可以理解为一张表，表一共有 :math:`N` 行，和输入序列长度相同；每一行代表一个向量，向量的维度和 Patch Embedding 的维度相同（ :math:`768` ）。 位置向量和 Patch Embedding 相加（保证维度不变），作为 Transformer Encoder 的输入。 实验结果表明，不管使用哪种位置编码方式，模型的精度都很接近，甚至不使用位置编码，模型的性能损失也没有特别大。 原因可能是 ViT 是作用在 Image Patch 上的，而不是 Pixel，对网络来说这些 Patch 之间的相对位置信息很容易理解。 当输入图片分辨率发生变化，输入序列的长度也发生变化，虽然 ViT 可以处理任意长度的序列，但是预训练好的位置编码无法再使用（例如原来是 :math:`3 \times 3 = 9` 个 Patch， 每个 Patch 的位置编码都是有明确意义的，如果 Patch 数量变多，位置信息就会发生变化）。 一种做法是使用插值算法，扩大位置编码表。但是如果序列长度变化过大，插值操作会损失模型性能，这是 ViT 在微调时的一种局限性。 CNN + Transformer 混合模型 +++++++++++++++++++++++++++++ 既然 CNN 具有归纳偏置的特性，Transformer 又具有很强全局归纳建模能力，使用 CNN + Transformer 的混合模型是不是可以得到更好的效果呢？ :math:`224 \times 224 \times 3` 的图像经过卷积层得到 :math:`14 \times 14 \times 768` 的 Feature Map，拉成一个 :math:`196 \times 768` 的二维 Tensor，后续操作和 ViT 相同。 .. note:: Patch Size 增大一倍，Patch 数量（即序列长度）会减少到原来的四分之一，在维持 Hidden Layer 维度不变的情况下，计算量将会大大减少。 `Swin Transformer `_ ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ .. image:: ./13_swin1.png :width: 400 px :align: center Swin Transformer 主要有两个创新： - 使用了层级式多分辨率的 Patch（Hierarchical Feature Maps）。 先是进行 4 倍下采样，然后是 8 倍，接着是 16 倍，一直下采样直到达到预期的下采样倍数。相比之下，ViT 只使用了 16 倍的下采样。 这样利于支持更多的图像下游任务，而不仅仅是分类。 - 窗口式的 Self-Attention（W-MSA，SW-MSA）。 将 Feature Map 划分成一个个的小窗口，将这些小窗口送入到 Swin Transformer Block 中，减少了计算量。 模型框架 ++++++++++++++++++ .. image:: ./13_swin2.png :width: 800 px :align: center .. note:: Transformer 的输入输出都是二维的矩阵，比如图中的 :math:`\frac{H}{4} \times \frac{W}{4} \times C` 其实是 :math:`\frac{HW}{16}\times C` 。 Patch Partition + Linear Embedding """""""""""""""""""""""""""""""""""""""" 同 ViT。 Patch Merging """""""""""""""""""""""""""""""""""""""" 实现 Feature Map 分辨率减半，通道数翻倍。 Swin Transformer Block """""""""""""""""""""""""""""""" W-MSA（Windows Multi-Head Attention）和 SW-MSA（Shifted Windows Multi-Head Attention）代替了原生版本的 Multi-Head Attention。 W-MSA 将 Feature Map 分成一个个 Windows，然后对每个 Windows 执行 Multi-Head Attention 操作，减少了计算量。 W-MSA 的各个 Windows 之间相互独立、缺乏交互，SW-MSA 会重新划分窗口，使得之前各个 Windows 的数据可以进一步融合、传递（需要 Mask 掉原本不相邻的数据块）。 .. image:: ./13_swin3.png :width: 500 px :align: center Relative Position Bias """""""""""""""""""""""""""""""" 相对位置偏置 :math:`B` 融合进 Attention 的公式中： .. math:: \mathrm{Attention}(Q,K,V) = \mathrm{softmax} \left( \frac{QK^T}{\sqrt{d}} + B \right)V .. note:: ViT 和 Swin Transformer 的实现： - https://huggingface.co/docs/transformers/model_doc/vit - https://huggingface.co/docs/transformers/model_doc/swin 参考资料 ---------------- 1. Transformer https://paddlepedia.readthedocs.io/en/latest/tutorials/pretrain_model/transformer.html 2. Transformer https://xiaomindog.github.io/2021/01/20/Transformer/ 3. The Illustrated Transformer https://jalammar.github.io/illustrated-transformer/ https://zhuanlan.zhihu.com/p/48508221 4. 举个例子讲下transformer的输入输出细节及其他 https://zhuanlan.zhihu.com/p/166608727 5. ViT（Vision Transformer）解析 https://zhuanlan.zhihu.com/p/445122996 6. Swin Transformer原理详解篇 https://juejin.cn/post/7157873015055712293 7. Hugging Face Transformers https://huggingface.co/docs/transformers/index