层次化的深度融合的细节特征增强网络针对红外小目标检测

Hierarchical Deep Fusion Detail Feature Enhancement Network for Infrared Small Target Detection

  • 摘要: 红外小目标检测从红外图像的复杂背景中识别出小目标,是一项关键且具有挑战性的任务。现有的方法在有效地针对性进行小目标信息的特征提取,减少噪声和保留细节信息有待改进,在小目标检测效果遇到问题。本文结合红外小目标检测近年来的U型神经模型,提出了层次化深度融合细节特征增强网络(HFE-Net)用来解决上面的问题。设计了通道特征的增强注意力模块(CFEAM),针对红外小目标检测进行设计,对信息的局部性和全局性进行特征整体增强,能够进行通道信息的增强,保留了小目标特征的细节。本文设计了一个多分支带状感知注意力模块(MSPA),结合卷积方法增强检测全局上下文信息,以降低深层中的干扰信息。此外,设计了一个浅层信息增强处理模块(SIEP),能够高效利用浅层特征信息,结合空间和频域注意力增强操作以保留局部细节并进一步去除浅层中的噪声,拓宽感受野,提高检测性能。本文方法在多个公开数据集上开展的大量实验,对比基准模型,在平均交并比(mIoU)指标上取得了显著提升,在NUDT-SIRST数据集上提升4.52%,在NUAA-SIRST数据集上提升2.25%。实验结果表明,本文所提出的算法检测精度优于目前较为先进的红外小目标检测算法,充分验证了所提出的算法具有更高的特征提取能力和更高的检测精度。

     

    Abstract: Infrared small target detection aims to identify tiny targets from complex infrared backgrounds, a task that is both critical and highly challenging. Existing methods still face difficulties in effectively extracting discriminative features of small targets, suppressing noise, and preserving fine details, which often limits detection performance. To address these issues, this paper proposes a Hierarchical Deep Fusion Detail Feature Enhancement Network (HFE-Net) based on recent U-shaped neural architectures for infrared small target detection. Specifically, a Channel Feature Enhancement Attention Module (CFEAM) is designed for infrared small target detection, which enhances both local and global information in a unified manner. This module strengthens channel-wise representations while preserving fine-grained details of small target features. Moreover, a Multi-branch Strip Perception Attention Module(MSPA) is introduced, which integrates convolution-based strategies to enhance global contextual information and effectively suppress interference in deeper layers. Furthermore, a Shallow Information Enhancement Processing Module (SIEP) is developed to efficiently exploit shallow features by combining spatial-frequency attention mechanisms, thereby preserving local details, suppressing shallow noise, and expanding the receptive field to improve detection performance. Extensive experiments conducted on multiple public datasets demonstrate that the proposed method significantly improves mean Intersection over Union (mIoU) compared with baseline models, achieving gains of 4.52% on NUDT-SIRST and 2.25% on NUAA-SIRST. The results confirm that the proposed algorithm not only achieves superior detection accuracy over state-of-the-art methods but also exhibits strong noise robustness and reliable detection effectiveness.

     

/

返回文章
返回