CSSA-YOLO: A Clutter-Suppressed and Scale-Aware Framework for Robust Object Detection in UAV Imagery

The widespread deployment of unmanned aerial vehicles (UAVs) in remote sensing has highlighted the necessity for robust object detection methods in UAV imagery. However, high-altitude UAV imagery suffers from severe background clutter that obscures target discriminability and extreme scale variations that degrade fine-grained features. To address these challenges, we propose CSSA-YOLO, a clutter-suppressed and scale-aware detection framework built upon YOLOv9. Specifically, we project dense spatial features into a low-rank token space via a Semantic Bottleneck Module (SBM). This projection acts as an information bottleneck, suppressing the background clutter while robustly retaining critical target semantic and structural priors. Furthermore, we develop a Scale-Aware Complete-IoU (SA-CIoU) loss to tackle gradient attenuation for small objects. By analytically integrating a scale-aware modulation factor with a dynamic alignment mechanism into localization optimization, SA-CIoU shifts the optimization priority to the precise localization of small and hard-to-detect instances. Extensive experiments on the VisDrone2019 benchmark demonstrate the superiority of our approach, with CSSA-YOLO achieving an [email protected] of 46.0% and an [email protected]:0.95 of 28.4%, yielding an absolute 1.4% improvement over the YOLOv9 baseline. Furthermore, when integrated with a P2-enhanced YOLOv9 architecture, our method achieves a remarkable [email protected] of 49.5%. Notably, evaluations across diverse scenarios, including the infrared (IR) thermal HIT-UAV benchmark and PCB defect detection datasets, further demonstrate the generalizability and robustness of our framework.