taindp98 / active-learning-object-detection

🚌 Official repository of Active Learning for Object Detection [Published in IEEE BigData 2023 special session Machine Learning]

Hierarchical Uncertainty Aggregation and Emphasis Loss for Active Learning in Object Detection

This repository contains supplemental materials for the IEEE BigData 2023 special session Machine Learning paper: [Paper]

Object detection has achieved remarkable advancements through the utilization of deep neural networks; however, its heavy reliance on extensive labeled data remains a significant challenge. We first delve into active learning strategies that leverage the power of uncertainty estimation and detector resilience. Departing from approaches biased toward high-performing classes, we present an innovative solution known as Hierarchical Uncertainty Aggregation and Emphasis Loss (HUALE). This approach introduces hierarchical uncertainty aggregation, which effectively retrieves and ranks unlabeled images for precise image selection. Secondly, we introduce the Emphasis loss, a novel loss augmentation to object detection training, and combine it with the EIoU loss and Synchronic IoU-Localization loss to elevate the accuracy of bounding box localization and address non-convexity challenges. Our method, built upon the Faster R-CNN framework, consistently outperforms the baseline (random selection) with average mAP (%) improvements of 4.53/3.44/1.57 across the PASCAL VOC 2007, PASCAL VOC 2012, and MS COCO datasets.

Table of Contents

• Update
• Results

Update

• June 2024: We release 1st version supplementary.

Results

Citation

Please cite this paper if it helps your research:

@inproceedings{nguyen2023hierarchical,
  title={Hierarchical Uncertainty Aggregation and Emphasis Loss for Active Learning in Object Detection},
  author={Nguyen, Tai and Nguyen, Khoa and Nguyen, Thanh and Nguyen, Tri and Nguyen, Anh and Kim, Karrman},
  booktitle={2023 IEEE International Conference on Big Data (BigData)},
  pages={5311--5320},
  year={2023},
  organization={IEEE}
}