Semantic Consistency and Uncertainty-Driven Small-Object Detection for Class Imbalance

Abstract

In aerial image small-object detection, complex imaging perspectives, arbitrary object orientations, and long-tailed category distributions jointly exacerbate sample imbalance, which significantly degrades detection stability and leads to frequent misclassification of minority categories. To address these challenges, this paper proposes a novel training framework termed SCUD. Specifically, in the label noise suppression strategy (LNSS), a contrastive learning mechanism based on semantic consistency is introduced to constrain the aggregation of similar samples in the feature space, thereby reducing the adverse impact of noisy samples on model optimization. In addition, a scale-aware resampling strategy (SARS) is designed to alleviate noise amplification and overfitting caused by excessive repetition of small objects during training. Furthermore, an adaptive instance selection mechanism (AISM) is developed by jointly modeling prediction uncertainty and global statistical priors, enabling the model to dynamically emphasize learning from informative samples. Extensive experiments are conducted on two publicly available unmanned aerial vehicle (UAV) aerial image datasets to validate the effectiveness of the proposed approach. The proposed method achieves an mAP50 of 70.7% on the DOTA-v1.0 dataset and 88.1% on the DIOR dataset. Notably, the detection accuracy of several rare categories is significantly improved, further demonstrating the effectiveness of the proposed method in addressing sample imbalance in aerial image small-object detection.