Enhancing Infrared Small Target Detection: A Multi-scale Feature Integration Approach with an Optimized Combined Loss Function
摘要
Infrared small target detection (IRSTD) remains a challenging task due to complex backgrounds, low signal-to-noise ratios, and the weak, small-scale nature of target features. Although existing methods have achieved promising results, they often face limitations in effectively distinguishing small targets from background clutter and maintaining robustness under varying environmental conditions. To address these challenges, this paper presents a novel IRSTD model based on an encoder-decoder architecture, specifically designed to enhance small target detection in complex infrared scenes. The encoder incorporates the Res2Net module combined with the convolutional block attention module (CBAM) to improve multi-scale feature extraction. In the decoder, we introduce residual convolution blocks (RCBs) and dilation bottleneck blocks (DBBs) to effectively capture global context and refine target localization. In addition, a combined loss function is proposed to improve detection accuracy while minimizing false alarm rate. Extensive experiments conducted on the SIRST, NUDT-SIRST and IRSTD-1K datasets demonstrate that our model outperforms state-of-the-art methods in both detection accuracy and false alarm reduction. Ablation studies validate the effectiveness of the proposed modules and show their impact on overall performance. Moreover, the model’s ability to maintain high processing speed using only CPU resources highlights its practicality for near real-time applications. The code is available at: https://github.com/long-nguyen12/irstd-pytorch.