arXiv:2412.17302cs.CV2024-12被引 12

用神经网络提升红外小目标检测的时空建模能力

Neural Spatial-Temporal Tensor Representation for Infrared Small Target Detection

  • 用神经网络建模时空特征,替代传统固定结构
  • 参数减少16.6倍,平均IoU提升19.19%
  • 适合处理动态背景下的红外小目标检测

基于优化的方法在红外小目标检测中占主导地位,因其利用红外图像固有的低秩性和稀疏性。然而,在多帧场景下,传统时空表示难以适应动态变化。为此,我们提出神经空间-时序张量(NeurSTT)模型,通过非线性网络增强背景近似中的时空特征相关性,实现无监督目标检测。具体地,采用神经层在低秩引导的深度框架中近似序列背景;设计神经三维总变差以优化背景平滑性,减少序列中静态目标类簇;将传统稀疏性约束融入损失函数以保留潜在目标。通过用深度更新策略替代复杂求解器,实现领域感知的简化优化过程。在多个数据集上的视觉与数值结果表明,该方法显著优于现有挑战。尤其在256×256序列上,参数量减少16.6倍,平均IoU较次优方法提升19.19%。

原文摘要 · Abstract (English)

Optimization-based approaches dominate infrared small target detection as they leverage infrared imagery's intrinsic low-rankness and sparsity. While effective for single-frame images, they struggle with dynamic changes in multi-frame scenarios as traditional spatial-temporal representations often fail to adapt. To address these challenges, we introduce a Neural-represented Spatial-Temporal Tensor (NeurSTT) model. This framework employs nonlinear networks to enhance spatial-temporal feature correlations in background approximation, thereby supporting target detection in an unsupervised manner. Specifically, we employ neural layers to approximate sequential backgrounds within a low-rank informed deep scheme. A neural three-dimensional total variation is developed to refine background smoothness while reducing static target-like clusters in sequences. Traditional sparsity constraints are incorporated into the loss functions to preserve potential targets. By replacing complex solvers with a deep updating strategy, NeurSTT simplifies the optimization process in a domain-awareness way. Visual and numerical results across various datasets demonstrate that our method outperforms detection challenges. Notably, it has 16.6$\times$ fewer parameters and averaged 19.19\% higher in $IoU$ compared to the suboptimal method on $256 \times 256$ sequences.

红外检测张量建模神经优化小目标

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