Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Multiple instance active learning for object detection

About

Despite the substantial progress of active learning for image recognition, there still lacks an instance-level active learning method specified for object detection. In this paper, we propose Multiple Instance Active Object Detection (MI-AOD), to select the most informative images for detector training by observing instance-level uncertainty. MI-AOD defines an instance uncertainty learning module, which leverages the discrepancy of two adversarial instance classifiers trained on the labeled set to predict instance uncertainty of the unlabeled set. MI-AOD treats unlabeled images as instance bags and feature anchors in images as instances, and estimates the image uncertainty by re-weighting instances in a multiple instance learning (MIL) fashion. Iterative instance uncertainty learning and re-weighting facilitate suppressing noisy instances, toward bridging the gap between instance uncertainty and image-level uncertainty. Experiments validate that MI-AOD sets a solid baseline for instance-level active learning. On commonly used object detection datasets, MI-AOD outperforms state-of-the-art methods with significant margins, particularly when the labeled sets are small. Code is available at https://github.com/yuantn/MI-AOD.

Tianning Yuan, Fang Wan, Mengying Fu, Jianzhuang Liu, Songcen Xu, Xiangyang Ji, Qixiang Ye (1) __INSTITUTION_7__ University of Chinese Academy of Sciences, Beijing, China, (2) Noah's Ark Lab, Huawei Technologies, Shenzhen, China, (3) Tsinghua University, Beijing, China)• 2021

Related benchmarks

TaskDatasetResultRank
Object DetectionMS-COCO--
208
Defect DetectionIndustrial Thin-Film Defect Dataset
F1 Score75.3
75
Object DetectionPascal VOC
mAP@5067.8
54
Object DetectionMS COCO (train val)
AP32.27
37
Showing 4 of 4 rows

Other info

Code

Follow for update