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

BEVT: BERT Pretraining of Video Transformers

About

This paper studies the BERT pretraining of video transformers. It is a straightforward but worth-studying extension given the recent success from BERT pretraining of image transformers. We introduce BEVT which decouples video representation learning into spatial representation learning and temporal dynamics learning. In particular, BEVT first performs masked image modeling on image data, and then conducts masked image modeling jointly with masked video modeling on video data. This design is motivated by two observations: 1) transformers learned on image datasets provide decent spatial priors that can ease the learning of video transformers, which are often times computationally-intensive if trained from scratch; 2) discriminative clues, i.e., spatial and temporal information, needed to make correct predictions vary among different videos due to large intra-class and inter-class variations. We conduct extensive experiments on three challenging video benchmarks where BEVT achieves very promising results. On Kinetics 400, for which recognition mostly relies on discriminative spatial representations, BEVT achieves comparable results to strong supervised baselines. On Something-Something-V2 and Diving 48, which contain videos relying on temporal dynamics, BEVT outperforms by clear margins all alternative baselines and achieves state-of-the-art performance with a 71.4\% and 87.2\% Top-1 accuracy respectively. Code will be made available at \url{https://github.com/xyzforever/BEVT}.

Rui Wang, Dongdong Chen, Zuxuan Wu, Yinpeng Chen, Xiyang Dai, Mengchen Liu, Yu-Gang Jiang, Luowei Zhou, Lu Yuan• 2021

Related benchmarks

TaskDatasetResultRank
Action RecognitionSomething-Something v2 (val)
Top-1 Accuracy71.4
535
Action RecognitionKinetics-400
Top-1 Acc80.6
413
Action RecognitionSomething-Something v2
Top-1 Accuracy70.6
341
Action RecognitionSomething-Something v2 (test)
Top-1 Acc70.6
333
Action RecognitionKinetics 400 (test)
Top-1 Accuracy80.6
245
Action RecognitionSomething-Something v2 (test val)
Top-1 Accuracy71.4
187
Video Action RecognitionKinetics-400
Top-1 Acc81.1
184
Video ClassificationSomething-Something v2 (test)
Top-1 Acc0.714
169
Video Action RecognitionKinetics 400 (val)
Top-1 Acc76.2
151
Video Action ClassificationSomething-Something v2
Top-1 Acc71.4
139
Showing 10 of 25 rows

Other info

Code

Follow for update