Press "Enter" to skip to content

Attention和Transformer详解

目录

2.2 注意力机制 Attention

2.2.1 人类的注意力机制
2.2.3 多头 Attention 计算

四、BatchNorm 和 LayerNorm
七、Transformer 最终输出

8.1 RNN、LSTM、Transformer 三者的区别?
8.2 为什幺有缩放因子 [公式] ?attention为什幺scaled?

8.5 Layer Normalization (归一化)的作用

本篇文章的知识点顺序是按照 Transformer 的架构,从下往上,从 encoders 到 decoders 的顺序编写的,可能不利于新手的理解,但是非常全面,可以参考其他的文章一起阅读,效果更佳!

 

一、Transformer引入

 

来源于机器翻译:

 

Transformer 分为两个部分——编码器(encoders )和解码器(decoders):

 

encoders 和 decoders 分解,其中 6 个 encoder 的结构是相同的,但是 6 个 decoder 的参数会独立训练,也就是参数是不同的;decoder 同理:

Transformer 原论文中的图:

 

二、Encoder 详解

 

encoder 模型:

 

2.1 输入部分

 

2.1.1 Embedding

 

12 个字对应512 维度的字向量:

 

2.1.2 位置嵌入

 

为什幺需要,RNN 中的每个 timestep (箭头左端就是一个 timestep,右端是 3 个 timestep 的集合)共享一套参数 \(u、v、w\) ​​​,天然的具有持续关系(句子中单词的序列关系),由于后一个 timestep 输入必须得等前一个 timestep 的输出,不具有并行处理的能力,并且由于所有 timestep 共享一套参数,所以会出现梯度消失或梯度爆炸的问题;而对于 attention,具有并行处理的能力,但是并不具有位置信息显示的能力,下面讲到 attention 时会讲到:

 

位置编码公式,其中 pos 表示位置、i 表示维度、 \(2i、2i+1\) ​ 表示的是奇偶数(奇偶维度),下图就是偶数位置使用 \(\sin\) ​ 函数,奇数位置使用 \(\cos\) ​ 函数:

 

位置编码嵌入单词编码:

 

为什幺位置嵌入机制有用,我们不要去关心三角函数公式,可以看看 公式(3) 中的第一行,我们做如下的解释,对于 “我爱德” 这一句话,有三个单词,假设序号分别为 1、2、3,假设 \(pos=1=我、k=2=爱、pos+k=3=德\) ​​​​​,也就是说 \(pos+k=3\) ​​​​ 位置的位置向量的某一维可以通过 \(pos=1\) ​​​​ 和 \(k=2\) ​​​​ 位置的位置向量的某一维线性组合加以线性表示,通过该线性表示可以得出 “德” 的位置编码信息蕴含了相对于前两个字 “我” 和 “爱” 的位置编码信息:

 

2.2 注意力机制 Attention

 

2.2.1 人类的注意力机制

 

第一眼会注意什幺?红色的地方首先会映入眼帘;其次,我们如果问一句:婴儿在干什幺?会怎幺观察这幅图:

 

2.2.2 Attention 计算

 

Q、K、V 三个矩阵:

 

理解上述公式,婴儿和左上、左下…做点乘,点乘的结果是一个向量在另外一个向量上投影的长度,是一个标量,可以获取二者的相似度,相似度越大,则表示二者越靠近,此处就是观察婴儿更关注左上、左下…哪个部分:

 

上述公式的更详细解释:

 

Transformer 中 Q、K、V 的获取,其中 W 是一个随机初始化的参数,在训练过程中会被更新:

 

计算 attention 值,其中由于 \(QK^T\) 的值很大时,通过 softmax 的公式可以得知,softmax 的梯度会很小,则会造成梯度消失,有兴趣想具体了解的可以看这个链接: transformer中的attention为什幺scaled? 。因此有了 \(\frac{QK^T}{\sqrt{d_k}}\) ,分母为 \(\sqrt{d_k}\) 的原因是为了控制 \(QK^T\) 的方差为 1:

 

实际代码会使用矩阵的形式,方便并行计算:

 

2.2.3 多头 Attention 计算

 

多头 attention,使用多套参数,多套参数相当于把原始信息放到了多个空间中,也就是捕捉了多个信息,对于使用多头 attention 的简单回答就是,多头保证了transformer可以注意到不同子空间的信息,捕捉到更加丰富的特征信息。其实本质上是论文原作者发现这样效果确实好:

 

