Instance-Aware Uncertainty for Active Learning in Object Detection
Active learning (AL) can significantly reduce the overall amount of training data required to achieve near-optimal performance. Despite the notable advancements in active learning for image recognition, there remains a lack of a lightweight active learning method specifically designed for object detection. In this paper, we embark on an investigation into the uncertainty of predicted instances’ boxes within the object detection process, addressing three distinct types of uncertainty related to position, size, and category in object detection. We define proposals after regression as instances and feature instances belong to one object as instances bags. We estimate image uncertainty by calculating the position, size, and category variances of these instance bags. In object detection tasks, experiments validate that the proposed method significantly reduces computational complexity and outperforms state-of-the-art methods by a substantial margin.
Authors
- Hongyi Tang
- Wenting Ma (ORCID: https://orcid.org/0000-0002-1245-9885)
- Xiaohang Yuan (ORCID: https://orcid.org/0000-0001-7883-576X)
- Zhiheng Zhou (ORCID: https://orcid.org/0000-0003-4040-0175)
- Zhenjie Yao (ORCID: https://orcid.org/0000-0003-1027-637X)
- Hao Yuan
- Meng Guo (ORCID: https://orcid.org/0009-0003-3506-0636)
- Zhipeng Zhang (ORCID: https://orcid.org/0009-0004-8957-4648)
Institutions
- China Mobile (China) (CN)
- Chinese Academy of Sciences (CN)
- Institute of Microelectronics (CN)
- South China University of Technology (CN)
Publication Details
- Published
- 2024-09-27
- DOI
- https://doi.org/10.1109/icip51287.2024.10647710
- Citations
- 6
- Primary Topic
- Machine Learning and Algorithms
- Type
- article
- Field-Weighted Citation Impact
- 2.08