187 / 2024-03-12 10:07:59

Global-margin Uncertainty and Collaborative Sampling for Active Learning in Complex Aerial Images Object Detection

Aerial images, object detection, active learning, global-margin uncertainty, collaborative sampling

Object detection in aerial images based on deep learning requires a large amount of labeled data, whereas manual annotation of aerial images is time-consuming and laborious. As a branch of machine learning, active learning can help humans find the valuable samples by designing some corresponding query strategies, which effectively reduces the cost of manual labeling. However, objects in aerial images are usually small, dense, and accompanied by the interference from complex backgrounds. These brings considerable challenges for active learning in selecting high-value aerial image samples. Currently, there is a relatively lack of study on active learning for aerial images object detection. Therefore, this paper proposes an novel active learning method, using global-margin uncertainty (GMU) and collaborative sampling (CS) to find out the high valuable aerial image samples to reduce the annotation cost and improve the training efficiency of models. In GMU, the predicted scores of categories are applied to calculate the global uncertainty and margin uncertainty of unlabeled aerial images, then those aerial images with high uncertainty scores are selected as the candidate samples. In CS, we train a main model and an auxiliary model respectively to detect the candidate samples, where the samples with large differences in detection results of the two models are selected for manual annotation. The experiments conduct on VisDrone2019 and DOTA-v1.5 datasets, which showes that the proposed method has a better performance compared with several state-of-the-art active learning methods.