arXiv:2601.05446cs.CV2026-01

通过追踪红外小目标引发的特征扰动轨迹,提升检测精度。

TAPM-Net: Trajectory-Aware Perturbation Modeling for Infrared Small Target Detection

  • 引入扰动路径模块与轨迹感知状态块,建模特征扰动的空间扩散行为。
  • 在NUAA-SIRST和IRSTD-1K数据集上达到当前最优性能。
  • 适合需要高精度红外小目标检测的军事或遥感应用。

红外小目标检测(ISTD)因信号对比弱、空间范围小和背景杂乱而长期面临挑战。尽管卷积神经网络(CNNs)和视觉变换器(ViTs)提升了性能,现有模型仍缺乏追踪小目标如何在特征空间中引发方向性、分层扰动的机制,而这正是区分红外场景中信号与结构化噪声的关键线索。为此,我们提出轨迹感知马尔可夫传播网络(TAPM-Net),显式建模目标诱导特征扰动的空间扩散行为。TAPM-Net由两个新组件构成:扰动引导路径模块(PGM)和轨迹感知状态块(TASB)。PGM从多层级特征构建扰动能量场,并提取反映局部响应方向性的梯度跟随特征轨迹。这些轨迹输入TASB——一种基于Mamba的状态空间单元,能够沿每条轨迹建模动态传播,同时融合速度约束扩散与词级和句级嵌入的语义对齐特征。与现有注意力方法不同,TAPM-Net在保持全局一致性的同时,实现沿空间轨迹的各向异性、上下文敏感状态转移,且计算成本低。在NUAA-SIRST和IRSTD-1K数据集上的实验表明,TAPM-Net在ISTD任务中达到当前最优性能。

原文摘要 · Abstract (English)

Infrared small target detection (ISTD) remains a long-standing challenge due to weak signal contrast, limited spatial extent, and cluttered backgrounds. Despite performance improvements from convolutional neural networks (CNNs) and Vision Transformers (ViTs), current models lack a mechanism to trace how small targets trigger directional, layer-wise perturbations in the feature space, which is an essential cue for distinguishing signal from structured noise in infrared scenes. To address this limitation, we propose the Trajectory-Aware Mamba Propagation Network (TAPM-Net), which explicitly models the spatial diffusion behavior of target-induced feature disturbances. TAPM-Net is built upon two novel components: a Perturbation-guided Path Module (PGM) and a Trajectory-Aware State Block (TASB). The PGM constructs perturbation energy fields from multi-level features and extracts gradient-following feature trajectories that reflect the directionality of local responses. The resulting feature trajectories are fed into the TASB, a Mamba-based state-space unit that models dynamic propagation along each trajectory while incorporating velocity-constrained diffusion and semantically aligned feature fusion from word-level and sentence-level embeddings. Unlike existing attention-based methods, TAPM-Net enables anisotropic, context-sensitive state transitions along spatial trajectories while maintaining global coherence at low computational cost. Experiments on NUAA-SIRST and IRSTD-1K demonstrate that TAPM-Net achieves state-of-the-art performance in ISTD.

红外检测小目标轨迹建模Mamba

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