17 天前

动态头:通过注意力机制统一目标检测头

Xiyang Dai, Yinpeng Chen, Bin Xiao, Dongdong Chen, Mengchen Liu, Lu Yuan, Lei Zhang
动态头:通过注意力机制统一目标检测头
摘要

目标检测中定位与分类任务的复杂耦合关系,推动了相关方法的蓬勃发展。以往研究虽致力于提升各类目标检测头的性能,却未能提供一个统一的视角。本文提出一种新颖的动态检测头框架,通过引入注意力机制,实现对各类检测头的统一建模。该方法通过协同整合多层次特征间的自注意力机制以实现尺度感知(scale-awareness),空间位置间的自注意力机制以实现空间感知(spatial-awareness),以及输出通道内部的自注意力机制以实现任务感知(task-awareness),在不引入任何计算开销的前提下,显著增强了检测头的表征能力。大量实验验证了所提动态检测头在COCO基准上的有效性与高效性。采用标准的ResNeXt-101-DCN骨干网络,本方法在多个主流检测器上实现了显著性能提升,并达到54.0 AP的新纪录。进一步结合最新的Transformer骨干网络及额外训练数据,可将当前COCO最佳结果提升至60.6 AP,创下新纪录。代码将开源,地址为:https://github.com/microsoft/DynamicHead。