Multi-Head Self-Attention与Multi-Head Attention
时间: 2023-09-16 20:11:48 浏览: 146
各种attention的实现
Multi-Head Self-Attention和Multi-Head Attention都是Transformer模型中的核心模块。
Multi-Head Self-Attention是指在一个句子中每个单词使用自注意力机制(Self-Attention)来计算其上下文表示。这个过程会将每个单词的原始表示分别映射成多个子空间,并在每个子空间中计算注意力权重,最后将这些子空间的注意力权重加权求和,得到每个单词的上下文表示。这种方式可以捕获单词之间的复杂依赖关系,并且具有可解释性。
Multi-Head Attention是指在一个句子的两个不同部分之间使用注意力机制(Attention)来计算它们之间的关系。这个过程也会将每个单词的原始表示分别映射成多个子空间,并在每个子空间中计算注意力权重,最后将这些子空间的注意力权重加权求和,得到不同部分之间的关系表示。这种方式可以捕获不同部分之间的语义关系,例如句子中的主语和谓语之间的关系。
总的来说,Multi-Head Self-Attention和Multi-Head Attention都是使用多个子空间来计算注意力权重,但是Multi-Head Self-Attention是单向的,只考虑一个句子中的单词之间的关系,而Multi-Head Attention是双向的,可以考虑两个不同的部分之间的关系。
阅读全文