HyperAIHyperAI

Command Palette

Search for a command to run...

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

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


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供