COCO 54.7mAP!DetectoRS目标检测:改进主干网,成就新高度!

2020-06-05 16:09:26 浏览数 (1)

COCO 数据集上的目标检测精度的最高纪录已经有将近一年时间停留在53.3 mAP,曾经报告达到过这一高度的算法有:

1)Cascade Mask R-CNN(Triple-ResNeXt152, multi-scale) ;

2)ResNeSt-200DCN (multi-scale);

均使用了多尺度测试,

它们分别来自论文:

1)CBNet: A Novel Composite Backbone Network Architecture for Object Detection, AAAI 2020;

2)ResNeSt: Split-Attention Networks, arXiv 2020;

这两篇论文均为在主干网上做文章,尤其是后者,其提出的主干网ResNeSt不仅大幅提高了目标检测的精度,在语义分割、实例分割中也登顶COCO数据集。

这不由得使我们思考,现有目标检测、语义分割等的任务架构是不是已经较为完备,而研究更好的主干网、增强网络的表达能力是提升精度更好的选择?

今天新出的文章 DetectoRS: Detecting Objects with Recursive Feature Pyramid and Switchable Atrous Convolution ,则再一次在主干网上做文章,其在COCO数据集上的单尺度测试精度已经达到53.3mAP(53.3真是一个神奇的数字!),而多尺度测试中 DetectoRS (ResNeXt-101-32x4d, multi-scale) 达到目前所有文献报告的最高精度:54.7mAP!代码已开源。

不仅如此,在COCO数据集上的实例分割和全景分割任务中,DetectoRS 也一举登顶,成为目前目标检测、语义分割、全景分割领域的全能王者!

DetectoRS 作者信息:

第一作者 Siyuan Qiao目前是约翰斯霍普金斯大学博士生,第二作者是著名的DeepLab 系列算法主要作者谷歌公司研究员Liang-Chieh Chen。

主要思想

DetectoRS的思想来自成功的目标检测算法如Faster RCNN 、Cascade R-CNN所体现出的哲学:looking and thinking twice (无论是两阶段还是级联检测器,都体现出信息的反复利用和提精)作者是把这一思想应用于主干网的改进上,主要创新如图:

1)宏观层次,提出递归特征金字塔网络(RFP,Recursive Feature Pyramid),集成FPN的反馈连接到主干网,使得主干网再训练得到的特征更好的适应检测或分割任务。

2)微观层次,提出可切换的空洞卷积(SAC,Switchable Atrous Convolution)替换主干网上的标准卷积,使得模型可自适应选择感受野。

以上新技术带来的改进是显著的,作者将其应用于实例分割算法HTC,在其他不变的情况下,检测和分割精度提升明显,如图:

递归特征金字塔网络(RFP)

大家所熟知的特征金字塔网络(FPN)是将主干网不同阶段(stage)特征组成特征金字塔,如下图(a),而RFP则添加了对于目标检测任务的梯度向主干网方向回传,如下图(b)。

图中(c)是将这个递归的过程展开的示意图,FPN反馈连接到主干网每个阶段的第一个block,使用的是ASPP结构(图中绿色圆点),图中画出的是2-step 序列RFP,FPN之间也进行了特征融合(图中红色部分)。

反馈连接网络ASPP:

RPN中的融合模块:

作者最终实验中使用的就是2-step RFP。

这种递归特征金字塔网络有什么好处呢?很明显它能够使得目标检测的错误回传信息更直接的反馈调整主干网的参数。

可切换的空洞卷积(SAC)

空洞卷积可增大网络感受野,这已经被证明对检测和分割任务是有效的。

而作者提出的可切换的空洞卷积(Switchable Atrous Convolution)则使得网络训练时尺度选择更灵活,如下图:

不同空洞率的空洞卷积捕捉到不同感受野的目标,网络可以学习到一个开关,以自适应调整选择哪个感受野的卷积结果。

其网络结构如图:

作者将主干网所有标准卷积替换为可切换的空洞卷积,而且为了更方便地从ImageNet预训练模型给主干网赋值,作者发明了可兼容标准卷积的可切换空洞卷积,具体细节请查看原论文。

实验结果

作者在消融实验中对上述两种改进进行了效果评估,如图:

可见HTC算法 RFP或者HTC SAC均获得了大幅的精度提升,且相比较之下HTC SAC的提升幅度更大,而同时加上RFP与SAC,取得了最高的精度。

一些检测结果示例:

下图为DetectoRS使用不同类型主干网与其他SOTA算法的比较:

使用ResNeXt-101-32x4d主干网,DetectoRS 取得了 COCO 目标检测单尺度和多尺度目前的最高精度,分别达到53.3mAP和54.7mAP。

在COCO实例分割和全景分割任务中的结果:

同样获得了目前最高的精度。

速度:在NVIDIA TITAN RTX显卡上运行帧率3.9fps。

最后让我们一起回顾一下COCO数据集目标检测近几年的SOTA(请点击并横屏查看大图):

图片来自paperswithcode.com

DetectoRS 已经站在风口浪尖,不知下一个出现在最右上角的是谁呢?

论文地址:

https://arxiv.org/abs/2006.02334

代码地址:

https://github.com/joe-siyuan-qiao/DetectoRS

0 人点赞