1. Introduction

2. 模型结构

2.1 Transformer模型

2.2 输入层

2.3 位置向量：给单词赋予上下文语境

2.4 注意力机制

2.4.1 Self-attention 和 Encoder-Decoder attention

2.4.2 Scaled Dot-Product Attention

2.4.3 Multi-Head Attention

2.5 前向神经网络

3. 本文使用的注意力机制

3.1 Encoder的结构

3.2 Decoder的结构

4. 总结

4.1 为什么使用self-attention

4.2 训练

4.3 结果

5. 本文开源代码

6. 参考资料

1. Introduction

本文是谷歌发表的文章，针对nlp里的机器翻译问题，提出了一种被称为”Transformer”的网络结构，基于注意力机制。文章提出，以往nlp里大量使用RNN结构和encoder-decoder结构，RNN及其衍生网络的缺点就是慢，问题在于前后隐藏状态的依赖性，无法实现并行，而文章提出的”Transformer”完全摒弃了递归结构，依赖注意力机制，挖掘输入和输出之间的关系，这样做最大的好处是能够并行计算了。
在此之前，针对机器翻译这个领域，为了应对RNN无法并行问题，已经有过一些使用CNN的解决方案了，例如谷歌的ByteNet，Facebook的FairSeq等等。
自注意力机制（Self-attention）能够把输入序列上不同位置的信息联系起来，然后计算出整条序列的某种表达，目前自注意力机制主要应用于阅读理解、提取摘要、文本推论等领域。

2. 模型结构

上图是谷歌提出的 transformer 架构，其本质上是一个Encoder-Decoder的结构。把英文句子输入模型，模型会输出法文句子。

要搭建Transformer，我们必须要了解5个过程:

词向量层
位置编码
创建Masks
多头注意层（The Multi-Head Attention layer）
Feed Forward层

2.1 Transformer模型

Transformer采用encoder-decoder结构，舍弃了RNN的循环式网络结构，完全基于注意力机制来对一段文本进行建模。
Transformer所使用的注意力机制的核心思想是去计算一句话中的每个词对于这句话中所有词的相互关系，然后认为这些词与词之间的相互关系在一定程度上反应了这句话中不同词之间的关联性以及重要程度。因此再利用这些相互关系来调整每个词的重要性（权重）就可以获得每个词新的表达。这个新的表征不但蕴含了该词本身，还蕴含了其他词与这个词的关系，因此和单纯的词向量相比是一个更加全局的表达。Transformer通过对输入的文本不断进行这样的注意力机制层和普通的非线性层交叠来得到最终的文本表达。

Transformer采用Encoder与Decoder的堆叠，具体为

Encoder：Transformer模型的Encoder由6个基本层堆叠起来，每个基本层包含两个子层，第一个子层是一个注意力机制，第二个是一个全连接前向神经网络。对两个子层都引入了残差边以及layer normalization。
Decoder：Transformer模型的Decoder也由6个基本层堆叠起来，每个基本层除了Encoder里面的那两个以外，还增加了一层注意力机制，同样引入残差边以及layer normalization。
Encoder的attention是self-attention；Decoder里面的attention首先是self-attention，然后是encoder-decoder attention。

2.2 输入层

编码器和解码器的输入就是利用学习好的embeddings将tokens（一般应该是词或者字符）转化为d维向量。对解码器来说，利用线性变换以及softmax函数将解码的输出转化为一个预测下一个token的概率。

2.3 位置向量：给单词赋予上下文语境

由于本文的模型结构没有使用任何递归结构或卷积结构，为了让模型能利用输入序列的顺序信息，必须引入某种能表达输入序列每个部分的绝对或相对位置的信息才行，也就是需要将tokens的相对以及绝对位置信息注入到模型中去。文章采取的方法是在输入embeddings的基础上加了一个位置编码（positional encoding），在送入encoder和decoder之前，先对输入进行编码，编码后的向量维度是dmodel。具体来说，采用正弦和余弦函数进行编码。
然而，只要稍微思考一下就会发现，这样的模型并不能捕捉序列的顺序！换句话说，如果将K、V按行打乱顺序（相当于句子中的词序打乱），那么Attention的结果还是一样的。这就表明了，到目前为止，Attention模型顶多是一个非常精妙的“词袋模型”而已。这问题就比较严重了，大家知道，对于时间序列来说，尤其是对于NLP中的任务来说，顺序是很重要的信息，它代表着局部甚至是全局的结构，学习不到顺序信息，那么效果将会大打折扣（比如机器翻译中，有可能只把每个词都翻译出来了，但是不能组织成合理的句子）。
于是Google再祭出了一招——Position Embedding，也就是“位置向量”，将每个位置编号，然后每个编号对应一个向量，通过结合位置向量和词向量，就给每个词都引入了一定的位置信息，这样Attention就可以分辨出不同位置的词了。

