【注意力机制】Self-attention注意力机制理论知识

输入输出类别（N指向量个数）：

• 第一种：输入 $N$，输出 $N$（例如：Sequence Labeling、POS tagging）
• 第二种：输入 $N$，输出 $1$（例如：Sentiment analysis）
• 第三种：输入 $N$，输出 $N^{\prime }$（seq2seq任务：由机器自己决定输出的个数）例如：翻译任务、语音辨识

以第一种为例Sequence Labeling任务：

例子： I saw a saw

序列输入到FC层得到各个单词的词性不能解决saw是不同词性的问题，采用考虑上下文的context，将前后几个向量串起来输入到FC，考虑Window包括多个前后多个frame

考虑整个input的sequence资讯，可将Window变大甚至盖住整个sequence，但是可能过拟合，FC参数过多。

Self-attention引入

考虑整个input的sequence资讯信息更好的办法：
self-attention考虑一整个资讯信息并返回与输入相同数目的vetor输出，再输入FC决定应该是什么。

self-attention可以叠加：
输入+self-attention+FC+self-attention+FC（交替使用，FC专注某一位置的咨询，self-attention关注整个资讯）

注：self-attention最相关文章：Attention is all you need.（提出了transformer，transformer最重要的module就是self-attention，Bert也有应用）

self-attention架构

self-attention输入是一串Vector（可能整个网络的input，也可能是隐层的output）

例：
输入： $a^1{a}^2{a}^3{a}^4$
输出： $b^1{b}^2{b}^3{b}^4$

self-attention怎么产生$b$

例子：产生$b^1$

找出与 $a^1$ 相关的其他向量（哪些向量是重要的，哪些部分跟判断 $a^1$ 是哪一个level有关系，哪些是决定 $a^1$ 得class或regression数值所需用到的资讯）
其中每一个向量与 $a^1$ 关联的程度，用数值 α 表示（计算 α 方法有：Dot-product（transformer中使用）、Additive）

$q^1=W^q{a}^1$（$q$称为query）
$k^2=W^k{a}^2$（$k$称为key），$k^3=W^k{a}^3$， $k^4=W^k{a}^4$，$k^1=W^k{a}^1$

$\alpha (1,2)=q^1\cdot k^2$（$\alpha (1,2)$代表query是1提供，key是2提供的，1与2的关联性，也称为attention score），$\alpha (1,3)=q^1\cdot k^3$，$\alpha (1,4)=q^1\cdot k^4$

本身也会计算：
$\alpha (1,1)=q^1\cdot k^1$
计算出$a^1$与每一个向量关联性后，做一个softmax，输出一排${\alpha }^{\prime }(1,i)$

根据 ${\alpha }^{\prime }$ 抽取这个sequence里重要的咨询：
$v^1=W^v{a}^1$、$v^2=W^v{a}^2$、$v^3=W^v{a}^3$、$v^4=W^v{a}^4$
再用 $v^i$ 乘以 ${\alpha }^{\prime }(1,i)$ ，然后再累加得到 $b^1$
${\alpha }^{\prime }(1,1)\cdot v^1+{\alpha }^{\prime }(1,2)\cdot v^2+{\alpha }^{\prime }(1,3)\cdot v^3+{\alpha }^{\prime }(1,4)\cdot v^4=b^1$

例子：产生$b^2$

主角：$a^2$
每一个向量与$a^2$关联的程度，用数值 $\alpha$ 表示（计算 $\alpha$ 方法：Dot-product）
$q^2=W^q{a}^2$
$k^1=W^k{a}^1$、$k^2=W^k{a}^2$、$k^3=W^k{a}^3$、$k^4=W^k{a}^4$
根据 $q^2$ 去对 $a1$ 到 $a4$ 求attention score
$\alpha (2,1)=q^2\cdot k^1$、$\alpha (2,2)=q^2\cdot k^2$、$\alpha (2,3)=q^2\cdot k^3$、$\alpha (2,4)=q^2\cdot k^4$
计算出 $a^2$ 与每一个向量关联性后，做一个softmax，输出一排 ${\alpha }^{\prime }(2,i)$
根据 ${\alpha }^{\prime }$ 抽取这个sequence里重要的咨询：
$v^1=W^v{a}^1$、$v^2=W^v{a}^2$、$v^3=W^v{a}^3$、$v^4=W^v{a}^4$
再用 $v^i$ 乘以 ${\alpha }^{\prime }(2,i)$，然后再累加得到 $b^2$
${\alpha }^{\prime }(2,1)\cdot v^1+{\alpha }^{\prime }(2,2)\cdot v^2+{\alpha }^{\prime }(2,3)\cdot v^3+{\alpha }^{\prime }(2,4)\cdot v^4=b^2$

