提升注意力模块性能!DCANet:学习卷积神经网络的连接注意力|ECC...
极市平台
共 4372字，需浏览 9分钟
·
2020-07-22 22:18
加入极市专业CV交流群，与 10000+来自港科大、北大、清华、中科院、CMU、腾讯、百度 等名校名企视觉开发者互动交流！
同时提供每月大咖直播分享、真实项目需求对接、干货资讯汇总，行业技术交流。关注 极市平台 公众号 ，回复 加群，立刻申请入群~
今天给大家分享一篇ECCV2020上关于注意力机制方面的优秀论文，该论文提出了深度连接注意力网络(DCANet)，这是一种新颖的设计，在不改变CNN模型内部结构的情况下可以大幅提升注意力模块的性能。这个工作也证明了注意力单元之间不同的连接方式和注意力模块间信息交流的重要性。
论文链接：https://arxiv.org/pdf/2007.05099.pdf摘要和创新点
虽然，自注意力机制在许多视觉任务中都显示出了优越的效果，但它每次却只考虑当前的特征，因此这种方式并不能充分利用注意力机制。在本文中，作者提出了深度连接注意力网络(DCANet)，这是一种新颖的设计，在不改变CNN模型内部结构的情况下提升注意力模块的性能。
为了达到这样的效果，作者将相邻的注意力模块连接起来，使得注意力模块之间可以互相进行信息流交换，DCANet将CNN模型中的所有注意块一起进行训练，提高了注意力学习的能力。而且DCANet是具有泛化性的。它不局限于特定的注意力模块或基础网络架构。在ImageNet和MS COCO基准测试上的实验结果表明，DCANet在所有测试用例中始终优于目前最先进的注意力模块，而且它的额外计算开销也是最少的。
本文的创新点总结如下：1、DCANet通过从前面的注意块中收集信息，并将其传递给下一个注意块，使得注意块间可以进行相互配合，从而提高了注意力模块的学习能力。
2、在不改变内部结构的情况下，研究了各种注意力模块并开发了一个通用注意力框架，其中注意块由三个组件组成：上下文提取、转换和融合。
3、DCANet比较简单、通用，而且能极大程度提高注意力模块性能，将DCANet应用于多个最先进的注意力模块和一些基础CNN架构上，实验结果表明，毫不夸张地说，经过DCANet模块增强的网络比所有原始的网络都要好。
DCA模块介绍
深度连接注意力网络，虽然在概念上其实是比较简单，但经验表明，这种网络是很强大的。本文通过分析各种注意力模块的内部结构，提出了一种不局限于特定注意块的通用连接方案。通过参数化的加法操作将之前的注意力特征与当前提取的注意力特征进行合并，保证了信息在各个注意力块之间以前馈的方式流动，有效地避免了在注意力模块间的信息的频繁变动。下图2是DCANet模块的展示：
1、Revisiting Self-Attention Blocks首先回顾几个流行的注意力模块来分析其内部结构。常见的做法是，通过横向添加额外的注意力块来增强CNN的基础架构。然而，不同的注意力块是针对不同的任务量身定制的，实现方式也是不同的。例如，SE块由两个完全连接的层组成，而GC块包括几个卷积层。因此，要直接提供一个标准的、足够通用的、能涵盖大多数注意块的连接方式是不容易的。
为了解决这个问题，作者研究了目前最先进的注意力模块，并总结了他们的处理方式和组成形式。根据最近注意力模块连接方式及其组件的一些启发(主要来自-SENet和NLNet)，研究了各种注意力模块并开发了一个通用注意力框架，其中注意块由三个组件组成：上下文提取、转换和融合。提取是一种简单的特征提取器，变换则是将提取的特征转化为新的非线性注意空间，融合是将注意和原始特征融合。这些组成部分是通用的，不局限于一个特别的注意块。图3为大家演示了这三个组件在四个比较流行的注意力块上的具体情况。
Extraction：提取主要就是从特征图中收集特征信息，对于给定的卷积块生成的feature map，其中 ，通过提取器g从X中提取特征:，其中是提取操作的参数，G是输出。当g是无参数操作时，就不需要了(比如池化操作)。g的灵活性使得G可以根据呈现不同的形状进行相适应的提取操作。例如，SENet和GCNet将feature map X视为一个向量()，而CBAM中的空间注意模块将feature map采集为一个张量()。Transformation：则是对提取到的特征进行处理，并将其转化到非线性的注意力空间。形式上，将t定义为特征变换操作，注意块的输出可以表示为。其中为变换操作中使用的参数，T为提取模块的输出。Fusion：将注意力映射与原始卷积块的输出融合在一起。注意引导输出可以表示为，其中i是feature map中的索引，“”表示一个融合函数;“”在设计为点积注意比例时执行元素向乘法，否则执行总和。2、Attention Connection
接下来，使用前面提到的注意组件提出了一个广义的注意力连接模式。不管实现细节如何，注意力模块的模型表示为:
如上一节所述，由转换组件生成的注意力特征图对于注意力学习至关重要。为了构造连接注意力，将先前的注意力映射输入当前转换组件，当前转换组件将先前的转换输出与当前提取输出合并。这种连接设计保证了当前转换模块既能学习提取的特征，又能学习之前的注意力信息。由此产生的注意力模块可以描述为:
其中i是特征的索引。在公式(3)中，可以看作是G的增强。Weighted Connection（加权连接）：加权求和主要是对直接连接进行了增广操作。为了避免引入额外的参数，计算重量使用α和β，连接函数此时可表示为：
与直接连接相比，加权连接在α和β之间产生了竞争关系。此外，它可以很容易地扩展到softmax形式。消融研究的实验结果(见表2a)表明，结果对连接模式不敏感，说明性能的提高更多地来自注意块之间的连接，而不是连接函数的具体形式。因此在方法中默认使用直接连接。3、Size MatchingCNN模型中的特征图在不同的阶段可能大小也不相同。因此，相应的注意力特征图的大小也可能发生变化，这种不匹配使得DCANet不可能在two-stage之间应用。为了解决这一问题，本文对通道上的注意力特征图的形状和空间维度进行了自适应匹配。对于通道，使用全连接层(然后是layer normalization和ReLU激活函数)来匹配size，从而使得通道转换为C通道，其中C'和C分别表示之前和当前通道的数量。为了清晰起见，本文为通道尺寸匹配引入的参数为，为了进一步减少注意力连接中的参数负担，本文用两个轻量级全连接层重新构造了全连接层；输出尺寸分别为C/r和C，其中r为缩减比。这一修改大大减少了引入参数的数量。关于通道匹配这一策略的实验影响见Table2e。所有的实验使用r = 16的全连接层来匹配信道大小。为了匹配空间分辨率，一个简单而有效的策略是采用平均池化层。本文将步幅和感受野大小设置为分辨率降低的尺度。Max-pooling在本文的方法中也很有效，但它只考虑部分信息而不是整个注意力信息。除了池化操作之外，另一种解决方案是learnable con volume操作。然而，作者认为它不适合本文目的，因为它引入了许多参数，并不能很好地推广。详细的空间分辨率尺寸匹配消融实验见Table2c。实验结果
本文在图像识别和目标检测任务详细评估了DCANet模型。并且在ImageNet和MS-COCO基准测试上的实验结果，证明了方法的有效性。继而进行了大量全面的消融实验，深入研究了DCANet网络内部性质。表1是在ImageNet验证集的详细实验结果展示。观察发现，与普通注意力模型相比，集成DCA模块在所有情况下都提高了分类精度。值得注意的是，这里比较的网络加入了注意力模块之后都会比原先基础网络性能更强。在测试的网络中，DCA-CBAM -ResNet50比 CBAM-ResNet50提高了0.51%的top-1精度，DCA-SE-MobileNetV2比SE-MobileNetV2提高了1.19%的top-1精度，但计算开销是相当的。这项改进实验证明了DCANet模块的效率。
消融实验结果展示如下：
下图4对ResNet50、CBAM -ResNet50和DCA进行了可视化实验比较，可以看到，使用DCA模块进行注意力增强的模型始终专注于图像中的关键部分，同时也注意到，最具判别性的特征部分在中间特征之间的变化很小。
在COCO数据集上的实验结果展示：
Conclusion
本文详尽地指出，目前对自注意力机制的研究还不够充分。为了更好地利用这个方法，本文提出了深度连接注意力网络(DCANet)，它通过增加注意力模块连接的方式使得注意力块之间的信息流在向前传播时能够进行自我调节。文章证明了DCANet可以用来对各种注意力方法进行持续的优化改进，并且在基于ImageNet基准测试方面，在CNN网络架构方面能达到最小的开销。此外，在MS-COCO数据集上的实验表明，DCANet在其他视觉任务上有很好的表现效果。DCANet模块以其优雅的设计和独特的前向反馈连接方式，使得它很容易与目前主流框架的各种注意力机制进行集成。
◎作者档案
Murufeng，一个紧跟前沿、乐于分享最新技术干货的DLer！个人公众号：深度学习技术前沿欢迎大家联系极市小编（微信ID:fengcall19）加入极市原创作者行列
△长按添加极市小助手
△长按关注极市平台，获取最新CV干货
觉得有用麻烦给个在看啦~
评论