强烈推荐:https://jalammar.github.io/illustrated-transformer/
特点: 简单明了,清晰易懂。对Transformer里的self-attention(multi-head), positional encoding这些concepts有一个基本的认识。
缺点:具体细节仍需要进一步阅读其他资料。
不喜欢阅读英文的同学,可以看下这个很棒的中文资料:https://kexue.fm/archives/4765
优点: 有写者自己更多的思考、分析。可以帮助读者对Transformer里的模块有更深刻的认识。我要为写者点赞!
https://jalammar.github.io/visualizing-neural-machine-translation-mechanics-of-seq2seq-models-with-attention/
此链接对seq2seq模型里的注意力进行了可视化解释
Transformer由encoding component和decoding component组成,encoding component由6个堆叠的Encoder组成,decoding component是6个堆叠的Decoder组成。
每一个Encoder有两个部分:self-attention + feed forward neural network
每一个Decoder有三个部分:self-attention + encoder-decoder attention + feed forward
关键的一点是,同RNN的输出一样,每个时刻都会输出一个向量表示。因此对每一个位置,都会进行self-attention以及feed forward nn。上面提到6个堆叠的Encoder,同样和堆叠RNN一样,把上一层每个时刻的输出作为当层每一时刻的输入。
在self-attention中,每个位置有其对应的Query vector, Key vector, Value vector。三个向量是通过矩阵运算得到的。
x
1
×
W
Q
=
q
1
x_1 × W_Q = q_1
x1×WQ=q1
x
1
×
W
K
=
k
1
x_1 × W_K = k_1
x1×WK=k1
x
1
×
W
V
=
v
1
x_1 × W_V = v_1
x1×WV=v1
接下来,使用
q
1
q_1
q1分别和
k
1
,
k
2
,
k
3
.
.
.
.
k_1,k_2, k_3....
k1,k2,k3....计算得分, 分数除以8,然后softmax,最后对
v
1
,
v
2
,
v
3
.
.
.
v_1,v_2,v_3...
v1,v2,v3...加权求和得到第一个位置的输出。
多头自注意力就是使用多个不同的 W K , W Q , W V W_K, W_Q, W_V WK,WQ,WV矩阵。将结果拼接起来,再经过矩阵变换。
上述的模型没有考虑单词的顺序信息,而在NLP任务中,顺序是很重要的信息。因此谷歌提出了position embedding并将其作用在输入端。
在Decoder部分,不同于Encoder的是加了Encoder-Decoder-Attention部分,这里将最顶层Encoder部分的输出作为Key, Value,Decoder部分作为Query。具体计算方式不变。