重要说明：本文从网上资料整理而来，仅记录博主学习相关知识点的过程，侵删。

一、参考资料

通道注意力，空间注意力，像素注意力

通道注意力机制和空间注意力机制

视觉 注意力机制——通道注意力、空间注意力、自注意力

二、通道注意力与空间注意力机制

1. 通道注意力机制(CAM)

通道注意力机制(Channel Attention Mechanism, CAM)，通过特征内部之间的关系来通道注意力机制。特征图的每个通道都被视作一个特征检测器，所以通道特征聚焦的是图像中有用的信息是"什么"（what）。

1.1 通道注意力的结构

通道注意力模块的结构，如下图所示：

1.2 通道注意力流程

通道注意力的思路流程是：

1. 首先，对一个尺寸为 H×W×C的输入特征图F进行空间维度的全局最大池化和全局平均池化，得到两个 1×1×C 的特征图；（在空间维度进行池化，压缩空间尺寸，便于后面学习通道的特征）
2. 然后，将全局最大池化和全局平均池化的结果，分别送入一个共享的多层感知机(MLP)中学习，得到两个 1×1×C 的特征图。MLP的第一层神经元个数为 C/r，激活函数为 Relu，第二层神经元个数为 C；（基于MLP学习通道维度的特征，和各个通道的重要性）
3. 最后，将MLP输出的结果进行Add操作，接着经过Sigmoid激活函数的映射处理，最终得到通道注意力权重矩阵 $M_c$；

1.3 通道注意力的计算公式

空间注意力权重矩阵 $M_c$，可以表示为：
$$M_c\in R^{C\times 1\times 1}$$
为了降低计算参数，在MLP中采用一个降维系数r，$M_c\in R^{C/r\times 1\times 1}$。

综上，通道注意力计算公式如下：
$$\begin{array}{c}{M}_c(F)=\sigma (MLP(AvgPool(F))+MLP(MaxPool(F)))\\ =\sigma (W_1(W_0(F_{avg}^c))+W_1(W_0(F_{max}^c)))\end{array}$$
上式中，$F_{avg}^c$ 和 $F_{max}^c$ 分别表示全局平均池化特征和最大池化特征。

2. 空间注意力(SAM)

通过对特征图空间内部的关系来产生空间注意力特征图。不同于通道注意力，空间注意力(Spatial Attention Mechanism, SAM)聚焦于特征图上的有效信息在"哪里"（where）。

2.1 空间注意力的结构

2.2 空间注意力流程

空间注意力的思路流程是：

1. 首先，对一个尺寸为 H×W×C的输入特征图F进行通道维度的全局最大池化和全局平均池化，得到两个 H×W×1 的特征图；（在通道维度进行池化，压缩通道大小，便于后面学习空间的特征）
2. 然后，将全局最大池化和全局平均池化的结果，按照通道拼接(concat)，得到特征图尺寸为HxWx2，
3. 最后，对拼接的结果进行7x7的卷积操作，得到特征图尺寸为 HxWx1，接着通过Sigmoid激活函数 ，得到空间注意力权重矩阵 $M_s$；

2.3 空间注意力的计算公式

空间注意力权重矩阵 $M_s$，可以表示为：
$$M_s(F)\in R^{H,W}$$
同上，在通道维度使用两种池化方法产生2D特征图：
$$\begin{gathered} F_{avg}^s\in R^{1\times H\times W} \\ {F}_{max}^s\in R^{1\times H\times W} \end{gathered}$$
综上，空间注意力的计算公式如下：
$$\begin{array}{rl}{M}_s(F)& =\sigma (f^{7\times 7}([AvgPool(F);MaxPool(F)]))\\ & =\sigma (f^{7\times 7}[F_{avg}^s;F_{max}^s])\end{array}$$

3. 混合注意力机制

在混合注意力机制中，通道注意力和空间注意力可以通过串联、或者并联的方式进行组合。关于混合注意力机制的介绍，请参考下文的CBAM网络模型。

三、SENet：通道注意力

SENet：[1]

深入理解SENet：自适应特征提取的注意力机制

//TODO

