图解transformer | The Illustrated Transformer

作者博客：@Jay Alammar

原文链接：The Illustrated Transformer – Jay Alammar – Visualizing machine learning one concept at a time. (jalammar.github.io)

---

翻译讲究：信、达、雅。要在保障意思准确的情况下传递作者的意图，并且尽量让文本优美。

但是大家对我一个理工科少女的语言要求不要太高，本文只能保证在尽量通顺的情况下还原原文。

注意本文的组成部分：翻译 + 我的注释。

添加注释是因为在阅读的过程中，我感觉有的地方可能表述的并不是特别详细，对于一些真正的小白，可能不太好理解，所以我做了一点注释。

---

在之前的文章我们讲了现代神经网络常用的一种方法——注意力机制（Attention）。

本文我们继续介绍一下Transformer——用注意力机制来提高模型训练速度的模型。
Transformer在某些特定任务上性能比谷歌的机器翻译模型更为优异，其优点在于并行化计算。并且谷歌云也推荐使用transformer作为参考模型来运行他们的TPU云服务。所以让我们来把它拆解开看一下它是如何发挥作用的。

• 论文：Attention is All You Need.

• 官方代码（TensorFlow）：GitHub - tensorflow/tensor2tensor: Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.

• 哈佛大学NLP组代码（Pytorch）：The Annotated Transformer (harvard.edu)

在本文中，我们会逐个概念进行介绍，希望能帮助没接触过Transformer的人能够更容易的理解。

从高层面看

我们先把整个Transformer模型看作是一个黑盒。在机器翻译中，它可以把句子从一种语言翻译成另一种语言。

打开这个黑盒，我们首先可以看到一个编码器（encoder）模块和一个解码器（decoder）模块，以及二者之间存在某种关联。

作者原文说的是“Popping open that Optimus Prime goodness”，也就是“噢我的老天鹅让我们打开这个擎天柱看一下。”

擎天柱：？？？你在叫我吗？

我之前看过一个大佬解释说transformer其实取的是“变压器”这个意象，但是因为变形金刚这个印象好像更为广为人知一点，所以人们提到transformen一般想到的都是变形金刚，作者在这里就直接默认是擎天柱了。
这倒我想起来微软和NVIDIA还合作了一个威震天（Megatron Turing—NLG），感兴趣的可以自己去了解一下。

再往里看一下，编码器模块是6个编码器组件堆叠在一起，同样解码器模块也是6个解码器组件堆。（为什么选6个呢？没有什么原因，论文原文就是这么写的，你也可以换成别的层数）

在这里这个encoder和decoder中的基础组件个数属于超参数，可以自己尝试设置不同的值。这是Transformer中第一个可调参数。

6个编码器组件的结构是相同的（但是他们之间的权重是不共享的），每个编码器都可以分为2个子层。

编码器的输入首先会进入一个自注意力层，这个注意力层的作用是：当要编码某个特定的词汇的时候，它会帮助编码器关注句子中的其他词汇。这个之后会进行详细讲解。

自注意力层的输出会传递给一个前馈神经网络，每个编码器组件都是在相同的位置使用结构相同的前馈神经网络。

这里要注意一点，在前面我们已经说了组件之间的参数是不共享的。这里的相同仅限于结构相同，它们的参数是不同的！！！

解码器组件也含有前面编码器中提到的两个层，区别在于这两个层之间还夹了一个注意力层，多出来的这个自注意力层的作用是让解码器能够注意到输入句子中相关的部分（和seq2seq中的attention一样的作用）。

图解张量

现在我们要开始了解整个模型了。

在一个已经训练好的Transformer模型中，输入是怎么变为输出的呢？

首先我们要知道各种各样的张量或者向量是如何在这些组件之间变化的。

与其他的NLP项目一样，我们首先需要把输入的每个单词通过词嵌入（embedding）转化为对应的向量。

在原文中每个词的嵌入向量是512维，这里为了便于理解，就用这几个格子进行表示。

虽然这里只画了4个格子，但是我们用4个格子表示512的维度，不是说用4个格子表示四维。

所有编码器接收一组向量作为输入，论文中的输入向量的维度是512。最底下的那个编码器接收的是嵌入向量，之后的编码器接收的是前一个编码器的输出。

