SODRS: Semisupervised Learning for One-Stage Small Object Detection in Remote Sensing Images

Abstract

Small object detection in remote sensing images faces challenges such as weak features, vulnerability to interference, and limited object visibility. These factors have made it a long-standing technical issue. Traditional object detection methods often perform poorly in this field, especially when dealing with high-resolution remote sensing images, where detection accuracy and robustness fail to meet practical application needs. Furthermore, due to the large volume of remote sensing image data and the high cost of annotation, effectively utilizing unlabeled data has become the key to enhancing model detection performance. To resolve these problems, this article proposes a one-stage semisupervised learning method for small object detection in remote sensing images called SODRS. This method uses FCOS as the baseline and introduces the squeeze-and-excitation attention mechanism to enhance feature representation, the soft focal loss module to optimize the class ambiguity between objects and background, and the confident pseudolabeling strategy to improve the quality of pseudolabels. Finally, the conditional random fields-based label refinement is applied to postprocess the predicted labels, improving spatial relationships and dependencies among objects. Experimental results demonstrate that SODRS excels at detecting small objects in complex remote sensing scenarios, with higher accuracy and robustness than existing classical methods. Notably, SODRS can accurately distinguish objects even when densely distributed, close to each other, or overlapping. This demonstrates the application potential of one-stage semisupervised methods in small object detection in remote sensing images.

Keywords

• Feature enhancement
• label refinement
• one-stage semisupervised learning (SSL)
• remote sensing images
• small object detection