最新的目标检测的深度架构 | 参数少一半、速度快3倍+(文末福利)

2019 年 8 月 30 日 计算机视觉战队


Matrix Nets

这是一个「矩阵网络」,它的参数少、效果好、训练快、显存占用低


简要介绍

研究作者提出了 Matrix Net (xNet),一种用于目标检测的新深度架构。xNets将具有不同大小尺寸和纵横比的目标映射到网络层中,其中目标在层内的大小和纵横比几乎是均匀的。因此,xNets提供了一种尺寸和纵横比感知结构。
研究者利用xNets增强基于关键点的目标检测。新的的架构实现了比任何其他单镜头检测器的时效性高,具有47.8的mAP在MS COCO数据集,同时使用了一半的参数而且相比于第二好框架,其在训练上快了3倍。
简单结果展示
上图所示,xNet的参数及效率要远远超过其它模型。其中FSAF在基于锚点的检测器中效果是最好的,它超过了经典的RetinaNet。研究者提出的模型在参数量类似的情况下性能超过了所有其他single-shot架构。
背景及现状

目标检测是计算机视觉中最广泛研究的任务之一,具有许多应用到其他视觉任务,如目标跟踪、实例分割和图像字幕。目标检测结构可以分为两类:single-shot检测器和two-stage检测器。Two-stage检测器利用区域候选网络找到固定数量的目标候选,然后使用第二个网络来预测每个候选的分数并改进其边界框。

常见的Two-stage算法

Single-shot检测器也可以分为两类:基于锚的检测器和基于关键点的检测器。基于锚的检测器包含许多锚边界框,然后预测每个模板的偏移量和类。最著名的基于锚的体系结构是RetinaNet,它提出了focal损失函数,以帮助纠正锚边界框的类不平衡。性能最好的基于锚的探测器是FSAF。FSAF将锚基输出与无锚输出头集成在一起,以进一步提高性能。

另一方面,基于关键点的检测器可以预测左上角和右下角的热图,并使用特征嵌入将它们匹配起来。最初的基于关键点的检测器是CornerNet,它利用一个特殊的coener池化层来准确地检测不同大小的目标。从那时起,Centerne通过预测目标中心和角,大大改进了CornerNet体系结构。
Matrix Nets

下图所示为Matrix nets(xNets),使用分层矩阵建模具有不同大小和丛横比的目标,其中矩阵中的每个条目i、j 表示一个层 li,j,矩阵左上角层 l1,1 中宽度降采样2^(i-1),高度降采样2^(j-1)。对角层是不同大小的方形层,相当于一个 FPN,而非对角层是矩形层(这是xNets所特有的)。 层l1,1是最大的层,每向右一步,层宽度减半,而每向下一步高度减半。

例如,层l3,4是层l3,3宽度的一半。对角层建模宽高比接近方形的目标,而非对角层建模宽高比不接近方形的目标。接近矩阵右上角或左下角的层建模宽高比极高或极低的目标。这类目标非常罕见,所以可以对它们进行剪枝以提升效率。

1
Layer Generation

生成矩阵层是一个关键的步骤,因为它影响模型参数的数量。参数越多,模型表达越强,优化问题越困难,因此研究者选择尽可能少地引入新的参数。对角线层可以从主干的不同阶段获得,也可以使用特征金字塔框架。上三角层是在对角线层上施加一系列具有1x2步长的共享3x3卷积得到的。类似地,左下角层是使用具有2x1步长的共享3x3卷积得到的。参数在所有下采样卷积之间共享,以最小化新参数的数量。

2
层范围

矩阵中的每个层都对具有一定宽度和高度的目标进行建模,因此我们需要定义分配给矩阵中每个层的目标的宽度和高度范围。范围需要反映矩阵层特征向量的感受野。矩阵中向右的每一步都有效地使水平维度中的感受野加倍,而每一步都使垂直维度上的感受场加倍。因此,当我们在矩阵中向右或向下移动时,宽度或高度的范围需要加倍。一旦定义了第一层l1,1的范围,我们就可以使用上述规则为矩阵层的其余部分生成范围。

