DropMAE: Masked Autoencoders with Spatial-Attention Dropout for Tracking Tasks

• Published in: 2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) pp. 14561 - 14571
• Main Authors: Wu, Qiangqiang, Yang, Tianyu, Liu, Ziquan, Wu, Baoyuan, Shan, Ying, Chan, Antoni B.
• Format: Conference Proceeding
• Language: English
• Published: IEEE 01.06.2023
• Subjects: Image reconstruction; Object segmentation; Object tracking; Pattern recognition; Self-supervised or unsupervised representation learning; Task analysis; Video tracking; Visualization
• ISSN: 2575-7075
• DOI: 10.1109/CVPR52729.2023.01399

In this paper, we study masked autoencoder (MAE) pretraining on videos for matching-based downstream tasks, including visual object tracking (VOT) and video object segmentation (VOS). A simple extension of MAE is to randomly mask out frame patches in videos and reconstruct the frame pixels. However, we find that this simple baseline heavily relies on spatial cues while ignoring temporal relations for frame reconstruction, thus leading to sub-optimal temporal matching representations for VOT and VOS. To alleviate this problem, we propose DropMAE, which adaptively performs spatial-attention dropout in the frame reconstruction to facilitate temporal correspondence learning in videos. We show that our DropMAE is a strong and efficient temporal matching learner, which achieves better finetuning results on matching-based tasks than the ImageNet-based MAE with 2\times faster pre-training speed. Moreover, we also find that motion diversity in pre-training videos is more important than scene diversity for improving the performance on VOT and VOS. Our pre-trained DropMAE model can be directly loaded in existing ViT-based trackers for fine-tuning without further modifications. Notably, DropMAE sets new state-of-the-art performance on 8 out of 9 highly competitive video tracking and segmentation datasets. Our code and pre-trained models are available at https://github.com/jimmy-dq/DropMAE.git.