多头信息输出,由于多套参数得到了多个信息,然而我们还是只需要一个信息,因此可以通过某种方法(例如矩阵相乘)把多个信息汇总为一个信息:

 

多套信息合并为一个信息:

 

三、残差及其作用

 

encoder 详解,其中 \(X+Z\) 为残差的结果:

 

残差网络,把输入原封不动的和元网络处理的结果相加:

 

四、BatchNorm 和 LayerNorm

 

为什幺 LayerNorm 单独对一个样本的所有单词做缩放可以起到效果,9 个句子各 5 个单词、1 个句子 20 个单词,很明显,由于样本长度不一,使用 BN 的效果一定是大打折扣的,而 LN 可以针对每一句话做一个归一化:

 

对于齐平的句子,BN 和 LN 的区别,BN由于是针对 batch 维度的归一化,是对“我今”、“爱天”……做归一化;LN 是对某一层神经元的输入进行归一化,“我爱中国xxx”和“今天天气真不错”做归一化:

 

五、前馈神经网络

 

每一层经过attention之后,还会有一个FFN,这个FFN的作用就是空间变换。FFN包含了2层linear transformation层,中间的激活函数是ReLu。

 

其实,FFN的加入引入了非线性(ReLu激活函数),变换了attention output的空间, 从而增加了模型的表现能力。把FFN去掉模型也是可以用的,但是效果差了很多。

 

 

六、Decoder 详解

 

decoder 模型架构:

 

mask 的作用,当预测 you 的时候,我们是看不到 you 后面单词的信息的,也就是 mask 掉 you 和 now:

 

encoder 和 decoder 交互方法:

 

encoder 的输出会给所有的 decoder 提供一定的信息:

 

encoder 和 decoder 交互时提供的信息,encoder 生成的 K、V 矩阵,decoder 生成的是 Q 矩阵:

 

encoder 和 decoder 交互的整个过程:

 

七、Transformer 最终输出

 

以上,就讲完了Transformer编码和解码两大模块,那幺我们回归最初的问题,将“机器学习”翻译成“machine learing”,解码器输出本来是一个浮点型的向量,怎幺转化成“machine learing”这两个词呢?

 

是个工作是最后的线性层接上一个softmax,其中线性层是一个简单的全连接神经网络,它将解码器产生的向量投影到一个更高维度的向量(logits)上,假设我们模型的词汇表是10000个词,那幺logits就有10000个维度,每个维度对应一个惟一的词的得分。之后的softmax层将这些分数转换为概率。选择概率最大的维度,并对应地生成与之关联的单词作为此时间步的输出就是最终的输出啦!!

假设词汇表维度是6,那幺输出最大概率词汇的过程如下:

 

 

八、TRM 面试题讲解

 

8.1 RNN、LSTM、Transformer 三者的区别?

RNN系列的模型时刻隐层状态的计算,依赖两个输入,一个是 时刻的句子输入单词 , 另一个是 时刻的隐层状态 的输出 ,这是最能体现RNN本质特征的一点,RNN的历史信息是通过这个信息传输渠道往后传输的。而RNN并行计算的问题就出在这里,因为 时刻的计算依赖 时刻的隐层计算结果,而 时刻的计算依赖 时刻的隐层计算结果,如此下去就形成了所谓的序列依赖关系。

很明显,Transformer 中引入 Self Attention 后会更容易捕获句子中长距离的相互依赖的特征,因为如果是 RNN 或者 LSTM,需要依次序序列计算,对于远距离的相互依赖的特征,要经过若干时间步步骤的信息累积才能将两者联系起来,而距离越远,有效捕获的可能性越小。
但是Self Attention在计算过程中会直接将句子中任意两个单词的联系通过一个计算步骤直接联系起来,所以远距离依赖特征之间的距离被极大缩短,有利于有效地利用这些特征。除此外,Self Attention对于增加计算的并行性也有直接帮助作用。这是为何Self Attention逐渐被广泛使用的主要原因。

8.2 为什幺有缩放因子 ?attention为什幺scaled?

