Paper Reading: SENet
介绍
本文研究了网络设计的另一个方面——特征在通道之间的关系
我们提出了一个新的架构单元,称之为Squeeze-and-Excitation(SE
) 块,其目的是通过明确地建模特征图通道之间的依赖关系来提高网络产生的表征的质量,利用全局信息,有选择地强调和压制特征。
SE
块的结构如下图,
是任意给定的变换(例如一个卷积),他将输入 映射到特征图 ,其中
特征图 首先经过一个名为 Squeeze
的操作,它通过在空间维度( )上聚合特征图来生成通道特征响应的全局分布的 Embedding
,使网络的全局接收场的信息被其所有层使用。
聚合特征图后(经过 Squeeze
后),跟随一个名为 excitation
的操作,它采用一种简单的自我门控机制,将嵌入作为输入,产生一组对于每个通道的权重
这些权重最终被应用于特征图 ,生成SE
块的输出,随货直接输入网络的后续层。
虽然 SE
块是通用的,但在整个网络中,它在不同深度作用也不同。
在前几层,它以一种与类别无关的方式突出了信息特征,加强共享的低层次表征
在后面的层中,SE块变得越来越对于不同类别专门化,并以高度特定类的方式对不同的输入作出反应。
因此,由 SE
块进行的特征重新校准的增益可以通过网络累积。
SE
块的结构很简单,可以直接在现有最先进的网络架构中使用,方法是用 SE
块替换组件,即可显著提升网络性能,且只增加了少量计算成本。
Squeeze-and-excitation 块
SE
块是一个建立在一个变换 上的计算单元。
将输入 映射到特征图 。
我们取 为一个卷积操作,用 来表示所学到的卷积核的集合。
我们将输出写作 ,其中
这里 代表卷积,,。是一个 2D 空间下的核,代表 中的单独一个通道,作用于 的相应通道上。为了简化符号省略了偏置项。
由于输出是通过所有通道的求和产生的,通道依赖性被隐含地嵌入,,但与滤波器捕捉到的局部空间相关性杂糅。
卷积所建模的通道关系本质上是隐式的且局部的,所以我们希望卷积特征的学习能够通过明确地建模通道的相互依赖性而得到加强,这样网络就能够提高其对信息特征的敏感性,这些特征可以被后续的转化所利用。
因此我们想让它获得全局信息,并重新校准特征图。
Squeeze
:全局信息 Embedding
每一个学习到的滤波器都在一个局部的接受领域内运行,因此,变换输出的每个单元U
都无法利用这个区域以外的上下文信息。
为了缓解这个问题,我们将全局空间信息压缩到一个通道描述符中。
我们通过使用全局平均池化来产生逐通道的统计数据来实现的。
统计数据 ,他通过收缩U的空间维度 产生。
的第 个元素的计算方法如下:
Excitation
:自适应重新校准
为了利用在挤压操作中汇总的信息,我们紧接着进行了第二个操作,使用一个简单的门控机制,采用sigmoid激活函数:
其中 是 ReLU,,
为了限制模型的复杂性和帮助推广,我们通过在非线性周围的两个 FC层
形成一个瓶颈来对门控机制进行参数化。即具有降低率 的降维层(本文中)。 一个ReLU,然后是一个增维层,还原到变换 输出的 的通道维度。
块的最终输出是通过用重新缩放得到的:
其中 ,是和特征图 按通道相乘得到.