self-attention 总结：

第一步：每一个a都要分别产生q k v ，I为Input

$q^i=W^q{a}^i$ ——> $\left(\begin{array}{cccc}{q}^1& q^2& q^3& q^4\end{array}\right)=W^q\left(\begin{array}{cccc}{a}^1& a^2& a^3& a^4\end{array}\right)$——> $Q=W^q\cdot I$

$k^i=W^k{a}^i$——> $\left(\begin{array}{cccc}{k}^1& k^2& k^3& k^4\end{array}\right)=W^k\left(\begin{array}{cccc}{a}^1& a^2& a^3& a^4\end{array}\right)$——> $K=W^k\cdot I$

$v^i=W^v{a}^i$——> $\left(\begin{array}{cccc}{v}^1& v^2& v^3& v^4\end{array}\right)=W^v\left(\begin{array}{cccc}{a}^1& a^2& a^3& a^4\end{array}\right)$——> $V=W^v\cdot I$

第二步：每一个 $q^i$ 都和每一个 $k$ 计算 Dot-product 得到 attention score

$\alpha (1,1)=k^1\cdot q^1$，$\alpha (1,2)=k^2\cdot q^1$，$\alpha (1,3)=k^3\cdot q^1$，$\alpha (1,4)=k^4\cdot q^1$

$\alpha (1,i)=k^i\cdot q^1$ ——> $\left(\begin{array}{c}{k}^1\\ k^2\\ k^3\\ k^4\end{array}\right)\cdot q^1=\left(\begin{array}{c}\alpha (1,1)\\ \alpha (1,2)\\ \alpha (1,3)\\ \alpha (1,4)\end{array}\right)$

$\left(\begin{array}{c}{k}^1\\ k^2\\ k^3\\ k^4\end{array}\right)\cdot \left(\begin{array}{c}{q}^1{q}^2{q}^3{q}^4\end{array}\right)=\left(\begin{array}{c}\alpha (1,1)\alpha (2,1)\alpha (3,1)\alpha (4,1)\\ \alpha (1,2)\alpha (2,2)\alpha (3,2)\alpha (4,2)\\ \alpha (1,3)\alpha (2,3)\alpha (3,3)\alpha (4,3)\\ \alpha (1,4)\alpha (2,4)\alpha (3,4)\alpha (4,4)\end{array}\right)$ ——>$K^T\cdot Q=A$ ——> $A^{\prime }=softmax(A)或Relu(A)$

第三步：输出 $b^1={\alpha }^{\prime }(1,i)\cdot v^i$

$\left(\begin{array}{c}{v}^1{v}^2{v}^3{v}^4\end{array}\right)\cdot \left(\begin{array}{c}\alpha (1,1)\alpha (2,1)\alpha (3,1)\alpha (4,1)\\ \alpha (1,2)\alpha (2,2)\alpha (3,2)\alpha (4,2)\\ \alpha (1,3)\alpha (2,3)\alpha (3,3)\alpha (4,3)\\ \alpha (1,4)\alpha (2,4)\alpha (3,4)\alpha (4,4)\end{array}\right)=\left(\begin{array}{c}{b}^1{b}^2{b}^3{b}^4\end{array}\right)$——>$V\cdot A^{\prime }=O$

$O$ 其中每一个column就是self-attention的输出。

self-attention中唯一需要学的参数：$W^q$、$W^k$、$W^v$

Multi-head Self-attention

self-attention是用 $q$ 去找相关的 $k$ ，相关有很多种不同的形式，也许不能只有一个 $q$，要多个 $q$负责不同种类的相关性。
$q^i=W^q{a}^i$（$q$称为query）然后 $q^i$ 再乘以另外2个矩阵得到$q^{i,1}$，$q^{i,2}$
解释：2个head，里面有2种不同相关性，$i$代表位置，1和2代表这个位置的第几个$q$，$q$有2个，对应 $k$ 和 $v$ 也有2个。


得到的 $b$可以都接起来，再乘以一个矩阵得到 $b^i$，再送到下一层。

Positional encoding

以上Self-attention的方法没有对位置信息考虑，需要考虑上位置的咨询。
Positional encoding：为每一个位置设定一个vetor（称为positional vetor：$e^i$（$i$代表位置）
$e^i+a^i$ 再计算 $q^i$、$k^i$、$v^i$

Positional encoding有不同的方法产生：例如sinusiodal/cos、RNN、Floater network、或者通过学习得到。

其他Self-attention应用

注：做语音识别、语音辨识整个sequence可能太长，使用Truncated Self-attention人为的设定一小个范围。

self-attention用于Graph上，结点和边对应到矩阵上，没有边对应的位置可以不计算attention score 直接设为0，这就是一种GNN