基于改进YOLOv5的轻量化通信信号检测算法
作者:
作者单位:

西南交通大学 数学学院,四川 成都 611756

作者简介:

李书婷(1999-),女,在读硕士研究生,主要研究方向为深度学习和计算机视觉.email:lsting@my.swjtu.edu.cn.
付鑫艺(1999-),男,在读硕士研究生,主要研究方向为深度学习和计算机视觉.
何星星(1982-),男,博士,副教授,主要研究为方向神经符号计算.
任芮彬(1990-),女,博士,副教授,主要研究方向为机器学习、应用统计等.

通讯作者:

任芮彬(1990-),女,博士,副教授,主要研究方向为机器学习、应用统计等. email:Airy_Ren@163.com.

基金项目:

国家自然科学基金青年基金资助项目(12102369);中央高校基本科研业务费专项资金资助项目(2682024ZTPY041);四川省科技计划资助项目(2023YFH0066);成都市科技资助项目(2023-RK00-00080-ZF)

伦理声明:



A lightweight communication signal detection algorithm based on improved YOLOv5
Author:
Ethical statement:

Affiliation:

College of Mathematics,Southwest Jiaotong University,Chengdu Sichuan 611756,China

Funding:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
    摘要:

    通用信号检测模型参数量大且计算复杂,难以直接在资源受限的边缘端部署,且模型轻量化是研究的热点之一。对此,本文提出一种轻量化的通信信号检测算法ISR-YOLO。设计了轻量化网络I-Shufflenet作为主干特征提取网络,通过重构特征提取模块,提高信号检测的精度;在颈部网络中,根据信号几何特征,设计了特征融合模块Rec-SPPF,进一步提高信号定位框的完整性;另外添加卷积块注意力模块(CBAM),提升轻量化后网络对细长信号敏感度;选用对边界框高、宽分开计算的增强交并比(EIoU) Loss作为新的边界框损失函数,以提高网络的定位精度。在IEEE SPAW2021公开数据集和仿真数据集上的实验表明,在保持与YOLOv5性能几乎一致的同时,ISR-YOLO的计算复杂度和模型大小分别降低了87.34%和82.64%,显著降低边缘端对高算力硬件的依赖性。

    Abstract:

    General signal detection models typically feature large parameter sizes and high computational complexity, making them difficult to deploy directly on resource-constrained edge devices. Consequently, model lightweighting has become one of the active research topics in this field. To address this challenge, this paper proposes a lightweight communication signal detection algorithm named ISR-YOLO(I-Shufflenet Rec-SPPF(Spatial Pyramid Pooling-Fast)-You Only Look Once). A lightweight network, IShufflenet, is designed as the backbone feature extraction network, where the feature extraction modules are reconstructed to improve signal detection accuracy. In the neck network, a feature fusion module called Rec-SPPF is designed based on signal geometric characteristics to further enhance the integrity of signal bounding boxes. Additionally, the Convolutional Block Attention Module(CBAM) is incorporated to improve the sensitivity of the lightweighted network to elongated signals. Furthermore, the Enhanced Intersection over Union(EIoU) Loss, which separately calculates the height and width of bounding boxes, is adopted as the new bounding box loss function to improve the network's localization accuracy. Experiments conducted on the IEEE SPAW2021 public dataset and simulated datasets demonstrate that ISR-YOLO achieves performance almost identical to YOLOv5 while reducing computational complexity and model size by 87.34% and 82.64%, respectively, significantly decreasing the dependency on high-computing-power hardware at the edge.

    参考文献
    相似文献
    引证文献
引用本文

李书婷,付鑫艺,何星星,等.基于改进YOLOv5的轻量化通信信号检测算法[J].太赫兹科学与电子信息学报,2026,24(3):397~408. DOI:10.11805/TKYDA2024629.
LI Shuting, FU Xinyi, HE Xingxing, et al. A lightweight communication signal detection algorithm based on improved YOLOv5[J]. Journal of Terahertz Science and Electronic Information,2026,24(3):397-408. DOI:10.11805/TKYDA2024629.

复制
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
历史
  • 收稿日期:2024-12-18
  • 最后修改日期:2025-02-11
  • 录用日期:
  • 在线发布日期: 2026-04-03
  • 出版日期:
关闭