Opto-Electronic Engineering, Volume. 52, Issue 5, 250061(2025)

Hypergraph computed efficient transmission multi-scale feature small target detection algorithm

Yemei Sun1, Xueting Sang1, Yan Zhang1, Guorui Liu1、*, and Shuaiyu Chen1、*
Author Affiliations
  • 1[in Chinese]
  • 1School of Computer and Information Engineering, Tianjin Chengjian University, Tianjin 300380, China
  • show less

    UAV aerial images have the characteristics of complex background, small and dense targets. Aiming at the problems of low precision and a large number of model parameters in UAV aerial image detection, an efficient multi-scale feature transfer small target detection algorithm based on hypergraph computation is proposed. Firstly, a multi-scale feature pyramid network is designed as a neck network to effectively reduce the problem of information loss caused by lengthy transmission paths by fusing multi-layer features in the middle layer and transmitting them directly to adjacent layers. In addition, the feature fusion process uses hypergraphs to model higher-order features, improving the nonlinear representation ability of the model. Secondly, a lightweight dynamic task-guided detection head is designed to effectively solve the problem of inaccurate detection targets caused by inconsistent classification and positioning task space in the traditional decoupling head with a small number of parameters through sharing mechanism. Finally, the pruning lightweight model based on layer adaptive amplitude is used to further reduce the model volume. The experimental results show that this algorithm has better performance than other architectures on VisDrone2019 dataset, with the accuracy mAP0.5 and parameter number reaching 42.4% and 4.8 M, respectively. Compared with the benchmark YOLOv8, the parameter number is reduced by 54.7%. The model achieves a good balance between detection performance and resource consumption.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Yemei Sun, Xueting Sang, Yan Zhang, Guorui Liu, Shuaiyu Chen. Hypergraph computed efficient transmission multi-scale feature small target detection algorithm[J]. Opto-Electronic Engineering, 2025, 52(5): 250061

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Article

    Received: Feb. 28, 2025

    Accepted: Apr. 8, 2025

    Published Online: Jul. 18, 2025

    The Author Email: Guorui Liu (刘国瑞), Shuaiyu Chen (陈帅宇)

    DOI:10.12086/oee.2025.250061

    Topics