Position Embedding并不算新鲜的玩意，在FaceBook的《Convolutional Sequence to Sequence Learning》也用到了这个东西。

（1）以前在RNN、CNN模型中其实都出现过Position Embedding，但在那些模型中，Position Embedding是锦上添花的辅助手段，也就是“有它会更好、没它也就差一点点”的情况，因为RNN、CNN本身就能捕捉到位置信息。但是在这个纯Attention模型中，Position Embedding是位置信息的唯一来源，因此它是模型的核心成分之一，并非仅仅是简单的辅助手段。

（2）在以往的Position Embedding中，基本都是根据任务训练出来的向量。而Google直接给出了一个构造Position Embedding的公式：

其中pos代表的是第几个词，i代表embedding中的第几维。Google在论文中说到他们比较过直接训练出来的位置向量和上述公式计算出来的位置向量，效果是接近的。因此显然我们更乐意使用公式构造的Position Embedding了。

在给词向量添加位置编码之前，我们要扩大词向量的数值，目的是让位置编码相对较小。这意味着向词向量添加位置编码时，词向量的原始含义不会丢失。

（3）Position Embedding本身是一个绝对位置的信息，但在语言中，相对位置也很重要，Google选择前述的位置向量公式的一个重要原因是：由于我们有sin(α+β)=sinαcosβ+cosαsinβ以及cos(α+β)=cosαcosβ−sinαsinβ，这表明位置p+k的向量可以表示成位置p的向量的线性变换，这提供了表达相对位置信息的可能性。

2.4 注意力机制

注意力机制（Attention）简单来说就是给定一个查找（query）和一个键值表（key-value pairs)，将query映射到正确的输入的过程。此处的query、key、value和最终的输出都是向量。输出往往是一个加权求和的形式，而权重则由query、key和value决定。
NLP领域中，Attention网络基本成为了标配，是Seq2Seq的创新。Attention网络是为了解决编码器-解码器结构存在的长输入序列问题。
Attention功能可以被描述为将查询和一组键值对映射到输出，其中查询、键、值和输出都是向量。输出可以通过对查询的值加权来计算。

Attention通常可以进行如下描述，表示为将query(Q)和key-value pairs映射到输出上，其中query、每个key、每个value都是向量，输出是V中所有values的加权，其中权重是由Query和每个key计算出来的，计算方法分为三步：

1）计算比较Q和K的相似度，用f来表示：

2）将得到的相似度进行softmax归一化：

3）针对计算出来的权重，对所有的values进行加权求和，得到Attention向量：

计算相似度的方法有以下4种，在本文中，我们计算相似度的方式是第一种：

本文提出的Attention机制称为Multi-Head Attention，不过在这之前，我们要先介绍它的简单版本 Scaled Dot-Product Attention。

2.4.1 Self-attention 和 Encoder-Decoder attention

计算Attention首先要有query、key和value。Encoder的attention是self-attention，Decoder里面的attention首先是self-attention，然后是encoder-decoder attention。这里的两种attention是针对query和key-value来说的：

对self-attention来说，计算得到query和key-value的过程都是使用的同样的输入，因为要算自己跟自己的attention嘛；
对encoder-decoder attention来说，query的计算使用的是decoder的输入，而key-value的计算使用的是encoder的输出，因为我们要计算decoder的输入跟encoder里面每一个的相似度嘛。经典模型：

因此本文下面对于attention的讲解，都是基于self-attention来说的，如果是encoder-decoder attention，只要改一下输入即可，其余过程都是一样的。

2.4.2 Scaled Dot-Product Attention

Scaled Dot-Product Attention 可以看作是只有一个 Head 的 Multi-Head Attention，是简化版本。

输入包含dk维的query和key，以及dv维的value。通过计算query和各个key的点积，除以√dk归一化，然后经过softmax激活变成权重，最后再乘value。点积注意力机制的优点是速度快、占用空间小。

图解参考：https://www.jianshu.com/p/b1030350aadb

例子：

