학술논문

AFE-Net: Attention-Guided Feature Enhancement Network for Infrared Small Target Detection
Document Type
article
Source
IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, Vol 17, Pp 4208-4221 (2024)
Subject
Attention mechanism
convolutional neural network (CNN)
infrared small target detection
nonlocal dependency
Ocean engineering
TC1501-1800
Geophysics. Cosmic physics
QC801-809
Language
English
ISSN
2151-1535
Abstract
Infrared small target detection is considerably challenging due to the few pixels in targets, low signal-to-noise ratio, and complex background. In this article, we propose an effective attention-guided feature enhancement network (AFE-Net), which can leverage the local and nonlocal features of targets and background in infrared images. The AFE-Net consists of three key modules, namely encoder and decoder interactive guidance (EDIG) module, cascading false alarm removal (CFAR) module, and random scale input (RSI) module. Specifically, in the EDIG module, we employ a CA mechanism on encoding and decoding layers to select feature channels with higher contribution. Then, we impose a bottom-up pointwise attention block to highlight the features of small infrared targets and suppress possible noise by incorporating the low-level detailed features into the high-level semantic features. The CFAR module extracts affluent global features by cascading nonlocal operations of different layers, which can remove clutters with similar features to infrared targets. The RSI module is placed in front of the entire detection network to extract multiscale features of infrared small targets, which can enhance the robustness of the proposed network. Experimental results on the SIRST dataset and comprehensive comparisons with representative methods demonstrate the superiority of our proposed method.