YOLO-DSNet for Small Target Detection
Source abstract
Abstract
Small target detection in Unmanned Aerial Vehicle (UAV) applications is often plagued by inherent challenges such as small object sizes, sparse information, and complex background interference. Traditional detection algorithms and existing YOLO series models suffer from limitations in detection accuracy and fine-grained detail preservation. To address this, this paper proposes YOLO-DSNet, a small target detection network based on YOLOv13n. First, we introduce the dual-stream attention module (DSAM), which enhances discriminative features by leveraging bidirectional context modeling. Second, we design the Multi-scale Attention C2f (MSA-C2f) module—an adaptive architecture that optimizes feature extraction via multi-scale enhancement, effectively preserving and integrating small target information. Finally, through dataset augmentation, we significantly improve the model’s detection performance. The proposed YOLO-DSNet achieves a mAP@0.5 improvement from 30.8% to 40.1% on the VisDrone2019 dataset with only 0.8 million additional parameters, yielding a 30% accuracy gain while increasing computational overhead by merely 11.6 Gigaflops (GFLOPs). Experiments demonstrate YOLO-DSNet’s effectiveness in small target detection tasks such as UAV aerial photography and remote sensing imagery, successfully balancing accuracy and efficiency with high practical value.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Advanced Neural Network Applications · Infrared Target Detection Methodologies · Advanced SAR Imaging Techniques
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Vecton AI, Sydney, NSW 2000, Australia
School of Computer Science and Engineering, University of New South Wales (UNSW), Sydney, NSW 2052, Australia
Computer Network Information Center, Chinese Academy of Sciences, Beijing 100190, China
References (25)
Show source-provided references
1. Faster R-CNN: Towards real-time object detection with region proposal networks10.1109/TPAMI.2016.2577031 ↗
2. Lin, T.-Y., Goyal, P., Girshick, R., He, K., and Dollár, P. (2017, January 22–29). Focal loss for dense object detection. Proceedings of the IEEE International Conference on Computer Vision, Venice, Italy.10.1109/ICCV.2017.324 ↗
3. Redmon, J., and Farhadi, A. (2018). YOLOv3: An incremental improvement. arXiv.
4. Lin, T.-Y., Dollár, P., Girshick, R., He, K., Hariharan, B., and Belongie, S. (2017, January 21–26). Feature pyramid networks for object detection. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Honolulu, HI, USA.10.1109/CVPR.2017.106 ↗
5. Liu, S., Qi, L., Qin, H., Shi, J., and Jia, J. (2018, January 18–23). Path aggregation network for instance segmentation. Proceedings of the 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, USA.10.1109/CVPR.2018.00913 ↗
6. Hu, J., Shen, L., and Sun, G. (2018, January 18–23). Squeeze-and-excitation networks. Proceedings of the 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, USA.10.1109/CVPR.2018.00745 ↗
7. Woo, S., Park, J., Lee, J.-Y., and Kweon, I.S. (2018). CBAM: Convolutional block attention module. Computer Vision—ECCV 2018, Springer.10.1007/978-3-030-01234-2_1 ↗
8. Wang, Q., Wu, B., Zhu, P., Li, P., Zuo, W., and Hu, Q. (2020, January 13–19). ECA-Net: Efficient channel attention for deep convolutional neural networks. Proceedings of the 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Seattle, WA, USA.10.1109/CVPR42600.2020.01155 ↗
9. Howard, A.G., Zhu, M., Chen, B., Kalenichenko, D., Wang, W., Weyand, T., Andreetto, M., and Adam, H. (2017). Mobilenets: Efficient convolutional neural networks for mobile vision applications. arXiv.
10. Bochkovskiy, A., Wang, C.-Y., and Liao, H.-Y.M. (2020). Yolov4: Optimal speed and accuracy of object detection. arXiv.
11. Yaseen, M. (2024). What is YOLOv8: An In-Depth Exploration of the Internal Features of the Next-Generation Object Detector. arXiv.
12. Lei, M., Li, S., Wu, Y., Yang, H., and Peng, L. (2025). YOLOv13: Real-Time Object Detection with Hypergraph-Enhanced Adaptive Visual Perception. arXiv.
13. DeepLab: Semantic image segmentation with deep convolutional nets, atrous convolution, and fully connected CRFs10.1109/TPAMI.2017.2699184 ↗
14. Zhu, P., Wen, L., Bian, X., Ling, H., and Hu, Q. (2018). Vision meets drones: A challenge. arXiv.
15. Xu, W., Sun, L., Zhen, C., Liu, B., Yang, Z., and Yang, W. (2022). Deep Learning-Based Image Recognition of Agricultural Pests. Appl. Sci., 12.10.3390/app122412896 ↗
16. Deep high-resolution representation learning for visual recognition10.1109/TPAMI.2020.2983686 ↗
17. Ioffe, S., and Szegedy, C. (2015, January 6–11). Batch normalization: Accelerating deep network training by reducing internal covariate shift. Proceedings of the International Conference on Machine Learning, Lille, France.
18. Yang, F., He, M., Liu, J., and Jin, H. (2025). RMH-YOLO: A Refined Multi-Scale Architecture for Small-Target Detection in UAV Aerial Imagery. Sensors, 25.10.3390/s25227088 ↗
19. Wu, J., Tang, X., Yang, Z., Hao, K., Lai, L., and Liu, Y. (2024). An Experimental Evaluation of LLM on Image Classification. Australasian Database Conference, Springer.10.1007/978-981-96-1242-0_37 ↗
20. Zhao, Y., Lv, W., Xu, S., Wei, J., Wang, G., Dang, Q., Liu, Y., and Chen, J. (2023). DETRs Beat YOLOs on Real-time Object Detection. arXiv.10.1109/CVPR52733.2024.01605 ↗
21. Huang, S., Xu, Z., Sun, X., Wang, Z., Jin, X., Li, X., and Zhang, X. (2024). DEIM: DETR with Improved Matching for Fast Convergence. arXiv.10.1109/CVPR52734.2025.01412 ↗
22. Yuan, Z., Gong, J., Guo, B., Wang, C., Liao, N., Song, J., and Wu, Q. (2024). Small Object Detection in UAV Remote Sensing Images Based on Intra-Group Multi-Scale Fusion Attention and Adaptive Weighted Feature Fusion Mechanism. Remote Sens., 16.10.3390/rs16224265 ↗
23. Tan, M., and Le, Q. (2019, January 9–15). Efficientnet: Rethinking model scaling for convolutional neural networks. Proceedings of the International Conference on Machine Learning, Long Beach, CA, USA.
24. Ghahramani, Z., Welling, M., Cortes, C., Lawrence, N., and Weinberger, K.Q. (2014). Generative adversarial nets. Advances in Neural Information Processing Systems, Curran Associates, Inc.
25. Model compression and hardware acceleration for neural networks: A comprehensive survey10.1109/JPROC.2020.2976475 ↗