概述
Decoder也是N=6层堆叠的结构,每层被分3层: 两个注意力层和前馈网络层,同Encoder一样在主层后都加有Add&Norm,负责残差连接和归一化操作。
Encoder与Decoder有三大主要的不同:
- 第一层 Masked Multi-Head Attention: 采用Masked操作
- 第二层 Multi-Head Attention: K, V矩阵是使用Encoder编码信息矩阵C进行计算,而Q使用上一个Decoder的输出计算。
- 概率计算输出: Linear和Softmax作用于前向网络层的输出后面,来预测对应的word的probabilities
Encoder的输入矩阵用X表示,输出矩阵用C表示
流程图表示如下:
graph LR
A(Output Embedding) --Positional Encoding--> B[带掩码的多头注意力层] --> C[多头注意力层] --> D[前馈网络层] --特征值 --> D1[Add&Norm] --> E[Linear] --> F[Softmax] --> G(OutputProbilities)
style A fill:#fff,stroke:#333,stroke-width:1px;
style G fill:#fff,stroke:#333,stroke-width:1px;
style D1 fill:#eee,stroke:#333,stroke-width:1px;
分析如下:
- 我们将 输入转换为嵌入矩阵,再加上位置编码,输入解码器
- 解码器收到输入,将其发送给带掩码的多头注意力层,生成注意力矩阵M
- 将注意力矩阵M和Encoder输出的特征值R作为多头注意力层的输入,输出第二层注意力矩阵
- 从第二层的多头注意力层得到注意力矩阵,送入前馈网络层,后者将解码后的特征作为输出
- 前馈网络层的输出经过Add&Norm后,做linear及Softmax回归,并输出目标句子的特征
Decoder 的输入
Decoder的输入结构与encoder的一样。