Robust Tracking via Mamba-based Context-aware Token Learning

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Xie, Jinxia, Zhong, Bineng, Liang, Qihua, Li, Ning, Mo, Zhiyi, Song, Shuxiang
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866916529754341376
author Xie, Jinxia
Zhong, Bineng
Liang, Qihua
Li, Ning
Mo, Zhiyi
Song, Shuxiang
author_facet Xie, Jinxia
Zhong, Bineng
Liang, Qihua
Li, Ning
Mo, Zhiyi
Song, Shuxiang
contents How to make a good trade-off between performance and computational cost is crucial for a tracker. However, current famous methods typically focus on complicated and time-consuming learning that combining temporal and appearance information by input more and more images (or features). Consequently, these methods not only increase the model's computational source and learning burden but also introduce much useless and potentially interfering information. To alleviate the above issues, we propose a simple yet robust tracker that separates temporal information learning from appearance modeling and extracts temporal relations from a set of representative tokens rather than several images (or features). Specifically, we introduce one track token for each frame to collect the target's appearance information in the backbone. Then, we design a mamba-based Temporal Module for track tokens to be aware of context by interacting with other track tokens within a sliding window. This module consists of a mamba layer with autoregressive characteristic and a cross-attention layer with strong global perception ability, ensuring sufficient interaction for track tokens to perceive the appearance changes and movement trends of the target. Finally, track tokens serve as a guidance to adjust the appearance feature for the final prediction in the head. Experiments show our method is effective and achieves competitive performance on multiple benchmarks at a real-time speed. Code and trained models will be available at https://github.com/GXNU-ZhongLab/TemTrack.
format Preprint
id arxiv_https___arxiv_org_abs_2412_13611
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Robust Tracking via Mamba-based Context-aware Token Learning
Xie, Jinxia
Zhong, Bineng
Liang, Qihua
Li, Ning
Mo, Zhiyi
Song, Shuxiang
Computer Vision and Pattern Recognition
How to make a good trade-off between performance and computational cost is crucial for a tracker. However, current famous methods typically focus on complicated and time-consuming learning that combining temporal and appearance information by input more and more images (or features). Consequently, these methods not only increase the model's computational source and learning burden but also introduce much useless and potentially interfering information. To alleviate the above issues, we propose a simple yet robust tracker that separates temporal information learning from appearance modeling and extracts temporal relations from a set of representative tokens rather than several images (or features). Specifically, we introduce one track token for each frame to collect the target's appearance information in the backbone. Then, we design a mamba-based Temporal Module for track tokens to be aware of context by interacting with other track tokens within a sliding window. This module consists of a mamba layer with autoregressive characteristic and a cross-attention layer with strong global perception ability, ensuring sufficient interaction for track tokens to perceive the appearance changes and movement trends of the target. Finally, track tokens serve as a guidance to adjust the appearance feature for the final prediction in the head. Experiments show our method is effective and achieves competitive performance on multiple benchmarks at a real-time speed. Code and trained models will be available at https://github.com/GXNU-ZhongLab/TemTrack.
title Robust Tracking via Mamba-based Context-aware Token Learning
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2412.13611