Employing Feature Mixture for Active Learning of Object Detection

Licheng Zhang,Siew-Kei Lam,Dingsheng Luo,Xihong Wu
DOI: https://doi.org/10.1016/j.neucom.2024.127883
IF: 6
2024-01-01
Neurocomputing
Abstract:Active learning aims to select the most informative samples for annotation from a large amount of unlabeled data, in order to reduce time-consuming and labor-intensive manual labeling efforts. Although active learning for object detection has made substantial progress in recent years, developing an accurate and efficient active learning algorithm for object detection remains a challenge. In this paper, we propose a novel unsupervised active learning method for deep object detection. This is based on our hypotheses that an object is more likely to be wrongly predicted by the model, if the prediction changes when its feature representations are slightly mixed by another feature representations at a very small ratio. Such unlabeled samples can be regarded as informative samples that can be selected by active learning. Our method employs base representations of all categories generated from the object detection network to examine the robustness of every detected object. We design a scoring function to calculate the informative score of each unlabeled image. We conduct extensive experiments on two public datasets, i.e., PASCAL VOC and MS-COCO. Experiment results show that our approach consistently outperforms state-of-the-art single-model based methods with significant margins. Our approach also performs on par with multi-model based methods, at much lesser computational cost.
What problem does this paper attempt to address?