HSF-Net: A Hierarchical Edge Enhancement and Sparse-Aware Fusion Network for Small Object Detection in UAV Aerial Imagery
Open Access
Abstract
Object detection in unmanned aerial vehicle (UAV) imagery is severely challenged by extremely small object scales, cluttered backgrounds, and pronounced foreground–background imbalance, which jointly degrade the accuracy of general-purpose detectors. This paper presents HSF-Net, a small-object detection network built upon YOLOv11s through three complementary enhancements. First, a Hierarchical Edge Enhancement Module (HEEM) employs orthogonal strip-convolution decomposition with zero-initialized residual fusion to enhance and re-weight the fine-scale edge and texture cues that are progressively attenuated in deep convolutional backbones. Second, a Sparse-Aware Feature Modulation (SAFM) module replaces concatenation-based fusion in the top-down neck pathway, coupling a sparse foreground gate with channel-wise scale modulation to confine cross-scale aggregation to object-bearing regions. Third, the detection head is restructured from {P3, P4, P5} to {P2, P3, P4}, introducing a high-resolution pathway for tiny targets while removing the original low-resolution P5 branch. On VisDrone, HSF-Net attains 46.4% mAP50 and 28.5% mAP50–95, exceeding the baseline by 7.1 and 4.8 percentage points, respectively, while reducing the parameter count from 9.4 million to 3.6 million. The model achieves an end-to-end throughput of 131.3 FPS on an NVIDIA RTX 3090, although the high-resolution P2 branch increases the computational cost to 39.5 GFLOPs. After dataset-specific training and evaluation on the markedly different TinyPerson benchmark, HSF-Net outperforms YOLOv11s by 5.2 percentage points in mAP50, indicating that its relative performance advantage persists under a substantially different target-scale distribution and maritime background.Keywords
Funding Information
- the Defense Industrial Technology Development Program (JCKY2025605C007)
This publication has 42 references indexed in Scilit:
- LYA-YOLO: a lightweight and accurate YOLO model in drone aerial image scenesExpert Systems with Applications, 2026
- LAM-YOLO: Drones-based small object detection on lighting-occlusion attention mechanism YOLOComputer Vision and Image Understanding, 2025
- MSD-YOLO11n: an improved small target detection model for high precision UAV aerial imageryJournal of King Saud University Computer and Information Sciences, 2025
- EBC-YOLO: a remote sensing target recognition model adapted for complex environmentsEarth Science Informatics, 2025
- LUD-YOLO: A novel lightweight object detection network for unmanned aerial vehicleInformation Sciences, 2024
- Deep learning-based object detection in low-altitude UAV datasets: A surveyImage and Vision Computing, 2020
- End-to-End Object Detection with Transformers2020
- CBAM: Convolutional Block Attention Module2018
- SSD: Single Shot MultiBox Detector2016
- Microsoft COCO: Common Objects in Context2014