【机器学习】李宏毅——自注意力机制(Self-attention)

前面我们所讲的模型，输入都是一个向量，但有没有可能在某些场景中输入是多个向量，即一个向量集合，并且这些向量的数目并不是固定的呢？

这一类的场景包括文字识别、语音识别、图网络等等。

那么先来考虑输出的类型，如果对于输入是多个数目不定的向量，可以有以下这几种输出方式：

最简单的思路当然是将每一个向量单独作为一个输入，然后将其丢入到一个全连接之中，每个单独产生结果。但是这种方法的问题在于没办法考虑前后向量之间的联系。那么转换另一个思路就是开一个window，每次将window中的向量一起输入到全连接网络中，如下图：

这种方式的问题在于如果遇到某些场景需要全部向量都考虑，但是我们无法事先知道这个场景中向量的数目，我们也就很难调整我们的window去适应。在这种情况下就需要用到自注意力机制了。

可以将自注意力机制看成一层，其接受所有向量的输入，然后输出的数目跟输入的数目的是一样的，可以认为它的每个输出都是考虑了整个向量集之后的结果，之后再将这些输出每个单独放入一个全连接网络中来得到输出，这样可以显著的降低全连接网络的参数量。并且这个也可以多次使用，自注意力后FC再自注意力再FC等等。

那么接下来的问题就是这个机制如何接受输入并作出相应的输出。这边以第一个向量对应的第一个输出为例：

最常见的是这种Dot-product，就是将两个向量分别乘以一个矩阵，得到新的向量，然后再进行点乘得到一个数值，也就是\(\alpha\)

哪一个向量对应的\(\alpha\)(attention score)越大，那么得到的b就越接近于它所计算出来的\(v\)，这就是最终结果了。

而经过上述的分析，可以看到需要学习的参数只有\(W^q,W^k,W^v\)而已。

这是认为不同的输入向量之间可能不止有一种相关性，因此：

各种参数都有多重，只不过各自算各自的。

在前面的自注意力过程中，我们并没有关注到各个输入向量之间的位置信息！这在一些场景里面是很关键的一部分信息，但是我们只是计算了各自的相关性没有考虑到位置信息，因此需要加上Position embedding：

只需要在每个输入向量上加上一个\(e^i\)即可，这个里面就包含了具体的位置信息，而如何产生\(e^i\)仍然是一个在研究的问题。

事实上，CNN是受限制的Self-attention，也就是Self-attention的特例，Self-attention可以通过某些设计和限制就可以变成CNN，完成和CNN同样的任务。

RNN也是处理输入是向量序列问题的算法，其具体过程就是一开始有一个memory，它和第一个输入向量一起输入到RNN中将会输出一个向量，该向量一方面放入全连接网络中得到一个输出，另一方面和下一个输入向量一起作为下一个RNN的输入，以此类推。

而它们的区别在于：

对不同位置的考虑：在RNN中如果最后一个黄色的输出向量希望它能够与第一个输入的蓝色向量有关，那么就需要蓝色向量从一开始输入就一直被记得，直到最后一个；而在Self-attention则不用考虑到这个问题，计算就已经是统筹兼顾了
计算的平行性：RNN是无法进行平行计算的，而Self-attention是可以的。

巴特西