轻量化红外小目标检测网络,抗噪能力强且参数少。
Lightweight Deep Unfolding Networks with Enhanced Robustness for Infrared Small Target Detection
- 用分层瓶颈结构压缩通道,降低参数量。
- 嵌入噪声抑制模块,复杂环境下检测更稳定。
- 结合注意力机制,在保持轻量的同时提升精度。
红外小目标检测(ISTD)是图像处理中的关键技术。尽管深度展开网络(DUNs)因模型可解释性和数据适应性在ISTD中表现优异,但现有方法仍面临参数量大和抗噪能力弱的挑战。为此,本文提出一种基于鲁棒主成分分析(RPCA)的轻量化框架L-RPCANet。技术上,构建分层瓶颈结构,对单通道红外图像进行通道维度的压缩与扩展,实现通道级特征精炼;每个模块中设计瓶颈层以提取特征,减少特征提取过程中的通道数,从而提升网络参数的轻量化程度。此外,嵌入噪声抑制模块以增强对复杂噪声的鲁棒性。同时,引入挤压-激励网络(SENets)作为通道注意力机制,聚焦不同通道特征的重要性差异,实现在轻量化与强鲁棒性下卓越的检测性能。在ISTD数据集上的大量实验表明,所提方法在性能上优于当前先进方法,包括RPCANet、DRPCANet和RPCANet++。代码将开源于https://github.com/xianchaoxiu/L-RPCANet。
原文摘要 · Abstract (English)
Infrared small target detection (ISTD) is one of the key techniques in image processing. Although deep unfolding networks (DUNs) have demonstrated promising performance in ISTD due to their model interpretability and data adaptability, existing methods still face significant challenges in parameter lightweightness and noise robustness. In this regard, we propose a highly lightweight framework based on robust principal component analysis (RPCA) called L-RPCANet. Technically, a hierarchical bottleneck structure is constructed to reduce and increase the channel dimension in the single-channel input infrared image to achieve channel-wise feature refinement, with bottleneck layers designed in each module to extract features. This reduces the number of channels in feature extraction and improves the lightweightness of network parameters. Furthermore, a noise reduction module is embedded to enhance the robustness against complex noise. In addition, squeeze-and-excitation networks (SENets) are leveraged as a channel attention mechanism to focus on the varying importance of different features across channels, thereby achieving excellent performance while maintaining both lightweightness and robustness. Extensive experiments on the ISTD datasets validate the superiority of our proposed method compared with state-of-the-art methods covering RPCANet, DRPCANet, and RPCANet++. The code will be available at https://github.com/xianchaoxiu/L-RPCANet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。