TY - JOUR
T1 - Dual-channel feature extraction hybrid attention network for detecting infrared small targets
AU - Nie, Suzhen
AU - Cao, Jie
AU - Miao, Jiaqi
AU - Hou, Haiyuan
AU - Hao, Qun
AU - Zhuang, Xuye
N1 - Publisher Copyright:
© 2024 IOP Publishing Ltd. All rights, including for text and data mining, AI training, and similar technologies, are reserved.
PY - 2024/12
Y1 - 2024/12
N2 - 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.
AB - 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.
KW - attention fusion
KW - feature compression
KW - infrared small target detection
KW - multilevel feature enhancement
UR - http://www.scopus.com/inward/record.url?scp=85205800119&partnerID=8YFLogxK
U2 - 10.1088/1361-6501/ad7972
DO - 10.1088/1361-6501/ad7972
M3 - Article
AN - SCOPUS:85205800119
SN - 0957-0233
VL - 35
JO - Measurement Science and Technology
JF - Measurement Science and Technology
IS - 12
M1 - 125405
ER -