向量长度这个超参数是我们可以设置的，一般来说是我们训练集中最长的那个句子的长度。

这是Transformer中第二个超参数，我们可以自行设置。到这里严格意义上讲，Transformer的超参数都出来了。

“这个简单的设计影响到后面一系列的网络，BERT啊、GPT啊其实只有两个参数可以调的。”——李沐

当我们的输入序列经过词嵌入之后得到的向量会依次通过编码器组件中的两个子层。

在这里，我们开始看到Transformer的一个关键属性，即每个位置上的单词在编码器中有各自的流通方向。在自注意力层中，这些路径之间存在依赖关系。 然而，前馈神经网络中没有这些依赖关系，因此各个路径可以在经过前馈神经网络层的时候并行计算。

接下来我们用一个短句（Thinking Machine）作为例子，看看在编码器的每个子层中发生了什么。

编码器

上边我们已经说了，每个编码器组件接受一组向量作为输入。在其内部，输入向量先通过一个自注意力层，再经过一个前馈神经网络，最后将其将输出给下一个编码器组件。

不同位置上的单词都要经过自注意力层的处理，之后都会经过一个结构完全相同的前馈神经网络。

自注意力

不要一看“self-attention”就觉得这是个每个人都很很熟悉的词，其实我个人感觉，在看《Attention is all you need》之前我都没有真正理解自注意力机制。现在让我们看一下自注意力机制。

假设我们要翻译下边这句话：

“The animal didn't cross the street because it was too tired”

这里it指的是什么？是street还是animal？人理解起来很容易，但是用算法处理就不那么容易了。

当模型处理it这个词的时候，自注意力会让it和animal关联起来。

当模型编码每个位置上的单词的时候，自注意力的作用就是：看一看输入句子中其他位置的单词，试图寻找一种对当前单词更好的编码方式。

如果你熟悉RNNs模型，回想一下RNN如何处理当前时间步的隐藏状态：将之前的隐藏状态与当前位置的输入结合起来。
在Transformer中，自注意力机制也可以将其他相关单词的“理解”融入到我们当前处理的单词中。

当我们在最后一个编码器组件中对it进行编码的时候，注意力机制会更关注The animal，并将其融入到it的编码中。

可以去Tensor2Tensor ，自己体验一下上图的可视化。

细说自注意力机制

先画图解释一下自注意力是怎么算的，之后再看一下实际实现中是怎么用矩阵计算。

第一步 对编码器的每个输入向量都计算三个向量，就是对每个输入向量都算一个query、key、value向量。
怎么算的？
把输入的词嵌入向量与三个权重矩阵相乘。权重矩阵是模型训练阶段训练出来的。

在同一个编码器组件中，一组输入向量使用相同的权重矩阵，就是所有输入都用一样的$W^Q、W^K、W^V$）

注意，这三个向量维度是64，比嵌入向量的维度小，嵌入向量、编码器的输入输出维度都是512。这三个向量不是必须比编码器输入输出的维数小，这样做主要是为了让多头注意力的计算更稳定。

将 $x_{1}$ 和 ${W}^{Q}$ 权重矩阵相乘得到 ${q}_{1}$, 就得到与单词${x}_{1}$ 相关的query。
按这样的方法，最终我们给输入的每一个单词都计算出一个“query”、一个 “key"和一个 “value"。

什么是 “query”、“key”、“value” 向量？

这三个向量是计算注意力时的抽象概念，继续往下看注意力计算过程，看完了就懂了。

第二步 计算注意力得分。

假设我们现在在计算输入中第一个单词Thinking。我们需要使用自注意力给输入句子中的每个单词打分，这个分数决定当我们编码某个位置的单词的时候，应该对其他位置上的单词给予多少关注度。

这个得分是query和key的点乘积得出来的。

举个栗子，我们要算第一个位置的注意力得分的时候就要将第一个单词的query和其他的key依次相乘，在这里就是$q_1·k_1$,$q_1·k_2$。

第三步 将计算获得的注意力分数除以8。

为什么选8？是因为key向量的维度是64，取其平方根，这样让梯度计算的时候更稳定。默认是这么设置的，当然也可以用其他值。

第四步 除8之后将结果扔进softmax计算，使结果归一化，softmax之后注意力分数相加等于1，并且都是正数。