3
Matrix Nets的优势
Matrix Nets的主要优点是它们允许方形卷积核准确地收集有关不同纵横比的信息。在传统的目标检测模型中,如RetinaNet,需要一个方形卷积核来输出不同的长宽比和尺度。这与直觉相反,因为不同方面的边界框需要不同的背景。在Matrix Nets中,由于每个矩阵层的上下文发生变化,所以相同的方形卷积核可以用于不同比例和长宽比的边界框。

由于目标大小在其指定的层内几乎是均匀的,因此与其他架构(例如FPN)相比,宽度和高度的动态范围更小。因此,回归目标的高度和宽度将变得更容易优化问题。最后Matrix Nets可用作任何目标检测架构、基于锚或基于关键点、one-shot或two-shots检测器。

Matrix Nets 用于基于关键点的检测

在CornerNet被提出来的时候,其是为了替代基于锚点的检测,它利用一对角(左上角和右下角)来预测边界框。 对于每个角来说,CornerNet可预测热图、偏移量和嵌入。

图是基于关键点的目标检测框架—— KP-xNet,它包含4个步骤。

  • (a-b):使用了xNet的主干;

  • (c):使用了共享输出子网络,而针对每个矩阵层,预测了左上角和右下角的热图和偏移量,并在目标层内对它们进行中心点预测;

  • (d):利用中心点预测匹配同一层中的角,然后将所有层的输出与soft非极大值抑制结合,从而得到最终输出。


实验结果
下表展示了在MS COCO数据集上的结果:

研究者还比较了新提出的模型与其他模型在不同的backbones 上基于参数的数目。 在第一张图中,我们发现KP-xNet在所有参数级别上都优于所有其他结构。研究者 认为这是因为KP-xNet使用了一种尺度和纵横比感知的体系结构。
论文地址: https://arxiv.org/pdf/1908.04646.pdf

如果想加入我们“计算机视觉战队”,请扫二维码加入学习群。计算机视觉战队主要涉及机器学习、深度学习等领域,由来自于各校的硕博研究生组成的团队,主要致力于人脸检测、人脸识别,多目标检测、目标跟踪、图像分割等研究方向。

我们开创一段时间的“计算机视觉协会”知识星球,也得到很多同学的认可,我们定时会推送实践型内容与大家分享,在星球里的同学可以随时提问,随时提需求,我们都会及时给予回复及给出对应的答复。

登录查看更多
15

相关内容

深度学习目标检测方法及其主流框架综述
专知会员服务
146+阅读 · 2020年6月26日
【文献综述】深度学习目标检测方法及其主流框架综述
专知会员服务
117+阅读 · 2020年6月26日
专知会员服务
159+阅读 · 2020年4月21日
CVPR2020 | 商汤-港中文等提出PV-RCNN:3D目标检测新网络
专知会员服务
43+阅读 · 2020年4月17日
专知会员服务
41+阅读 · 2020年2月20日
深度学习算法与架构回顾
专知会员服务
78+阅读 · 2019年10月20日
总结-CNN中的目标多尺度处理
极市平台
17+阅读 · 2019年7月24日
角网络——目标检测(文后有paper地址及源码)
计算机视觉战队
5+阅读 · 2019年6月5日
三分支网络——目前目标检测性能最佳网络框架
人工智能前沿讲习班
6+阅读 · 2019年3月5日
【泡泡图灵智库】基于CPU的实时6D物体姿态估计(arXiv)
泡泡机器人SLAM
12+阅读 · 2019年1月26日
Mesh R-CNN
Arxiv
4+阅读 · 2019年6月6日
Scale-Aware Trident Networks for Object Detection
Arxiv
4+阅读 · 2019年1月7日
Arxiv
3+阅读 · 2018年10月18日
Arxiv
8+阅读 · 2018年5月17日
Arxiv
3+阅读 · 2018年3月27日
Arxiv
7+阅读 · 2017年12月26日
Arxiv
4+阅读 · 2016年12月29日
VIP会员
相关VIP内容
相关论文
Mesh R-CNN
Arxiv
4+阅读 · 2019年6月6日
Scale-Aware Trident Networks for Object Detection
Arxiv
4+阅读 · 2019年1月7日
Arxiv
3+阅读 · 2018年10月18日
Arxiv
8+阅读 · 2018年5月17日
Arxiv
3+阅读 · 2018年3月27日
Arxiv
7+阅读 · 2017年12月26日
Arxiv
4+阅读 · 2016年12月29日
Top
微信扫码咨询专知VIP会员