Long-Short Temporal Contrastive Learning of Video Transformers
arXiv:2106.09212
Abstract
Video transformers have recently emerged as a competitive alternative to 3D CNNs for video understanding. However, due to their large number of parameters and reduced inductive biases, these models require supervised pretraining on large-scale image datasets to achieve top performance. In this paper, we empirically demonstrate that self-supervised pretraining of video transformers on video-only datasets can lead to action recognition results that are on par or better than those obtained with supervised pretraining on large-scale image datasets, even massive ones such as ImageNet-21K. Since transformer-based models are effective at capturing dependencies over extended temporal spans, we propose a simple learning procedure that forces the model to match a long-term view to a short-term view of the same video. Our approach, named Long-Short Temporal Contrastive Learning (LSTCL), enables video transformers to learn an effective clip-level representation by predicting temporal context captured from a longer temporal extent. To demonstrate the generality of our findings, we implement and validate our approach under three different self-supervised contrastive learning frameworks (MoCo v3, BYOL, SimSiam) using two distinct video-transformer architectures, including an improved variant of the Swin Transformer augmented with space-time attention. We conduct a thorough ablation study and show that LSTCL achieves competitive performance on multiple video benchmarks and represents a convincing alternative to supervised image-based pretraining.
Accepted in CVPR 2022
References in corpus (9)
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
- Two-Stream Convolutional Networks for Action Recognition in Videos
- Bootstrap your own latent: A new approach to self-supervised Learning
- The Kinetics Human Action Video Dataset
- Is Space-Time Attention All You Need for Video Understanding?
- Exploring Simple Siamese Representation Learning
- VATT: Transformers for Multimodal Self-Supervised Learning from Raw Video, Audio and Text
- An Image is Worth 16x16 Words, What is a Video Worth?
- Multiscale Vision Transformers