arXiv:2505.10595cs.CV2025-05被引 5

提出新型网络提升红外小目标检测精度与鲁棒性

ARFC-WAHNet: Adaptive Receptive Field Convolution and Wavelet-Attentive Hierarchical Network for Infrared Small Target Detection

  • 自适应感受野卷积融合多尺度特征,增强目标判别力
  • 小波频域重构提升目标特征,抑制背景噪声,漏检率降37.2%
  • 适合复杂背景下红外小目标检测,尤其对弱信号敏感

红外小目标检测在民用和军事应用中至关重要。由于红外图像纹理和结构信息有限,准确检测尤为困难。尽管近期基于深度学习的方法有所改进,但传统卷积核难以适应复杂场景和多样化目标,且下采样操作常导致特征丢失。为此,本文提出自适应感受野卷积与小波注意力分层网络(ARFC-WAHNet)。该网络引入多感受野特征交互卷积(MRFFIConv)模块,通过门控单元融合多分支卷积,自适应提取判别特征;设计小波频率增强下采样(WFED)模块,利用哈尔小波变换与频域重建强化目标特征、抑制背景噪声;提出高低层特征融合(HLFF)模块,整合低层细节与高层语义;并引入全局中值增强注意力(GMEA)模块,通过全局注意力提升特征多样性与表达能力。在SIRST、NUDT-SIRST和IRSTD-1k三个公开数据集上的实验表明,ARFC-WAHNet在检测精度和鲁棒性上均优于现有最先进方法,尤其在复杂背景下的表现更优。代码已开源。

原文摘要 · Abstract (English)

Infrared small target detection (ISTD) is critical in both civilian and military applications. However, the limited texture and structural information in infrared images makes accurate detection particularly challenging. Although recent deep learning-based methods have improved performance, their use of conventional convolution kernels limits adaptability to complex scenes and diverse targets. Moreover, pooling operations often cause feature loss and insufficient exploitation of image information. To address these issues, we propose an adaptive receptive field convolution and wavelet-attentive hierarchical network for infrared small target detection (ARFC-WAHNet). This network incorporates a multi-receptive field feature interaction convolution (MRFFIConv) module to adaptively extract discriminative features by integrating multiple convolutional branches with a gated unit. A wavelet frequency enhancement downsampling (WFED) module leverages Haar wavelet transform and frequency-domain reconstruction to enhance target features and suppress background noise. Additionally, we introduce a high-low feature fusion (HLFF) module for integrating low-level details with high-level semantics, and a global median enhancement attention (GMEA) module to improve feature diversity and expressiveness via global attention. Experiments on public datasets SIRST, NUDT-SIRST, and IRSTD-1k demonstrate that ARFC-WAHNet outperforms recent state-of-the-art methods in both detection accuracy and robustness, particularly under complex backgrounds. The code is available at https://github.com/Leaf2001/ARFC-WAHNet.

红外检测小目标注意力机制小波变换

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。