CNN能同时兼顾速度与准确度吗?CMU提出AdaScale

栏目: 数据库 · 发布时间: 5年前

内容简介:论文地址:https://arxiv.org/abs/1902.02910在机器人和自动汽车等具备视觉能力的自动系统中,视频目标检测发挥着关键的作用。为了提供可靠的操作,视频目标检测的速度和准确度都是重要的因素。我们在这篇论文中表明的关键见解是当涉及到图像缩放时,速度和准确度并无必要权衡。我们的结果表明将图像的尺寸重新调整到更低的分辨率时,有时会得到更好的准确度。基于这一观察,我们提出了一种全新的方法 AdaScale,可以自适应地选择输入图像的尺寸,从而同时提升视频目标检测的准确度和速度。我们在 Imag

论文:AdaScale: Towards Real-time Video Object Detection Using Adaptive Scaling

CNN能同时兼顾速度与准确度吗?CMU提出AdaScale

论文地址:https://arxiv.org/abs/1902.02910

在机器人和自动汽车等具备视觉能力的自动系统中,视频目标检测发挥着关键的作用。为了提供可靠的操作,视频目标检测的速度和准确度都是重要的因素。我们在这篇论文中表明的关键见解是当涉及到图像缩放时,速度和准确度并无必要权衡。我们的结果表明将图像的尺寸重新调整到更低的分辨率时,有时会得到更好的准确度。基于这一观察,我们提出了一种全新的方法 AdaScale,可以自适应地选择输入图像的尺寸,从而同时提升视频目标检测的准确度和速度。我们在 ImageNet VID 和 mini YouTube-BoundingBoxes 数据集上进行了实验,结果分别在加速 1.6 和 1.8 倍的情况下实现了 1.3 和 2.7 个百分点的平均精度均值(mAP)提升。此外,我们还将 ImageNet VID 数据集上当前最佳的视频加速工作提速了额外 1.25 倍,且 mAP 也略好一些。

引言

对于自动汽车、无人机和机器人等未来的自动智能体而言,视觉目标检测是视觉认知的一个基本构建模块。因此,为了构建性能可靠的系统,检测器必须要快速且准确。尽管目标检测非常适合静态图像(Dai et al., 2016; Girshick, 2015; He et al., 2014; Liu et al., 2016; Ren et al., 2015),但在视频目标检测方面还存在一些特有的挑战,包括由物体移动造成的运动模糊、相机对焦失败(Zhu et al., 2017a)以及自动智能体的实时速度限制。但是,除了这些难题之外,视频目标检测也会带来可以利用的新机会。之前一些关注视频目标检测的研究试图通过利用视频的一种独特特征来提升平均精度(Zhu et al., 2017a; Feichtenhofer et al., 2017; Kang et al., 2017),即时间一致性(连续帧有相似的内容)。另外,在速度方面,之前有研究(Zhu et al., 2017b; 2018b; Buckler et al., 2018)依靠这种时间一致性来降低独立的目标检测器所需的计算。类似地,我们的目标也是利用时间一致性,并使用一种名为自适应缩放测试(AdaScale/ adaptive-scale testing)的全新技术来同时提升独立目标检测器的速度和准确度。

输入图像的尺寸会同时影响基于 CNN 的现代目标检测器的速度和准确度(Huang et al., 2017)。之前与图像缩放相关的研究针对的是两个方向:(1)为了得到更好准确度的多尺寸测试,(2)为了实现更高速度的图像下采样。第一类的例子包括将图像调整为多个尺寸(图像金字塔)并使它们通过 CNN 以实现多个尺寸的特征提取(Dai et al., 2016; Girshick, 2015; He et al., 2014),然后通过一张单尺寸的输入图像生成的不同层来融合特征图(Lin et al., 2017a; Cai et al., 2016; Bell et al., 2016)。但是,相比于仅有单个尺寸的输入,这样的方法会引入额外的计算开销。第二类的例子包括通过调整输入图像尺寸的 Pareto 最优搜索(Lin et al., 2017b; Liu et al., 2016; Redmon & Farhadi, 2017; Huang et al., 2017)以及根据输入图像进行的动态图像尺寸调整(Chin et al., 2018)。但是,这样的方法的结果表明,在进行图像缩放时,更高速度的代价是准确度更低。

不同于之前的研究,我们发现 下采样 有时候有助于提升准确度。具体而言,图像下采样能带来两类提升:(1)减少假正例(false positive)的数量,而关注不必要的细节可能会引入假正例;(2)增加真正例(true positive)的数量,方法是通过将过大的目标缩放到目标检测器更有信心处理的尺寸。图 1 表明,在我们在 ImageNet VID 数据集上使用基于区域的全卷积网络(R-FCN)(Dai et al., 2016)目标检测器的实验中,下采样时得到的结果更好的图像。

