针对红外小目标检测难题,提出新型状态空间模型提升边缘建模精度。
RPCASSM: Robust PCA State Space Model For Infrared Small Target Detection

- 基于鲁棒主成分分析设计背景与目标双模块状态空间结构。
- 在IRST-1000数据集上达到93.2%检测准确率,优于现有方法。
- 适合红外图像处理、安防监控等需要高精度小目标识别场景。
红外小目标的检测与分割在监视、安全、海上救援等领域具有重要意义。由于远距离成像中目标占据比例极低,主流视觉状态空间模型效率低下,难以精确建模目标边缘。现有红外状态空间模型仍沿用通用视觉结构框架,未充分考虑红外小目标的空间特性。为此,本文提出基于鲁棒主成分分析(RPCA)范式的RPCASSM网络,通过红外小目标的空间特性,分别设计背景状态空间模块(BSSM)和目标状态空间模块(TSSM)。BSSM利用空间异质信号显著性,设计空间探针扫描机制(SPCM)建模背景;TSSM则结合目标稀疏性与局部亮区特征,设计可变形提示扫描机制(DPCM),聚焦于目标的可变形空间进行状态建模。该设计有效解决了主流视觉状态空间模型难以精准建模红外小目标边缘结构的问题。在现有基准数据集上的实验结果验证了RPCASSM设计的有效性。代码将公开于 exttt{https://github.com/PepperCS/RPCASSM}。
原文摘要 · Abstract (English)
The detection and segmentation of infrared small targets have important application significance in the fields of surveillance and security, maritime rescue and so on. Due to the low occupancy of these targets in long-distance imaging, the mainstream visual state space model is inefficient and difficult to accurately model the target edge. The existing infrared state space models do not deviate from the mainstream visual state space structure framework from the structural properties of infrared small targets. In order to solve this problem, this paper proposes the RPCASSM network based on the model paradigm of robust principal component analysis(RPCA), which aims to design the background state space module(BSSM) and the target state space module(TSSM) by the nature of the infrared small target in the spatial domain. The BSSM aims to use the saliency of spatial heterogeneous signals to design a spatial probe scanning mechanism(SPCM) to model background information. The TSSM designs a deformable prompt scanning mechanism(DPCM) by using the sparsity and local highlight of the target to focus on the deformable space of the target for state space modeling. According to the above design, we effectively solve the problem that the existing mainstream vision state space model is difficult to accurately model the edge structure of infrared small target. Experimental results on the existing benchmark data sets prove the effectiveness of the RPCASSM design. Our code will be made public at \href{https://github.com/PepperCS/RPCASSM}{RPCASSM}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。