arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2501.02467cs.CV

DeTrack:用于视觉目标跟踪的模型内潜在去噪学习

DeTrack: In-model Latent Denoising Learning for Visual Object Tracking

Fudan University(复旦大学);Shanghai Key Lab of Intelligent Information Processing(上海市智能信息处理重点实验室);School of Computer Science, Fudan University(复旦大学计算机科学技术学院);Shanghai Engineering Research Center of AI & Robotics(上海市人工智能与机器人工程技术研究中心);Academy for Engineering and Technology, Fudan University(复旦大学工程与技术学院)

Xinyu Zhou, Jinglun Li, Lingyi Hong, Kaixun Jiang, Pinxue Guo, Weifeng Ge, Wenqiang Zhang

AI总结:

本文提出 DeTrack,将视觉跟踪建模为模型内潜在去噪学习,通过去噪 ViT 逐块消除边界框噪声,并结合轨迹与视觉记忆,在保证实时性的同时提升鲁棒性和跟踪性能。

摘要:

以往的视觉目标跟踪方法采用图像特征回归模型或坐标自回归模型进行边界框预测。图像特征回归方法严重依赖匹配结果,且未利用位置先验;而自回归方法只能使用训练集中可用的边界框进行训练,在测试时面对未见过的数据可能导致次优性能。受扩散模型启发,去噪学习可增强模型对未见过数据的鲁棒性。因此,我们向边界框中引入噪声,生成带噪边界框用于训练,从而增强模型在测试数据上的鲁棒性。我们提出了一种新范式,将视觉目标跟踪问题表述为去噪学习过程。然而,跟踪算法通常需要实时运行,直接将扩散模型应用于目标跟踪会严重损害跟踪速度。因此,我们将去噪学习过程分解到模型内部的每个去噪块中,而不是多次运行模型,并据此将所提出的范式概括为模型内潜在去噪学习过程。具体而言,我们提出了一种去噪 Vision Transformer(ViT),它由多个去噪块组成。在去噪块中,模板嵌入和搜索嵌入被投影到每个去噪块中作为条件。一个去噪块负责去除预测边界框中的噪声,多个堆叠的去噪块协同完成整个去噪过程。随后,我们利用图像特征和轨迹信息来细化去噪后的边界框。此外,我们还利用轨迹记忆和视觉记忆来提高跟踪稳定性。实验结果验证了本方法的有效性,并在多个具有挑战性的数据集上取得了具有竞争力的性能。

英文摘要:

Previous visual object tracking methods employ image-feature regression models or coordinate autoregression models for bounding box prediction. Image-feature regression methods heavily depend on matching results and do not utilize positional prior, while the autoregressive approach can only be trained using bounding boxes available in the training set, potentially resulting in suboptimal performance during testing with unseen data. Inspired by the diffusion model, denoising learning enhances the model's robustness to unseen data. Therefore, We introduce noise to bounding boxes, generating noisy boxes for training, thus enhancing model robustness on testing data. We propose a new paradigm to formulate the visual object tracking problem as a denoising learning process. However, tracking algorithms are usually asked to run in real-time, directly applying the diffusion model to object tracking would severely impair tracking speed. Therefore, we decompose the denoising learning process into every denoising block within a model, not by running the model multiple times, and thus we summarize the proposed paradigm as an in-model latent denoising learning process. Specifically, we propose a denoising Vision Transformer (ViT), which is composed of multiple denoising blocks. In the denoising block, template and search embeddings are projected into every denoising block as conditions. A denoising block is responsible for removing the noise in a predicted bounding box, and multiple stacked denoising blocks cooperate to accomplish the whole denoising process. Subsequently, we utilize image features and trajectory information to refine the denoised bounding box. Besides, we also utilize trajectory memory and visual memory to improve tracking stability. Experimental results validate the effectiveness of our approach, achieving competitive performance on several challenging datasets.

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

↑