Transformer笔记

Transformer

参考视频:https://www.bilibili.com/video/BV19Y411b7qx?p=4&spm_id_from=pageDriver&vd_source=b070ed3f7d11d76ba6c8c88f7c6bbf28

计算注意力

Transformer笔记_第1张图片
a,b向量分别与WQ,WK,WV相乘得到 Queries,Keys,Values
Transformer笔记_第2张图片
divid by 8 将score÷8,64跟词向量编码的维度有关(82=64)。
上面说到向量计算,下面说矩阵运算,计算机效率更高。
Transformer笔记_第3张图片
单头注意力和多头注意力,多组QKV。
Transformer笔记_第4张图片
Transformer笔记_第5张图片
多头注意力计算过程,将Zi(i=0-7),横着拼到一起再与W0相乘。

词向量编码

Transformer笔记_第6张图片
Transformer笔记_第7张图片
Transformer笔记_第8张图片

MASK

因为句子长度不一致,为了统一,取最长句子的长度,将短的句子用PADK补全,在计算其他词对PAD的注意力时用MASK代替,但是计算PAD对其他词的注意力时,正常计算。
Transformer笔记_第9张图片
根据a来预测出b所以不能让a看到b;同样的c对b来说也是未来词,不能注意到。
Transformer笔记_第10张图片
可以对两者取并集。

Transformer计算流程

Transformer笔记_第11张图片
将词首先进行词向量编码,然后跟之前讲到的位置编码PE矩阵相加,得到最终编码;使用编码向量进行自注意力计算得到Z,然后将Z与X相加(短接运算,理解为X和A的简单相加),随后做一个数据标准化,然后做一个全连接的运算。输出shape:单词个数每个向量编码长度【10512】(10个单词,每个单词编码为512维的向量)
Transformer笔记_第12张图片
encoder可以是多个。在decoder中初始输入与encoder是一样的,都是词向量编码+位置编码后的最终编码结果,只不过step=0时输入是起始符(后面依次输入是起始符,单词1(单词1是上词预测出的结果再作为输入);起始符,单词1,单词2;起始符,单词1,单词2,单词3…),进行自注意力计算,进行X,Z相加和标准化。下面就不一样了,encoder-decoder attention这一层,也可以看作是自注意力计算不过这里的qkv不是按照x1和x2来计算的,是按照encoder的计算结果来作为kv用自己自注意力的结果作为q计算的。
transformer参考链接

你可能感兴趣的:(机器学习,transformer,人工智能)