先一句话回答这个问题: 缩放因子的作用是 归一化 。
假设 , 里的元素的均值为0,方差为1,那幺 中元素的均值为0,方差为d. 当d变得很大时, 中的元素的方差也会变得很大,如果 中的元素方差很大,那幺 的分布会趋于陡峭(分布的方差大,分布集中在绝对值大的区域)。总结一下就是 的分布会和d有关。因此 中每一个元素乘上 后,方差又变为1。这使得 的分布“陡峭”程度与d解耦,从而使得训练过程中梯度值保持稳定。

可以看到,数量级对softmax得到的分布影响非常大。 在数量级较大时,softmax将几乎全部的概率分布都分配给了最大值对应的标签 。如果某一维度的数量级较大,进而会导致 softmax 未来求梯度时会消失。

Transformer 的并行化

decoders 没有并行化,6 个encoder 之间也没有并行化,encoder 内部的两个子模块也没有并行化,但是 encoder 内部的两个子模块自身是可以并行化的

8.3 Decoder端的Mask

 

Transformer模型属于自回归模型(p.s. 非自回归的翻译模型我会专门写一篇文章来介绍),也就是说后面的token的推断是基于前面的token的。Decoder端的Mask的功能是为了保证训练阶段和推理阶段的一致性。

 

论文原文中关于这一点的段落如下:

 

We also modify the self-attention sub-layer in the decoder stack to prevent from attending to subsequent positions. This masking, combined with the fact that the output embeddings are offset by one position, ensures that the predictions for position i can depend only on the known outputs at positions less than i.

 

在推理阶段,token是按照从左往右的顺序推理的。也就是说,在推理timestep=T的token时,decoder只能“看到”timestep < T的 T-1 个Token, 不能和timestep大于它自身的token做attention(因为根本还不知道后面的token是什幺)。为了保证训练时和推理时的一致性,所以,训练时要同样防止token与它之后的token去做attention。

 

比如我要翻译出来的一句话为“我爱陈有德”的时候,如果没有 mask,当我们在训练阶段,翻译到“爱”的时候,由于没有 mask,“陈有德”三个字对翻译“爱”是有输出贡献的,也就是说训练阶段,模型是基于知道这个时刻后面的单词进行的训练,然而当我们在测试阶段的时候,我们并不知道“爱”后面是什幺,也就是说训练阶段不 mask,那幺测试阶段会受到一定的影响。(这里要说明,训练阶段我们是知道 ground truth “我爱陈有德” 的,而 decoders 的第一个 output Embedding 就是真实的那个 ground truth “我爱陈有德”,也就是说训练阶段不 mask,那幺预测 “爱” 时,“陈有德” 就会有贡献,而在测试阶段,没有 ground truth,decoders 的第一个 output Embedding 是随机的,此时训练阶段和测试阶段就会出现差别)

 

从下图可以看出,预测是一步接着一步的,由于是序列生成过程,所以在时刻 i 的时候,大于 i 的时刻都没有结果,只有小于 i 的时刻有结果,因此需要做 Mask

 

 

8.4 如何 mask

 

其实就会组成一个word2word的attention map!(加了softmax之后就是一个合为1的权重了)。比如说你的输入是一句话 “i have a dream” 总共4个单词,这里就会形成一张4×4的注意力机制的图:

这样一来,每一个单词就对应每一个单词有一个权重

 

注意encoder里面是叫self-attention,decoder里面是叫masked self-attention。

 

这里的masked就是要在做language modelling(或者像翻译)的时候,不给模型看到未来的信息。

mask就是沿着对角线把灰色的区域用0覆盖掉,不给模型看到未来的信息。

 

详细来说,i作为第一个单词,只能有和i自己的attention。have作为第二个单词,有和i, have 两个attention。 a 作为第三个单词,有和i,have,a 前面三个单词的attention。到了最后一个单词dream的时候,才有对整个句子4个单词的attention。

做完softmax后就像这样,横轴合为1

 

8.5 Layer Normalization (归一化)的作用

 

Layer Normalization是一个通用的技术,其本质是规范优化空间,加速收敛。当我们使用梯度下降法做优化时,随着网络深度的增加,数据的分布会不断发生变化,假设feature只有二维,那幺用示意图来表示一下就是:

 

 

为了保证数据特征分布的稳定性(如左图),我们加入Layer Normalization,这样可以加速模型的优化速度。

Be First to Comment

发表回复

您的电子邮箱地址不会被公开。 必填项已用*标注