这个softmax之后的注意力分数表示在计算当前位置的时候，其他单词受到的关注度的大小。显然在当前位置的单词肯定有一个高分，但是有时候也会注意到与当前单词相关的其他词汇。

这一步中我们会得到一个句子中不同单词对于Thinking的注意力分数。

第五步 将每个value向量乘以注意力分数。
这是为了留下我们想要关注的单词的value，并把其他不相关的单词丢掉。

第六步 将上一步的结果相加，输出本位置的注意力结果。

第一个单词的注意力结果就是$z_1$

这就是自注意力的计算。计算得到的向量直接传递给前馈神经网络。但是为了处理的更迅速，实际是用矩阵进行计算的。接下来我们看一下怎么用矩阵计算。

用矩阵计算self-attention

计算Query, Key, Value矩阵。直接把输入的向量打包成一个矩阵$X$，再把它乘以训练好的$W^Q$、$W^K$、$W^V$。

X矩阵每一行都代表输入句子中的一个词，整个矩阵代表输入的句子。

$X$矩阵中的一行相当于输入句子中的一个单词。
我们看一下维度的差异：原文中嵌入矩阵的长度为 512 , $q、k、v$ 矩阵的长度为 64 ；
在这里我们分别用 4 个格子表示和3个格子表示。

因为我们现在用矩阵处理，所以可以直接将之前的第二步到第六步压缩到一个公式中一步到位获得最终的注意力结果$Z$。

补充一个有趣的知识。注意力的计算方法不止这一种，常见的有点积注意力、加性注意力等。但是论文中用的是点积注意力。作者的理由是：“简单”。虽然二者难度是差不多的，但是点积注意力计算起来更快，代码写起来更方便。

多头注意力

论文进一步改进了自注意力层，增加了一个机制，也就是多头注意力机制。这样做有两个好处：

1. 它扩展了模型专注于不同位置的能力。

   在上面例子里只计算一个自注意力的的例子中，编码Thinking的时候，虽然最后$Z_1$或多或少包含了其他位置单词的信息，但是它实际编码中还是被Thinking单词本身所支配。

   如果我们翻译一个句子，比如The animal didn’t cross the street because it was too tired，我们会想知道“it”指的是哪个词，这时模型的“多头”注意力机制会起到作用。

   我个人理解：每个词最终的编码都是主要受该单词本身的影响，虽然关注到其他位置的词，但是它的关注点可能和你想要的点不一样。但是你有多head的时候就好办了，一个词一共才多少种含义，总能有一个关注到我想的方面吧。作者举it这个例子，如果你现在感觉看的有点迷惑没关系，看完这一小节最后那两张多头自注意力的可视化就理解了。

2. 它给了注意层多个“表示子空间”。

   就是在多头注意力中同时用多个不同的$W^Q$、$W^K$、$W^V$权重矩阵(Transformer使用8个头部，因此我们最终会得到8个计算结果)，每个权重都是随机初始化的。经过训练每个$W^Q$、$W^K$、$W^V$都能将输入的矩阵投影到不同的表示子空间。

在多头注意力中, 我们给每个头单独的权重矩阵, 从而产生不同的Q、 K 、V 矩阵。

Transformer中的一个多头注意力（有8个head）的计算，就相当于用自注意力做8次不同的计算，并得到8个不同的结果$Z$。

但是这会存在一点问题，多头注意力出来的结果会进入一个前馈神经网络，这个前馈神经网络可不能一下接收8个注意力矩阵，它的输入需要是单个矩阵（矩阵中每个行向量对应一个单词），所以我们需要一种方法把这8个合并成一个矩阵。

怎么做呢？我们将这些矩阵连接起来，然后将乘以一个附加的权重矩阵$W^O$。

以上就是多头自注意力的全部内容。让我们把多头注意力上述内容 放到一张图里看一下子：

现在我们已经看过什么是多头注意力了，让我们回顾一下之前的一个例子，再看一下编码“it”的时候每个头的关注点都在哪里：

编码it，用两个head的时候：其中一个更关注the animal，另一个更关注tired。 此时该模型对it的编码。除了it本身的表达之外，同时也包含了the animal和tired的相关信息

如果我们把所有的头的注意力都可视化一下，就是下图这样，但是看起来事情好像突然又复杂了。

使用位置编码表示序列的位置

