BERT基础(二):Transformer 详解
Transformer 中 self - attention 的详解参考:
BERT基础(一):self_attention自注意力详解
在 Transformer 之前,多数基于神经网络的机器翻译方法依赖于循环神经网络(RNN),后者利用循环(即每一步的输出馈入下一步)进行顺序操作(例如,逐词地翻译句子)。尽管 RNN 在建模序列方面非常强大,但其序列性意味着该网络在训练时非常缓慢,因为长句需要的训练步骤更多,其循环结构也加大了训练难度。与基于 RNN 的方法相比,Transformer 不需要循环,而是并行处理序列中的所有单词或符号,同时利用自注意力机制将上下文与较远的单词结合起来。通过并行处理所有单词,并让每个单词在多个处理步骤中注意到句子中的其他单词。
下面就细致理解一下Transformer的模型结构,理解每一步在做什么。
0. 总体结构
- 先把Transformer想象成一个黑匣子,在机器翻译的领域中,这个黑匣子的功能就是输入一种语言然后将它翻译成其他语言。如下图:
- 这个黑匣子由2个部分组成,一个Encoders和一个Decoders。
- 对这个黑匣子进一步的剖析,发现每个Encoders中分别由6个Encoder组成(论文中是这样配置的)。而每个Decoders中同样也是由6个Decoder组成。
- 对于Encoders中的每一个Encoder,他们结构都是相同的,但是并不会共享权值。每层Encoder有2个部分组成,如下图:
Self-attention的输出会被传入一个全连接的前馈神经网络,每个encoder的前馈神经网络参数个数都是相同的,但是他们的作用是独立的。
- 每个Decoder也同样具有这样的层级结构,但是在这之间有一个Attention层,帮助Decoder专注于与输入句子中对应的那个单词。
1. Positional Encoding
使用Self-Attention方式进行编码时无法处理以下的顺序问题:
我从北京到上海
我从上海到北京
上述两个语句进行编码时,第一句的北京与上海两个词的编码与第二句的北京与上海两个词的编码会完全相同,但这两个语句中的两个地点代表着不同的含义:起始位置、终点位置。为了解决此问题,进行自编码时在输入层引入了一个位置编码,如下所示:
与输入的计算过程:拼接
2. self-attention
- Multi-head
3. Layer Normalization和残差连接
(1) 残差连接
我们在上一步得到了经过注意力矩阵加权后的V,也就是, 对它进行转置(图中的Z),使其和的维度一致,然后把它们加起来做残差连接,直接进行元素相加,因为他们的维度一致。
(2)LayerNorm
作用是把神经网络中隐藏层归一化为标准正态分布,以起到加快训练速度,加速收敛的作用:
- 上式中以矩阵的行(row)为单位求均值:
- 上式中以矩阵的行(row)为单位求方差:
- 然后用每一行的每一个元素减去这行的均值,再除以这行的标准差,从而得到归一化以后的数值:
式中的是为了防止除0;之后引入两个可训练参数 来弥补归一化过程中损失掉的信息。此处一般初始化为全1,而 为全0。
最后输出 :(X为输入词向量,为self-attention输出序列)
4. Feed Forward
两层线性映射并用**函数**,比如ReLU:
ReLU :
5. Encoder和Decoder之间的信息交互
解码器最后输出的是一个向量,如何把它变成一个单词,这就要靠它后面的线性层和 softmax 层
线性层是一个简单的全连接神经网络,它是由Decoder堆栈产生的向量投影到一个更大,更大的向量中,称为对数向量;
在线性层之后是一个softmax层,softmax将这些分数转换为概率。选取概率最高的索引,然后通过这个索引找到对应的单词作为输出。