用动态重构的鲁棒主成分分析提升红外小目标检测精度
DRPCA-Net: Make Robust PCA Great Again for Infrared Small Target Detection
- 将稀疏先验融入可学习网络,动态生成迭代参数
- 在多个公开数据集上检测准确率显著优于现有方法
- 适合需要高效、可解释小目标检测的遥感与工业场景
红外小目标检测在遥感、工业监控及各类民用领域中至关重要。尽管深度学习取得进展,但多数端到端卷积模型通过堆叠复杂架构追求性能,常牺牲可解释性、参数效率和泛化能力。这些模型往往忽略红外小目标的固有稀疏性——一个可显式建模以同时提升性能与效率的关键线索。为此,本文重新审视基于模型的鲁棒主成分分析(RPCA)范式,提出动态RPCA网络(DRPCA-Net),一种将稀疏感知先验融入可学习架构的新型深度展开网络。不同于依赖静态全局参数的传统深度展开方法,DRPCA-Net通过轻量级超网络引入动态展开机制,使模型能根据输入场景自适应生成逐次迭代参数,从而增强对多样化背景的鲁棒性与泛化能力。此外,设计了动态残差组(DRG)模块以更好捕捉背景中的上下文变化,实现更精准的低秩估计并提升小目标分离效果。在多个公开红外数据集上的大量实验表明,DRPCA-Net在检测准确率上显著优于现有最先进方法。代码已开源:https://github.com/GrokCV/DRPCA-Net。
原文摘要 · Abstract (English)
Infrared small target detection plays a vital role in remote sensing, industrial monitoring, and various civilian applications. Despite recent progress powered by deep learning, many end-to-end convolutional models tend to pursue performance by stacking increasingly complex architectures, often at the expense of interpretability, parameter efficiency, and generalization. These models typically overlook the intrinsic sparsity prior of infrared small targets--an essential cue that can be explicitly modeled for both performance and efficiency gains. To address this, we revisit the model-based paradigm of Robust Principal Component Analysis (RPCA) and propose Dynamic RPCA Network (DRPCA-Net), a novel deep unfolding network that integrates the sparsity-aware prior into a learnable architecture. Unlike conventional deep unfolding methods that rely on static, globally learned parameters, DRPCA-Net introduces a dynamic unfolding mechanism via a lightweight hypernetwork. This design enables the model to adaptively generate iteration-wise parameters conditioned on the input scene, thereby enhancing its robustness and generalization across diverse backgrounds. Furthermore, we design a Dynamic Residual Group (DRG) module to better capture contextual variations within the background, leading to more accurate low-rank estimation and improved separation of small targets. Extensive experiments on multiple public infrared datasets demonstrate that DRPCA-Net significantly outperforms existing state-of-the-art methods in detection accuracy. Code is available at https://github.com/GrokCV/DRPCA-Net.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。