Hierarchical Uncertainty Aggregation and Emphasis Loss for Active Learning in Object Detection
Nguyen Tai, Nguyen Khoa, Nguyen, Thanh, Nguyen Tri, Nguyen Anh, Kim Karrman
Unverified — Be the first to reproduce this paper.
ReproduceAbstract
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.