Embedding在进入到Attention之前，有3个分叉，那表示说从1个向量，变成了3个向量Q、K、V，它是通过定义一个 $W^{Q}$ 矩阵（这个矩阵随机初始化，通过前向反馈网络训练得到），将Embedding和 $W^{Q}$ 矩阵做乘法，得到查询向量q，假设输入Embedding是512维，在上图中我们用4个小方格表示，输出的查询向量是64维，上图中用3个小方格以示不同。然后类似地，定义 $W^{K}$ 和 $W^{V}$ 矩阵，将Embedding和 $W^{K}$ 做矩阵乘法，得到键向量k；将embeding和 $W^{V}$ 做矩阵乘法，得到值向量v。对每一个Embedding做同样的操作，那么每个输入就得到了3个向量，查询向量、键向量和值向量。需要注意的是，查询向量和键向量要有相同的维度，值向量的维度可以相同，也可以不同，但一般也是相同的。至于将获得的Q、K、V矩阵具体操作，总的来说，就是以下这幅图：

获得的Z和目标值进行比较，获得的损失反向传播，优化的参数是， $W^{Q}$ 、 $W^{K}$ 和 $W^{V}$ 。

2.4.3 Multi-Head Attention

Multi-Head Attention就是把Scaled Dot-Product Attention的过程做H次，然后把输出拼接起来，再经过一次的线性操作，论文中，它的结构图如下：

不同的head可以学到不同的语义关系，如一组head用于做指代消解，一组head用于做上下文。

用h个不同的线性变换（本文取8）分别将dmodel维的key、value和query映射成dk维、dk维和dv维，然后再代入注意力机制，产生总共h×dv维输出，然后拼起来，再用一个线性变换得到最终的输出。
这个是Google提出的新概念，是Attention机制的完善。不过从形式上看，它其实就再简单不过了，就是把Q、K、V通过参数矩阵映射一下，然后再做Attention，把这个过程重复做h次，结果拼接起来就行了，可谓“大道至简”了。具体来说：

所谓“多头”（Multi-Head），就是只多做几次同样的事情（参数不共享），然后把结果拼接。

2.5 前向神经网络

这是一个 Position-wise 前向神经网络，encoder和decoder的每一层都包含一个全连接前向神经网络，激活函数顺序是线性、RELU、线性。每层由两个支层，attention层就是其中一个，而attention之后的另一个支层就是一个前馈的网络。公式描述如下：

3. 本文使用的注意力机制

本文使用的是Multi-Head Attention，具体体现在三个方面。

在“encoder-decoder attention”层中，query来自前一个decoder层，而key和value是encoder的输出。这允许decoder的每个位置都去关注输入序列的所有位置。
encoder包含self-attention层，在self-attention层中所有的key、value和query都来自前一层的encoder。这样encoder的每个位置都能去关注前一层encoder输出的所有位置。
decoder包含self-attention层

可以看到，Encoder的每一层有两个操作，分别是Self-Attention和Feed Forward；而Decoder的每一层有三个操作，分别是Self-Attention、Encoder-Decoder Attention以及Feed Forward操作。这里的Self-Attention和Encoder-Decoder Attention都是用的是Multi-Head Attention机制。

3.1 Encoder的结构

Encoder有N层，默认是6层，每层包括两个sub-layers:
（1）第一个sub-layer是Self-attention：使用multi-head self-attention mechanism，用来计算输入的self-attention;
（2）第二个sub-layer是简单的全连接网络。
每一个sub-layer都模拟了残差网络的结构，其网络示意图如下：

3.2 Decoder的结构

Decoder有N(默认是6)层，每层包括三个sub-layers:
（1）第一个sub-layer是Self-Attention：使用 Masked multi-head self-attention，也是计算输入的self-attention，但是因为是生成过程，因此在时刻 i 的时候，大于 i 的时刻都没有结果，只有小于 i 的时刻有结果，因此需要做Mask.
（2）第二个sub-layer是Encoder-Decoder Attention：对encoder的输入进行attention计算，这里仍然是 Multi-head self-attention结构，只不过输入的分别是decoder的输入和encoder的输出。
（3）第三个sub-layer是全连接网络：与Encoder相同。

其网络示意图如下：

4. 总结

模型的整体框架基本介绍完了，其最重要的创新应该就是Self-Attention和Multi-Head Attention的架构。在摒弃传统CNN和RNN的情况下，还能提高表现，降低训练时间。Transformer用于机器翻译任务，表现极好，可并行化，并且大大减少训练时间。并且也给我们开拓了一个思路，在处理问题时可以增加一种结构的选择。

4.1 为什么使用self-attention

从三个方面去对比self-attention和递归结构、卷积结构的优劣性，首先是每一层的计算复杂度，其次是能够被并行的计算量，最后是网络中长期依赖的路径长度。对比显示，self-attention表现最好。

4.2 训练

