arXiv:2507.01535cs.CV2025-07被引 3

用嵌套Mamba结构提升无人机追踪速度与精度

TrackingMiM: Efficient Mamba-in-Mamba Serialization for Real-time UAV Object Tracking

  • 采用嵌套Mamba架构,分别处理时空一致的图像块
  • 在5个无人机追踪数据集上达到领先精度与更快推理速度
  • 适合对实时性要求高的无人机视觉追踪场景

视觉变换器(ViT)模型长期面临二次复杂度挑战,尤其在需要实时处理的无人机(UAV)追踪系统中尤为突出。本文探索了近期提出的状态空间模型Mamba,利用其计算高效和长序列建模能力,有效处理追踪任务中的密集图像序列。首先,指出现有基于Mamba的方法存在时间不一致性问题,未能考虑Mamba扫描机制中的时间连续性。其次,提出TrackingMiM——一种轻量级的Mamba-in-Mamba架构,用于处理追踪中的图像序列。在本框架中,Mamba扫描以嵌套方式执行,独立处理时空一致的图像块;模板帧被编码为查询令牌,并在每次扫描中用于追踪。在五个无人机追踪基准上的大量实验表明,所提TrackingMiM在保持领先精度的同时,显著提升了追踪速度。

原文摘要 · Abstract (English)

The Vision Transformer (ViT) model has long struggled with the challenge of quadratic complexity, a limitation that becomes especially critical in unmanned aerial vehicle (UAV) tracking systems, where data must be processed in real time. In this study, we explore the recently proposed State-Space Model, Mamba, leveraging its computational efficiency and capability for long-sequence modeling to effectively process dense image sequences in tracking tasks. First, we highlight the issue of temporal inconsistency in existing Mamba-based methods, specifically the failure to account for temporal continuity in the Mamba scanning mechanism. Secondly, building upon this insight,we propose TrackingMiM, a Mamba-in-Mamba architecture, a minimal-computation burden model for handling image sequence of tracking problem. In our framework, the mamba scan is performed in a nested way while independently process temporal and spatial coherent patch tokens. While the template frame is encoded as query token and utilized for tracking in every scan. Extensive experiments conducted on five UAV tracking benchmarks confirm that the proposed TrackingMiM achieves state-of-the-art precision while offering noticeable higher speed in UAV tracking.

无人机追踪Mamba序列建模

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