5 个月前

多实例主动学习在目标检测中的应用

多实例主动学习在目标检测中的应用

摘要

尽管在图像识别领域的主动学习已经取得了显著进展,但仍然缺乏专门针对目标检测的实例级主动学习方法。本文提出了一种多实例主动目标检测(Multiple Instance Active Object Detection, MI-AOD)方法,通过观察实例级不确定性来选择最具信息量的图像用于检测器训练。MI-AOD 定义了一个实例不确定性学习模块,该模块利用在已标注数据集上训练的两个对抗性实例分类器之间的差异,预测未标注数据集中的实例不确定性。MI-AOD 将未标注图像视为实例包,并将图像中的特征锚点视为实例,通过以多实例学习(Multiple Instance Learning, MIL)的方式重新加权这些实例来估计图像的不确定性。迭代的实例不确定性学习和重新加权有助于抑制噪声实例,从而弥合实例级不确定性和图像级不确定性之间的差距。实验验证了 MI-AOD 为实例级主动学习提供了一个坚实的基础。在常用的目标检测数据集上,MI-AOD 在显著的性能差距下优于现有最先进方法,特别是在已标注数据集较小时表现尤为突出。代码可在 https://github.com/yuantn/MI-AOD 获取。

代码仓库

yuantn/MI-AOD
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
active-object-detection-on-cocoRetinaNet
AP: (7.3, 13.8, 16.9, 19.1, 20.8) on 2% ~ 10%
active-object-detection-on-pascal-voc-07-12SSD
mAP: (53.62, 62.86, 66.83, 69.33, 70.80, 72.21, 72.84, 73.74, 74.18, 74.91) on 1k ~ 10k
active-object-detection-on-pascal-voc-07-12RetinaNet
mAP: (47.18, 58.41, 64.02, 67.72, 69.79, 71.07, 72.27) on 5% ~ 20%

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
多实例主动学习在目标检测中的应用 | 论文 | HyperAI超神经