arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 1604.03540cs.CVcs.LG
收藏

基于硬示例在线挖掘的区域目标检测器训练

Training Region-based Object Detectors with Online Hard Example Mining

Carnegie Mellon University(卡内基梅隆大学);Facebook AI Research(脸书人工智能研究院)

Abhinav Shrivastava, Abhinav Gupta, Ross Girshick

摘要:

目标检测领域在基于区域的ConvNets浪潮中取得了重大进展,但它们的训练过程仍然包括许多优化成本高昂的算法和超参数。我们提出了一个简单但令人惊讶的有效的在线硬示例挖掘(OHEM)算法,用于训练基于区域的ConvNet检测器。我们的动机和以往一样--检测数据集包含大量简单的示例和少量困难的示例。自动选择这些困难的例子可以使训练更加有效和高效。OHEM是一种简单直观的算法,它消除了几种常用的算法和超参数。但更重要的是,它在PASCAL VOC 2007和2012等基准测试中的检测性能得到了一致和显著的提升。它的有效性随着数据集变得更大和更困难而增加,正如MS COCO数据集上的结果所证明的那样。此外,结合该领域的互补性进展,OHEM在PASCAL VOC 2007和2012上分别获得了78.9%和76.3%的最新结果。

英文摘要:

The field of object detection has made significant advances riding on the wave of region-based ConvNets, but their training procedure still includes many heuristics and hyperparameters that are costly to tune. We present a simple yet surprisingly effective online hard example mining (OHEM) algorithm for training region-based ConvNet detectors. Our motivation is the same as it has always been -- detection datasets contain an overwhelming number of easy examples and a small number of hard examples. Automatic selection of these hard examples can make training more effective and efficient. OHEM is a simple and intuitive algorithm that eliminates several heuristics and hyperparameters in common use. But more importantly, it yields consistent and significant boosts in detection performance on benchmarks like PASCAL VOC 2007 and 2012. Its effectiveness increases as datasets become larger and more difficult, as demonstrated by the results on the MS COCO dataset. Moreover, combined with complementary advances in the field, OHEM leads to state-of-the-art results of 78.9% and 76.3% mAP on PASCAL VOC 2007 and 2012 respectively.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