强烈安利一个详细解释位置编码的文章：Transformer的位置编码详解

到现在我们还没提到过如何表示输入序列中词汇的位置。

Transformer在每个输入的嵌入向量中添加了位置向量。这些位置向量遵循某些特定的模式，这有助于模型确定每个单词的位置或不同单词之间的距离。将这些值添加到嵌入矩阵中，一旦它们被投射到Q、K、V中，就可以在计算点积注意力时提供有意义的距离信息。

为了让模型能知道单词的顺序，我们添加了位置编码，位置编码是遵循某些特定模式的。

位置编码向量和嵌入向量的维度是一样的，比如下边都是四个格子：

举个例子，当嵌入向量的长度为4的时候，位置编码长度也是4

一直说位置向量遵循某个模式，这个模式到底是什么。

在下面的图中，每一行对应一个位置编码。所以第一行就是我们输入序列中第一个单词的位置编码，之后我们要把它加到词嵌入向量上。

看个可视化的图：

这里表示的是一个句子有20个词，词嵌入向量的长度为512。 可以看到图像从中间一分为二，因为左半部分是由正弦函数生成的。右半部分由余弦函数生成。 然后将它们二者拼接起来，形成了每个位置的位置编码。

你可以在get_timing_signal_1d()中看到生成位置编码的代码。
这不是位置编码的唯一方法。但是使用正余弦编码有诸多好处，具体可以看这里：Transforme 结构：位置编码详解

但是需要注意注意一点，上图的可视化是官方Tensor2Tensor库中的实现方法，将sin和cos拼接起来。但是和论文原文写的不一样，论文原文的3.5节写了位置编码的公式，论文不是将两个函数concat起来，而是将sin和cos交替使用。论文中公式的写法可以看这个代码：transformer_positional_encoding_graph，其可视化结果如下：

这里表示的是一个句子有10个词，词嵌入向量的长度为64。

残差

在继续往下讲之前，我们还需再提一下编码器中的一个细节：每个编码器中的每个子层（自注意力层、前馈神经网络）都有一个残差连接，之后是做了一个层归一化（layer-normalization）。

将过程中的向量相加和层归一化可视化如下所示：

当然在解码器子层中也是这样的。

我们现在画一个有两个编码器和解码器的Transformer，那就是下图这样的：

解码器

现在我们已经介绍了编码器的大部分概念，（因为编码器的解码器组件差不多）我们基本上也知道了解码器的组件是如何工作的。那让我们直接看看二者是如何协同工作的。

编码器首先处理输入序列，将最后一个编码器组件的输出转换为一组注意向量K和V。每个解码器组件将在“encoder-decoder attention”层中使用编码器传过来的K和V，这有助于解码器将注意力集中在输入序列中的适当位置：

完成编码阶段后，我们开始进行解码阶段。
在解码阶段每一轮计算都只往外蹦一个输出，在本例中是输出一个翻译之后的英语单词。

解码器不是咔咔咔一个句子一下给你输出出来，是每次只输出一个词！！！

输出步骤会一直重复，直到遇到句子结束符表明transformer的解码器已完成输出。

每一步的输出都会在下一个时间步喂给底部解码器，解码器会像编码器一样运算并输出结果（每次往外蹦一个词）。

跟编码器一样，在解码器中我们也为其添加位置编码，以指示每个单词的位置。

解码器中的自注意力层和编码器中的不太一样：

在解码器中，自注意力层只允许关注已输出位置的信息。实现方法是在自注意力层的softmax之前进行mask，将未输出位置的信息设为极小值。

“encoder-decoder attention”层的工作原理和前边的多头自注意力差不多，但是Q、K、V的来源不用，Q是从下层创建的（比如解码器的输入和下层解码器组件的输出），但是其K和V是来自编码器最后一个组件的输出结果。

编码器是对整个输入序列进行编码嘛，然后将其结果转化成K和V传给解码器，这个K、V包含整个句子的所有信息。但是解码器的输入是什么？是拼接之前解码器的输出的单词。所以解码器造出来的Q仅包含已经输出的内容。

最后的线性层和softmax层

Decoder输出的是一个浮点型向量，如何把它变成一个词？

这就是最后一个线性层和softmax要做的事情。

