深度学习：注意力机制(Attention Mechanism)

原创最新推荐文章于 2026-04-21 11:42:04 发布 1w 阅读 AI写同款

收录于

人工智能初探

源启智能

码龄20年墨衍会员 · AI 创作全网分发

3万+粉丝285篇原创

关注

注意力机制详解 注意力机制详解Attention机制由来Attention定义Encoder-Decoder框架存在问题：Attenion机制的引入Attention机制的类别Hard AttentionSoft AttentionGlobal AttentionLocal AttentionSelf AttentionMulti-head Attention附录—Transformer全局视角Tensor视角Encoder部分Self-AttentionSelf-Attention向量计算Self-Attention的矩阅读详情

AI摘要

本文介绍了注意力机制在深度学习，尤其是NLP和计算机视觉中的关键作用，包括其基本原理、在自然语言处理和计算机视觉任务中的应用，以及如何通过Encoder-Decoder框架实现。重点讨论了加性模型、点积模型和双线性模型在注意力计算中的区别。

1 注意力机制概述

1.1 定义

注意力机制（Attention Mechanism）是深度学习领域中的一种重要技术，特别是在序列模型如自然语言处理（NLP）和计算机视觉中。它使模型能够聚焦于输入数据的重要部分，从而提高整体性能和效率。

1.2 基本原理

聚焦关键信息：注意力机制模仿人类的注意力过程，使模型能夠集中处理输入数据中的关键信息，同时忽略不太相关的部分。
权重分配：通过为输入数据的不同部分分配不同的权重（或注意力分数），模型能够识别最重要的信息。

1.3 应用领域

自然语言处理（NLP）：在机器翻译、文本摘要、情感分析等任务中，注意力机制帮助模型关注文本的关键部分。
计算机视觉：在图像分类、目标检测和图像字幕生成等任务中，注意力机制使模型能够专注于图像的关键区域。
语音识别：注意力机制帮助模型关注语音信号的重要部分，提高语音识别的准确性。

1.4 优势

提高性能：通过集中处理重要信息，注意力机制可以提高模型的准确性和效率。
提供可解释性：注意力权重可以用来解释模型的决策过程，增加模型的透明度。
灵活性强：可以与多种模型结构结合，如循环神经网络（RNN）、卷积神经网络（CNN）和Transformer。

2 注意力机制分析

2.1 Encoder-Decoder框架

下面基于Encoder-Decoder框架对注意力机制深入分析，Encoder-Decoder框架的原理如下：

对于句子对<source，target>:

编码器对输入source进行编码，转换成中间语义向量C:

对于解码器Decoder，根据中间语义向量C和当前已生成的历史信息来生成下一时刻要生成的单词：

基本的Encoder-Decoder框架没有体现出注意力机制

把Decoder生成拆开来看：

f是非线性变换函数，在生成目标单词时，使用的语义向量C都是一样的，所以，source中任意单词对某个单词y_i来说，影响力都是相同的。

如果输入句子比较短，对于输出影响不是很大，但如果输入句子很长，这时所有的语义都通过一个语义向量C来表示，单词自身的信息会消失，很多细节信息会被丢失，最终的输出也会受到影响，所以要引入注意力机制。

2.2 经典注意力机制

在机器翻译中，比如输入source为Tom chase Jerry。输出想得到中文：汤姆追逐杰瑞。在翻译Jerry这个单词时，在普通Encoder-Decoder模型中，source里的每个单词对“杰瑞”贡献是相同的，但这样明显和实际不是很相符，在翻译“杰瑞”的时候，我们更关注的应该是"Jerry"，对于另外两个单词，关注的会少一些。

引入Attention模型后，在生成“杰瑞“的时候，会体现出输入source的不同影响程度，比如：

每个概率代表了翻译当前单词“杰瑞”时注意力分配模型分配给不同英文单词的注意力大小。

在注意力机制中，对于target中每个单词都有一个对应source中单词的注意力概率。

由于注意力模型的加入，原来在生成target单词时候的中间语义C就不再是固定的，而是会根据注意力概率变化的 $C_{i}$ ，加入了注意力模型的Encoder-Decoder框架如下：

生成target的过程就变成了如下形式：

每个 $C_i$ 可能对应着不同的注意力分配概率分布，比如对于上面的英汉翻译来说，其对应的信息可能如下：

其中， $f_2$ 表示Encoder对输入英文单词的某种变换函数，比如如果Encoder是用RNN模型的话，这个 $f_2$ 函数的结果往往是某个时刻输入后隐层节点的状态值。g代表Encoder根据单词的中间表示合成整个句子中间语义表示的变换函数，一般g函数就是对构成元素的加权求和：

其中， $L_x$ 代表输入句子Source的长度，a_i_j代表在Target输出第i个单词时，Source输入第j个单词的注意力分配系数； $h_{ij}$ 代表Source输入句子中第j个单词的语义编码。举例输出汤姆如下：

2.3 注意力分布

注意力分布α就是判断什么信息重要，什么信息不重要，分别赋予不同的权重。

如何计算α：

用X表示输入当输入 N 个向量：[x1,...,xN]，想要从中选出对于目标而言比较重要的信息，需要引入目标任务的表示，称为查询向量（query vector），此时问题可以转换为考察输入的不同内容和查询向量之间的相关度，也就是对不同内容进行打分，赋予与当前任务比较相关的部分更大的权重，再通过一个softmax层得到分布，也就是输入信息的不同部分的权重。

注意力变量z∈[1, N]来表示被选择信息的索引位置，即z=i来表示选择了第i个输入信息，然后计算在给定了q和X的情况下，选择第i个输入信息的概率αi：