1 个月前

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

Abhinav Shrivastava; Abhinav Gupta; Ross Girshick
基于区域的在线难例挖掘训练目标检测器
摘要

目标检测领域在基于区域的卷积神经网络(ConvNets)的推动下取得了显著进展,但其训练过程仍然包含许多启发式方法和需要耗费大量成本调整的超参数。本文介绍了一种简单而令人惊讶地有效的在线难例挖掘(OHEM)算法,用于训练基于区域的卷积神经网络检测器。我们的动机始终如一——检测数据集中包含了大量的简单示例和少量的困难示例。自动选择这些困难示例可以提高训练的有效性和效率。OHEM是一种简单直观的算法,它消除了常用的一些启发式方法和超参数。更重要的是,该算法在PASCAL VOC 2007和2012等基准测试中表现出一致且显著的性能提升。随着数据集变得更大、更复杂,其有效性也进一步增强,这一点在MS COCO数据集的结果中得到了验证。此外,结合领域内的其他互补进展,OHEM分别在PASCAL VOC 2007和2012上取得了78.9%和76.3%的平均精度均值(mAP),达到了当前最佳水平。