训练数据使用WMT English-German数据集，包含450w对语句。句子都被编码过了，使用了一个大小约37000个token的字典。样本被分为若干个batch，每个batch大概25000个token，每个batch中的句子长度保持基本一致。硬件上使用了8块GPU。Optimizer使用了Adam。过拟合方面使用了dropout和Label Smoothing。

4.3 结果

不论是英语-德语还是英语-法语的翻译任务，对比之前的一些模型，本文提出的模型都达到更好的BELU值，同时Training Cost也最低。

5. 本文开源代码

https://github.com/tensorflow/tensor2tensor

attention实现keras版： https://github.com/bojone/attention/blob/master/attention_keras.py
attention实现tensorflow版： https://github.com/bojone/attention/blob/master/attention_tf.py

6. 参考资料

从Encoder-Decoder(Seq2Seq)理解Attention的本质：https://www.cnblogs.com/huangyc/p/10409626.html#_label3
《Attention is All You Need》浅读（简介+代码）https://kexue.fm/archives/4765
Attention Is All You Need https://arxiv.org/abs/1706.03762
一文读懂「Attention is All You Need」| 附代码实现 https://yq.aliyun.com/articles/342508
对Attention is all you need 的理解https://blog.csdn.net/mijiaoxiaosan/article/details/73251443

论文《Attention Is All You Need》及Transformer模型相关推荐

谷歌NIPS论文Transformer模型解读：只要Attention就够了
作者 | Sherwin Chen 译者 | Major,编辑 | 夕颜出品 | AI科技大本营(ID:rgznai100) 导读:在 NIPS 2017 上,谷歌的 Vaswani 等人提出了 T ...
【论文导读】DLP-KDD2019｜BST---使用Transformer进行序列推荐
前言本次分享一篇使用Transformer作为用户行为序列建模的CTR预估文章.论文较短,只有4页,感兴趣的同学可以阅读下原文[1]. 本文约1.7k字,预计阅读10分钟. 概要本文"B ...
【Transformer专题】一、Attention is All You Need（Transformer入门）
目录前言一.整体架构二.Transfomer输入 2.1.单词Embedding 2.2.位置Embedding 三.Self-Attention结构 3.1.Self-Attention QK ...
李沐论文精读系列一： ResNet、Transformer、GAN、BERT
文章目录一. ResNet 1.0 摘要,论文导读 1.1 导论 1.1.1 为什么提出残差结构? 1.1.2 实验验证 1.2 相关工作 1.3 实验部分 1.3.1 不同配置的ResNet结构 ...
transformer论文解读_一文读懂transformer(CV专享)
导言: transformer在这两年大火,CV界广泛应用transformer模型,充分吸收其idea到自己的领域.考虑到transformer本是应用于NLP的论文,其中很多内容都形成了共识,因此 ...
Paper：Transformer模型起源—2017年的Google机器翻译团队—《Transformer：Attention Is All You Need》翻译并解读
Paper:Transformer模型起源-2017年的Google机器翻译团队-<Transformer:Attention Is All You Need>翻译并解读目录 <T ...
深度学习：论文阅读：（ICLR-2021）Vision Transformer
这里写目录标题论文详情 transormer与cnn 预备知识- Self-Attention和Multi-Head Attention Self-Attention Multi-head atte ...
北航成AAAI 2021最大赢家，两篇一作斩获最佳论文、提名奖，研究皆与Transformer相关...
杨净发自凹非寺量子位报道 | 公众号 QbitAI 当地时间2月4日,AAAI 2021最佳论文奖出炉! 由AAAI 2021程序委员会主席Mausam在开幕式上宣布,共有3篇论文获得了最佳论 ...
【论文解读】A Survey on Visual Transformer及引文理解
A Survey on Visual Transformer阅读,以及自己对相关引文的理解. Transformer 作为NLP领域的大杀器,目前已经在CV领域逐渐展露锋芒,大有替代CNN的趋势,在图 ...

论文《Attention Is All You Need》及Transformer模型

1. Introduction

2. 模型结构

2.1 Transformer模型

2.2 输入层

2.3 位置向量：给单词赋予上下文语境

2.4 注意力机制

2.4.1 Self-attention 和 Encoder-Decoder attention

2.4.2 Scaled Dot-Product Attention

2.4.3 Multi-Head Attention

2.5 前向神经网络

3. 本文使用的注意力机制

3.1 Encoder的结构

3.2 Decoder的结构

4. 总结

4.1 为什么使用self-attention

4.2 训练

4.3 结果

5. 本文开源代码

6. 参考资料

论文《Attention Is All You Need》及Transformer模型相关推荐

最新文章

热门文章