arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.09178cs.CV
收藏

MambaTrack:一种基于状态空间模型的多目标跟踪简单基线

MambaTrack: A Simple Baseline for Multiple Object Tracking with State Space Model

National University of Defense Technology(国防科技大学);JD Explore Academy(京东探索研究院)

Changcheng Xiao, Qiong Cao, Zhigang Luo, Long Lan

AI总结:

针对传统多目标跟踪方法难以处理非线性复杂运动、轨迹易因遮挡中断的问题,该研究提出基于Mamba的运动预测器MTP,构建MambaTrack跟踪器,在复杂运动场景基准上实现先进性能。

摘要:

基于检测的跟踪一直是多目标跟踪(Multi-object Tracking, MOT)领域的主流范式。这类方法通常依赖Kalman Filter(卡尔曼滤波器)估计目标的未来位置,其前提假设是目标做线性运动。但在舞蹈、体育等场景中,当目标呈现非线性且多样化的运动模式时,这类方法的表现会有所不足。此外,目前MOT领域对基于学习的运动预测器的研究关注较为有限。\n为应对上述挑战,我们着手探索数据驱动的运动预测方法。受Mamba等状态空间模型(state space models, SSMs)在长序列建模中具备近线性复杂度的巨大潜力启发,我们提出了一种基于Mamba的运动模型,名为Mamba moTion Predictor(MTP,Mamba运动预测器)。MTP旨在对舞者、运动员等目标的复杂运动模式进行建模。具体而言,MTP以目标的时空位置动态变化作为输入,通过bi-Mamba编码层捕捉运动模式,进而预测下一步的运动。\n在真实场景中,目标可能因遮挡或运动模糊而被漏检,导致其轨迹提前终止。为解决这一问题,我们进一步拓展了MTP的应用场景:我们以自回归的方式使用MTP,将其自身的预测结果作为输入来补全缺失的观测值,从而得到更连贯的轨迹。我们提出的跟踪器MambaTrack在Dancetrack和SportsMOT等以复杂运动和严重遮挡为特征的基准测试上展现出了先进性能。

英文摘要:

Tracking by detection has been the prevailing paradigm in the field of Multi-object Tracking (MOT). These methods typically rely on the Kalman Filter to estimate the future locations of objects, assuming linear object motion. However, they fall short when tracking objects exhibiting nonlinear and diverse motion in scenarios like dancing and sports. In addition, there has been limited focus on utilizing learning-based motion predictors in MOT. To address these challenges, we resort to exploring data-driven motion prediction methods. Inspired by the great expectation of state space models (SSMs), such as Mamba, in long-term sequence modeling with near-linear complexity, we introduce a Mamba-based motion model named Mamba moTion Predictor (MTP). MTP is designed to model the complex motion patterns of objects like dancers and athletes. Specifically, MTP takes the spatial-temporal location dynamics of objects as input, captures the motion pattern using a bi-Mamba encoding layer, and predicts the next motion. In real-world scenarios, objects may be missed due to occlusion or motion blur, leading to premature termination of their trajectories. To tackle this challenge, we further expand the application of MTP. We employ it in an autoregressive way to compensate for missing observations by utilizing its own predictions as inputs, thereby contributing to more consistent trajectories. Our proposed tracker, MambaTrack, demonstrates advanced performance on benchmarks such as Dancetrack and SportsMOT, which are characterized by complex motion and severe occlusion.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