压缩和激励网络（Squeeze-and-Excitation Networks，SENet）是通道注意力机制的代表模型。SENet 分为压缩和激励两个部分，其中压缩部分的目的是对全局空间信息进行压缩，然后在通道维度进行特征学习，得到各个通道的权重，最后通过激励部分对各个通道进行分配不同权重。

上图是SE模块的结构， 在压缩部分，输入特征图的尺寸为 H×W×C。压缩部分的功能是将维数从 H×W×C 压缩至1×1×C，即把 H×W 压缩为 1×1 维，这个过程由**全局平均池化(GAP)**实现。

在激励部分，需要将压缩部分得到的 1×1×C 的维度融入全连接层，预测各个通道的重要程度，然后再激励到前面特征图对应通道上进行操作。采用简单的门控机制与Sigmoid 激活函数。

在通道注意力机制，学习各个通道的重要性时，是先对特征图的空间进行压缩，然后在通道维度进行学习，得到各个通道的重要性。

四、STN：空间注意力

原始论文：[2]

空间变形网络——STN

详细解读Spatial Transformer Networks（STN）-一篇文章让你完全理解STN了

通俗易懂的Spatial Transformer Networks(STN)(一)

通俗易懂的Spatial Transformer Networks(STN)(二)

从仿射变化到STN网络

1. STN网络简介

空间变换神经网络 (Spatial Transformer Networks，STN)是空间注意力机制的代表模型，STN 能够对各种形变数据在空间中进行转换并自动捕获重要区域特征。它能够保证图像在经过裁剪、平移或者旋转等操作后，依然可以获得和操作前的原始图像相同的结果。

举个例子，在MNIST 数字分类的中应用STN，该分类过程一共包含 4 个步骤：

1. MNIST中的数字，是经过随机平移、缩放和旋转处理；把它们输入到STN网络中；
2. 通过STN网络，预测前面输入数字的变换（是平移了？还是缩放了？或是旋转了？）
3. STN网络预测出“变换前的数字”，即没经过变换的数字是怎样的
4. 最终进行分类预测

上图中，a是输入图片，b 是 STN 中的 localisation 网络检测到的物体区域，c是 STN 对检测到的区域进行线性变换后输出，d 是有 STN 的分类网络的最终输出。

2. STN网络结构

STN更准确地说应该是 STL(spatial transformer layer)，它就是网络中的一层，并且可以在任何两层之间添加一个或者多个。如下图所示，STL由三部分组成，分为是局部网络(localisation net)、网格生成器(grid generator, 参数化网络采样))和差分图像采样。

上图中，输入的特征图U被局部网络处理得到参数 $\theta$，然后经过网格生成器得到采样器，映射到原图U上，从而得到输出V。

2.1 局部网络

局部网络 用于预测输入数字的变换（是平移了？还是缩放了？或是旋转了？）

2.2 网格生成器

网格生成器获得输出特征图坐标点在输入特征图中坐标点的对应位置。

五、CBAM：混合注意力

BAM：[3]

CBAM：[4]

github代码：attention-module

注意力机制BAM和CBAM详细解析（附代码）

CBAM:通道注意力和空间注意力的集成者

CBAM网络简介

卷积注意力模块（Convolutional Block Attention Module，CBAM）是混合注意力机制的代表模型，它包括通道注意力模块、和空间注意力模块。

CBAM的模型结构如下，它对输入的特征图，首先进行通道注意力模块处理；得到的结果，再经过空间注意力模块处理，最后得到调整后特征。

六、参考文献

[1] Hu J, Shen L, Sun G. Squeeze-and-excitation networks[C]//Proceedings of the IEEE conference on computer vision and pattern recognition. 2018: 7132-7141.

[2] Jaderberg M, Simonyan K, Zisserman A. Spatial transformer networks[J]. Advances in neural information processing systems, 2015, 28.

[3] Park J, Woo S, Lee J Y, et al. Bam: Bottleneck attention module[J]. arxiv preprint arxiv:1807.06514, 2018.

[4] Woo S, Park J, Lee J Y, et al. Cbam: Convolutional block attention module[C]//Proceedings of the European conference on computer vision (ECCV). 2018: 3-19.