Dual-channel feature extraction hybrid attention network for detecting infrared small targets

被引:0
|
作者
Nie, Suzhen [1 ]
Cao, Jie [2 ]
Miao, Jiaqi [1 ]
Hou, Haiyuan [1 ]
Hao, Qun [2 ]
Zhuang, Xuye [1 ]
机构
[1] Shandong Univ Technol, Sch Mech Engn, Zibo, Peoples R China
[2] Beijing Inst Technol, Sch Opt & Photon, Beijing, Peoples R China
关键词
attention fusion; feature compression; multilevel feature enhancement; infrared small target detection; MODEL;
D O I
10.1088/1361-6501/ad7972
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
For military early warning, forest fire prevention, and maritime search and rescue, infrared small target detection is critical. However, because of the low contrast and inconspicuous features of infrared small targets, rendering most existing methods ineffective in restoring target edge details or misidentifying the background as a target. This paper proposes a dual-channel feature extraction network (DCFE-Net) with hybrid attention, which enables the network to suppress the background and enhance the target by designing dual-channel feature extraction and multi-layer feature fusion. Specifically, the dual-channel mainly consists of a convolutional attention fusion module, which adaptively integrates feature map correlations by introducing a hybrid attention module to capture global information while enhancing the feature representation of small targets, and a feature compression extraction module, which utilizes depth-separable convolutional combinations to carry out fine-grained target feature extraction while reducing the loss of details. In addition, the multilevel feature enhancement module ensures that the network can capture targets at different scales through skip connection operations, while avoiding small targets from being overwhelmed by deep features, making them simultaneously semantically informative and detailed. Therefore, the network can fuse multilevel features for effective information extraction. According to the experimental results, DCFE-Net performs best in false alarm rate and detection probability.
引用
下载
收藏
页数:14
相关论文
共 50 条
  • [1] DHAFormer: Dual-channel hybrid attention network with transformer for polyp segmentation
    Huang, Xuejie
    Wang, Liejun
    Jiang, Shaochen
    Xu, Lianghui
    PLOS ONE, 2024, 19 (07):
  • [2] DAFCNN: A Dual-Channel Feature Extraction and Attention Feature Fusion Convolution Neural Network for SAR Image and MS Image Fusion
    Luo, Jiahao
    Zhou, Fang
    Yang, Jun
    Xing, Mengdao
    REMOTE SENSING, 2023, 15 (12)
  • [3] Local Contrast Attention Guide Network for Detecting Infrared Small Targets
    Nian, Bingkun
    Jiang, Bo
    Shi, Huajun
    Zhang, Yi
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2023, 61
  • [4] DC-Net: A Dual-Channel and Cross-Scale Feature Fusion Infrared Small Target Detection Network
    Liu, Ying-Bin
    Huang, Han-Yan
    Zeng, Yu-Hui
    IEEE Transactions on Geoscience and Remote Sensing, 2024, 62
  • [5] Latent graph learning with dual-channel attention for relation extraction
    Tang, Guogen
    Li, Ping
    He, Yupeng
    Chen, Yan
    Zhong, Yuan
    Gan, Fangji
    KNOWLEDGE-BASED SYSTEMS, 2022, 252
  • [6] DCCL: Dual-channel hybrid neural network combined with self-attention for text classification
    Li, Chaofan
    Qiong, Liu
    Kai, Ma
    MATHEMATICAL BIOSCIENCES AND ENGINEERING, 2023, 20 (02) : 1981 - 1992
  • [7] Dual-Channel Hybrid Neural Network for Modulation Recognition
    Lu, Min
    Peng, Tianjun
    Yue, Guangxue
    Ma, Bolin
    Liao, Xiangbai
    IEEE ACCESS, 2021, 9 : 76260 - 76269
  • [8] Radar Signal Recognition Based on Dual-Channel Model with HOG Feature Extraction
    Tang Z.
    Quan D.
    Wang X.
    Jin N.
    Zhang D.
    IEEE Journal on Miniaturization for Air and Space Systems, 2023, 4 (04): : 358 - 367
  • [9] A Baseline Similarity Attention-Based Dual-Channel Feature Fusion Network for Machine Remaining Useful Life Prediction
    Hu, Yawei
    Li, Xuanlin
    Wang, Hang
    Liu, Yongbin
    Liu, Xianzeng
    Cao, Zheng
    IEEE TRANSACTIONS ON INSTRUMENTATION AND MEASUREMENT, 2024, 73 : 1 - 12
  • [10] Financial Anti-Fraud Based on Dual-Channel Graph Attention Network
    Wei, Sizheng
    Lee, Suan
    JOURNAL OF THEORETICAL AND APPLIED ELECTRONIC COMMERCE RESEARCH, 2024, 19 (01): : 297 - 314