注意力机制Attention详解

article/2025/9/16 8:53:58

注意力机制Attention详解

一、前言

2018年谷歌提出的NLP语言模型Bert一提出，便在NLP领域引起热议，之所以Bert模型能够火出圈，是由于Bert模型在NLP的多项任务中取得了之前所有模型都不能达到的出色效果。那么Bert模型它是如何提出的呢？我们来追溯Bert模型的发展历史：
首先Bert模型是基于Transformer提出的，Transformer同样是谷歌团队在2017年发表的论文 《Attention is all you need》 中提出，Transformer又是基于attention机制，注意力（Attention）机制由Bengio团队于2014年提出。而目前大部分attention模型都是依附于Encoder-Decoder框架进行实现。因此，我们要讲解Bert，必须要从Encoder-Decoder框架开始说起。

二、注意力Attention简述

什么是注意力： 对于人类来说，视野非常开阔，但焦点往往只有小部分范围，甚至一个点，这就叫做注意力。它可以帮助人类在众多的信息中去筛选对人类有帮助的信息。
注意力机制的核心目标是： 从众多信息中选择出对当前任务目标更关键的信息，将注意力放在上面。

在这里插入图片描述

三、Encoder-Decoder框架

在这里插入图片描述
上图是最常见的Encoder-Decoder框架图，在文本处理和语音识别的Encoder部分通常采用RNN模型；在图像处理的Encoder部分通常采用CNN模型。

举个栗子： 机器翻译
英文：Tom chase Jerry
中文：汤姆追逐杰瑞

对应在图中，输入序列 {x₁=Tom，x₂=chase ，x₃=Jerry} ，经过Encoder，被编码成语义编码C，语义编码是一块高度抽象的内容。

这里的语义编码C的编码方式是啥呢？
编码方式有很多种，在NLP中常见的RNN/LSTM/以及它们的变种，这里我们以RNN为例。
输入序列{x₁,x₂,x₃,x₄}，通过RNN生成隐藏层的状态值{h₁,h₂,h₃,h₄}，最简单的办法可以直接采用最后时刻输出的h_t作为C的状态值，这里也就是h₄作为语义编码C的值，也可以将所有时刻的隐藏层的值进行汇总，然后生成语义编码C的值，这里就是C=q(h1,h2,h3,h4)，q是非线性激活函数。

再通过Decoder将其解码为中文内容，同样解码也可以采用RNN/LSTM/以及它们的变种进行解码。Encoder编码和Decoder解码的方式可以任意搭配。
解码完成后输出序列 {Y₁=“汤姆”，Y₂=“追逐”，Y₃=“杰瑞”}。在此过程中，每次生成一个目标单词，每个目标单词的有效信息都不一样，例如“杰瑞”，与之最重要的信息是“Jerry”，其他的可以看作成噪声或者相关度不大的信息。

相比于注意力机制而言，还有一种分心模型：
体现在翻译输出时，对于输入贡献的注意力是一样的。例如翻译：“杰瑞”时，“Tom”，“chase”，“Jerry”三者贡献的注意力相同（同一个C）。

分心模型的计算：
Y₁ = f ( C ) ； Y₂ = f ( C ，Y₁ ) ； Y₃ = f ( C ，Y₁，Y₂ )

引入注意力机制之后：
在这里插入图片描述

此时会分配不同的语义编码C，语义编码C会告诉输入哪一个更为相关。

引入注意力机制后的计算：
Y₁ = f₁ ( C₁ ) ； Y₂ = f₁ ( C₂ ，Y₁ ) ； Y₃ = f₁ (C₃ ，Y₁，Y₂ )
C_汤姆 = g ( 0.6 * f₂(Tom)，0.2 * f₂(chase)，0.2 * f₂(Jerry))
C_追逐 = g ( 0.2 * f₂(Tom)，0.7 * f₂(chase)，0.1 * f₂(Jerry))
C_杰瑞 = g ( 0.3 * f₂(Tom)，0.2 * f₂(chase)，0.5 * f₂(Jerry))
g(x) 是对构成的元素进行加权求和：a_ij表示权重

四、Attention机制的具体计算过程

在这里插入图片描述
步骤：

根据Query和Key计算两者相似性或相关性；
常见方法：
(1) 两者向量点积： Similarity(Query，Key_i) = Query · Key_i
(2) Cosine 相似性： Similarity(Query，Key_i) =(Query · Key_i)/||Query||·||Key_i||
(3)引入额外的神经网络 MLP网络： Similarity(Query，Key_i) = MLP(Query，Key_i)
经过步骤1得到原始得分s₁（例如Query = 汤姆，Key1 = Tom，经过F(Q，K)函数计算后得到两者相关性的得分，其他的同理。由于s₁是一个实际的分值，通过softmax函数进行归一化处理，将得分映射到0-1之间得到权重系数（a1，a2，a3，a4）。
根据权重系数对Value进行加权求和。(权重系数a1，a2，a3，a4先分别和Value1，Value2，Value3，Value4相乘，最后相加)
最终得到Attention Value

Attention计算公式：
在这里插入图片描述

五、Attention机制优缺点

优点：
1.速度快。Attention机制不再依赖于RNN，解决了RNN不能并行计算的问题。这里需要说明一下，基于Attention机制的seq2seq模型，因为是有监督的训练，所以咱们在训练的时候，在decoder阶段并不是说预测出了一个词，然后再把这个词作为下一个输入，因为有监督训练，咱们已经有了target的数据，所以是可以并行输入的，可以并行计算decoder的每一个输出，但是再做预测的时候，是没有target数据地，这个时候就需要基于上一个时间节点的预测值来当做下一个时间节点decoder的输入。所以节省的是训练的时间。
2.效果好。因为注意力机制，能够获取到局部的重要信息，能够抓住重点。
缺点：
1.只能在Decoder阶段实现并行运算，Encoder部分依旧采用的是RNN，LSTM这些按照顺序编码的模型，Encoder部分还是无法实现并行运算，不够完美。
2.因为Encoder部分目前仍然依赖于RNN，所以对于中长距离之间，两个词相互之间的关系没有办法很好的获取。

六、Self-Attention机制

在上面的Attention机制结构图中，Self-Attention其实表示的是Query、Key、Value三者是相同的值。(例如Query = Tom，Key1 = Value1 = Tom，Key2 = Value2 = chase，Key3 = Value3 = Jerry)。
同样下图中Query表示：making…(句子中的每一个词)；Key/Value表示整句话中的每个单词；权重系数表示句子中哪个词和我的这个词关联最大（making，more，difficu）
Self-Attention可以捕获同一个句子中单词之间的一些句法特征(比如下图中展示的有一定距离的短语结构)或者语义特征(图中its的指代对象是Law)
在这里插入图片描述