Transformer

强烈推荐: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。具体计算方式不变。

你可能感兴趣的:(深度学习)