Transformer学习-self-attention
这里写自定义目录标题
- Self-attention
- Multi-head self-attention
- 用self-attention解决其他问题
Self-attention
- 用Wq、Wk、Wv分别乘输入向量得到q、k、v向量
- 用每个q向量乘所有的k向量得到对应项的attention，即用每项的query向量去匹配所有的key向量，得到该项对所有项的注意力打分。可以用矩阵优化运算。激活函数softmax可以用relu等替换。
- 用q、v的相乘的结果乘v得到self-attention的输出b
- 综上，如下图
Multi-head self-attention
- 通过多组q、k、v来表示不同的相关性
- 将多头的输出融合到一起
- 位置编码，将位置信息编码为向量（每个位置一个专属向量），加到输入中即可
用self-attention解决其他问题
- 语音解析：输入维度过大，使用truncated self-attention
- 图像处理：每个像素(三个通道)可以看作一个三维向量，如下图可以将50个三维向量作为输入
- self-attention vs cnn：cnn相当于简化版的self-attention，卷积核的运算相当于对卷积核区域内的像素点求k、v，self-attention的感受野是自己学习的，而cnn的卷积核是手动设置的
cnn就是self-attention的特例。self-attention更灵活，但是如果训练集小可能更容易过拟。
4. self-attention vs RNN
5. self-attention for Graph：用attention来表示nodes之间的关联