线性层就是一个简单的全连接神经网络，它将解码器生成的向量映射到logits向量中。
假设我们的模型词汇表是10000个英语单词，它们是从训练数据集中学习的。那logits向量维数也是10000，每一维对应一个单词的分数。

然后，softmax层将这些分数转化为概率（全部为正值，加起来等于1.0），选择其中概率最大的位置的词汇作为当前时间步的输出。

这张图从下往上看，假设具体上的那个向量是解码器的输出，然后将其转换为最终输出的单词。

训练过程概述

现在我们已经了解了Transformer的整个前向传播的过程，那我们继续看一下训练过程。
在训练期间，未经训练的模型会进行相同的前向传播过程。由于我们是在有标记的训练数据集上训练它，所以我们可以将其输出与实际的输出进行比较。

为了便于理解，我们假设预处理阶段得到的词汇表只包含六个单词（“a”, “am”, “i”, “thanks”, “student”, “<eos>”）。

注意这个词汇表是在预处理阶段就创建的，在训练之前就已经得到了。

一旦我们定义好了词汇表，我们就可以使用长度相同的向量（独热码，one-hot 向量）来表示词汇表中的每个单词。例如，我们可以用以下向量表示单词“am”：

接下来让我们讨论一下模型的损失函数，损失函数是我们在训练阶段优化模型的指标，通过损失函数，可以帮助我们获得一个准确的、我们想要的模型。

损失函数

假设我们正要训练我们的模型。

假设现在是训练阶段的第一步，我们用一个简单的例子（一个句子就一个词）来训练模型：把“merci” 翻译成 “thanks”。
这意味着，我们希望输出是表示“谢谢”的概率分布。但由于这个模型还没有经过训练，所以目前还不太可能实现。

由于模型的参数都是随机初始化的，未经训练的模型为每个单词生成任意的概率分布。 我们可以将其与实际输出进行比较，然后使用反向传播调整模型的权重，使输出更接近我们所需要的值。

如何比较两种概率分布？在这个例子中我们只是将二者相减。实际应用中的损失函数请查看交叉熵损失和Kullback–Leibler散度。

上述只是最最简单的一个例子。现在我们来使用一个短句子（一个词的句子升级到三三个词的句子了），比如输入 “je suis étudiant”预期的翻译结果为： “i am a student”。

所以我们希望模型不是一次输出一个词的概率分布了，能不能连续输出概率分布，最好满足下边要求：

• 每个概率分布向量长度都和词汇表长度一样。我们的例子中词汇表长度是6，实际操作中一般是30000或50000。
• 在我们的例子中第一个概率分布应该在与单词“i”相关的位置上具有最高的概率
• 第二种概率分布在与单词“am”相关的单元处具有最高的概率
• 以此类推，直到最后输出分布指示“<eos>”符号。除了单词本身之外，单词表中也应该包含诸如“<eos>”的信息，这样softmax之后指向“<eos>”位置，标志解码器输出结束。

对应上面的单词表，我们可以看出这里的one-hot向量是我们训练之后想要达到的目标。

在足够大的数据集上训练模型足够长的时间后，我们希望生成的概率分布如下所示：

这个是我们训练之后最终得到的结果。当然这个概率并不能表明某个词是否是训练集之中的词汇。 在这里你可以看到softmax的一个特性，就是即使其他单词并不是本时间步的输出， 也会有一丁点的概率存在，这一特性有助于帮助模型进行训练。

模型一次产生一个输出，在这么多候选中我们如何获得我们想要的输出呢？现在有两种处理结果的方法：

一种是贪心算法（greedy decoding）：模型每次都选择分布概率最高的位置，输出其对应的单词。

另一种方法是束搜索（beam search）：保留概率最高前两个单词（例如，“I”和“a”），然后在下一步继续选择两个概率最高的值，以此类推，在这里我们把束搜索的宽度设置为2，当然你也可以设置其他的束搜索宽度。

更多内容

如果你想更深入了解Transformer：

• 阅读论文原文Attention Is All You Need、 谷歌相关博客Transformer: A Novel Neural Network Architecture for Language Understanding和Tensor2Tensor announcement。

• 观看Łukasz Kaiser的授课视频。

• 自己尝试一下代码：Jupyter Notebook provided as part of the Tensor2Tensor repo

• 探究其源码GitHub - tensorflow/tensor2tensor: Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.