Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Long-Short Temporal Contrastive Learning of Video Transformers

About

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.

Jue Wang, Gedas Bertasius, Du Tran, Lorenzo Torresani• 2021

Related benchmarks

TaskDatasetResultRank
Action RecognitionSomething-Something v2 (val)
Top-1 Accuracy67
535
Action RecognitionKinetics-400
Top-1 Acc81.5
413
Video ClassificationKinetics 400 (val)
Top-1 Acc81.5
204
Video ClassificationSomething-Something v2 (test)
Top-1 Acc0.67
169
Video RecognitionHMDB51
Accuracy75.9
89
Video ClassificationKinetics-600 (test)
Top-1 Accuracy83.6
48
Video ClassificationUCF101 (full)
Accuracy96.8
17
Showing 7 of 7 rows

Other info

Follow for update