CNN能同时兼顾速度与准确度吗?CMU提出AdaScale 图 1:下采样后的图像得到的检测结果更好的示例。蓝框是检测结果,数字是置信度。这个检测器是在 600(短边的像素)的单尺寸上训练的。(a) 和 (c) 列是在 600 尺寸上的测试结果,(b) 列是在 240 尺寸上的测试结果,(c) 列则是 480。

受此启发,我们的目标是将图像调整至它们的最佳尺寸,以同时得到更高的速度和准确度。在这项研究中,我们提出了 AdaScale 来提升独立目标检测器的准确度和速度。具体来说,我们使用当前帧来预测下一帧的最佳尺寸。我们在 ImageNet VID 和 mini YouTube-BoundingBoxes 数据集上进行了实验,结果分别在加速 1.6 和 1.8 倍的情况下实现了 1.3 和 2.7 个百分点的平均精度均值(mAP)提升。此外,通过结合在 ImageNet VID 数据集上当前最佳的视频加速工作(Zhu et al., 2017b),我们为其提速了额外 25%,且 mAP 也略好一些。

自适应缩放 

图 2 展示了 AdaScale 方法的概况。其中包含微调目标检测器、使用所得到的检测器生成最优的尺寸标签、使用所生成的标签训练尺寸回归器以及 AdaScale 在视频目标检测中的部署。

CNN能同时兼顾速度与准确度吗?CMU提出AdaScale 图 2:AdaScale 方法

CNN能同时兼顾速度与准确度吗?CMU提出AdaScale 图 3:决定最优尺寸。首先,根据 4 个尺寸选择相同数量的预测前景。然后,选择损失最低的尺寸作为最优尺寸。

CNN能同时兼顾速度与准确度吗?CMU提出AdaScale 图 4:尺寸回归模块

为了将自适应缩放(AdaScale)整合进视频环境中,我们施加了一个时间一致性假设。更确切地说,我们假设两个连续帧的最优尺寸是相近的,我们的实验结果也验证了这一假设。算法 1 是一个利用 AdaScale 进行视频目标检测的例子。

CNN能同时兼顾速度与准确度吗?CMU提出AdaScale 算法 1:在测试阶段使用 AdaScale 的伪代码

实验

CNN能同时兼顾速度与准确度吗?CMU提出AdaScale 图 5:几种类别的精度-回调曲线,MS/AdaScale 在 (a)(b)(c) 中有更好的性能,在 (d) 中性能相当,在 (e)(f) 中相较 SS/SS 更差

CNN能同时兼顾速度与准确度吗?CMU提出AdaScale 图 6:在验证集中的所有图像上,不同方法在这些类别上得到的归一化的真正例和假正例情况

CNN能同时兼顾速度与准确度吗?CMU提出AdaScale 图 7:在 ImageNet VID 数据集上与之前最佳方法的 mAP 和速度比较。将我们的 AdaScale 应用于 RFCN (Dai et al., 2016)、DFF (Zhu et al., 2017a) 和 SeqNMS (Han et al., 2016) 时都能实现进一步的速度和准确度提升。

CNN能同时兼顾速度与准确度吗?CMU提出AdaScale 图 8:SS/SS 和 MS/AdaScale 结果的定性比较。(a) 和 (c) 列是 SS/SS 得到的结果,(b) 和 (d) 是 MS/AdaScale 得到的结果。MS/AdaScale 使用的尺寸标注在黑底白字矩形框中。


以上所述就是小编给大家介绍的《CNN能同时兼顾速度与准确度吗?CMU提出AdaScale》,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对 码农网 的支持!

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

分布式算法导论

分布式算法导论

泰尔 / 霍红卫 / 机械工业出版社 / 2004年09月 / 39.0

分布式算法20多年来一直是倍受关注的主流方向。本书第二版不仅给出了算法的最新进展,还深入探讨了与之相关的理论知识。这本教材适合本科高年级和研究生使用,同时,本书所覆盖的广度和深度也十分适合从事实际工作的工程师和研究人员参考。书中重点讨论了点对点消息传递模型上的算法,也包括计算机通信网络的实现算法。其他重点讨论的内容包括分布式应用的控制算法(如波算法、广播算法、选举算法、终止检测算法、匿名网络的随机......一起来看看 《分布式算法导论》 这本书的介绍吧!

RGB转16进制工具
RGB转16进制工具

RGB HEX 互转工具

Markdown 在线编辑器
Markdown 在线编辑器

Markdown 在线编辑器

HSV CMYK 转换工具
HSV CMYK 转换工具

HSV CMYK互换工具