VideoMAE: Masked Autoencoders are Data-Efficient Learners for Self-Supervised Video Pre-Training
Abstract
Pre-training video transformers on extra large-scale datasets is generally required to achieve premier performance on relatively small datasets.
In this paper, we show that video masked autoencoders (VideoMAE) are data-efficient learners for self-supervised video pre-training (SSVP).
We are inspired by the recent ImageMAE [31] and propose customized video tube masking with an extremely high ratio.
This simple design makes video reconstruction a more challenging and meaningful self-supervision task, thus encouraging extracting more effective video representations during the pre-training process.
We obtain three important findings with VideoMAE: (1) An extremely high proportion of masking ratio (i.e., 90% to 95% ) still yields favorable performance for VideoMAE.
The temporally redundant video content enables higher masking ratio than that of images.
(2) VideoMAE achieves impressive results on very small datasets (i.e., around 3k-4k videos) without using any extra data.
This is partially ascribed to the challenging task of video reconstruction to enforce high-level structure learning.
(3) VideoMAE shows that data quality is more important than data quantity for SSVP.
Domain shift between pre-training and target datasets is an important factor.
Notably, our VideoMAE with the vanilla ViT backbone can achieve 87.4% on Kinects-400, 75.4% on SomethingSomething V2, 91.3% on UCF101, and 62.6% on HMDB51, without using any extra data.
Code is available at https://github.com/MCG-NJU/VideoMAE.
Introduction
Transformer [71] has brought significant progress in natural language processing [18, 7, 55].
The vision transformer [21] also improves a series of computer vision tasks including image classification [67, 89], object detection [8, 38], semantic segmentation [81], object tracking [14, 17], and video recognition [6, 3].
The multi-head self-attention upon linearly projected image/video tokens is capable of modeling global dependency among visual content either spatially or temporally.
The inductive bias is effectively reduced via this flexible attention mechanism.
Training effective vision transformers (ViTs) typically necessitates large-scale supervised datasets.
Initially, the pre-trained ViTs achieve favorable performance by using hundreds of millions of labeled images [21].
For video transformers [3, 6], they are usually derived from image-based transformers and heavily depend on the pre-trained models from large-scale image data (e.g., ImageNet [58]).
Previous trials [3, 6] on training video transformers from scratch yield unsatisfied results (except for MViT [22] with a strong inductive bias).
Therefore, the learned video transformers are naturally biased by image-based models, and it still remains a challenge that how to effectively and efficiently train a vanilla vision transformer on the video dataset itself without using any pre-trained model
[Figure]
[Figure]
or extra image data.
Moreover, the existing video datasets are relatively small compared with image datasets, which further increases the difficulty of training video transformers from scratch.
Meanwhile, self-supervised learning has shown remarkable performance by using large-scale image datasets [15, 9].
The learned representations have outperformed the ones via supervised learning when being transferred to downstream tasks.
It is expected that this self-supervised learning paradigm can provide a promising solution to address the challenge of training video transformers.
Following the success of masked autoencoding in NLP [18] and images [31, 4], we present a new selfsupervised video pre-training (SSVP) method, termed as Video Masked Autoencoder (VideoMAE).
Our VideoMAE inherits the simple pipeline of masking random cubes and reconstructing the missing ones.
However, the extra time dimension of videos makes them different from images in this masked modeling.
First, video frames are often densely captured, and their semantics varies slowly in time [88].
This temporal redundancy would increase the risk of recovering missing pixels from the spatiotemporal neighborhood with little high-level understanding.
Furthermore, video could be viewed as the temporal evolution of static appearance, and there exists a correspondence between frames.
This temporal correlation could lead to information leakage (i.e., masked spatiotemporal content re-occurrence) during reconstruction unless a specific masking strategy is considered.
In this sense, for each masked cube, it is easy to find a corresponding and unmasked copy in adjacent frames.
This property would make the learned models identify some “shortcut” features that are hard to generalize to new scenarios.
To make video masked modeling more effective, in this paper, we present a customized design of tube masking with an extremely high ratio in our VideoMAE.
First, due to temporal redundancy, we use an extremely high masking ratio to drop the cubes from the downsampled clips.
This simple strategy not only effectively increases the pre-training performance but also greatly reduces the computational cost due to the asymmetric encoder-decoder architecture.
Second, to consider temporal correlation, we devise a simple yet effective tube masking strategy, which turns out to be helpful in relieving the risk of information leakage for cubes with no or negligible motion during reconstruction.
With this simple yet effective design in our VideoMAE, we are able to successfully train vanilla ViT backbones on the relatively small-scale video datasets such as Something-Something [26], UCF101 [61], and
HMDB51 [35], which significantly outperform the previous state of the art under the setting without extra data.
In summary, the main contribution of this paper is threefold:
• We present a simple but effective video masked autoencoder that unleashes the potential of vanilla vision transformer for video recognition.
To the best of our knowledge, this is the first masked video pre-training framework of simply using plain ViT backbones.
To relieve the information leakage issue in masked video modeling, we present the tube masking with an extremely high ratio, which brings the performance improvement to the VideoMAE.
• Aligned with the results in NLP and Images on masked modeling, our VideoMAE demonstrates that this simple masking and reconstruction strategy provides a good solution to self-supervised video pre-training.
The models pre-trained with our VideoMAE significantly outperform those trained from scratch or pre-trained with contrastive learning methods.
• We obtain extra important findings on masked modeling that might be ignored in previous research in NLP and Images.
(1) We demonstrate that VideoMAE is a data-efficient learner that could be successfully trained with only 3.5k videos.
(2) Data quality is more important than quantity for SSVP when a domain shift exists between the source and target dataset.
Related Work
Video representation learning.
Learning good video representations has been heavily investigated in the literature.
The supervised learning methods [59, 76, 70, 10, 6] usually depend on the image backbones.
The video encoder backbones are first pre-trained with image data in a supervised form.
Then, these backbones are fine-tuned on the video dataset for classifying human actions.
Meanwhile, some methods [68, 23, 22] directly train video backbones from videos in a supervised manner.
Besides supervised learning, semi-supervised video representation learning has also been studied [60].
The representations of labeled training samples are utilized to generate supervision signals for unlabeled ones.
Supervised or semi-supervised representation learning mainly uses a top-down training paradigm, which is not effective in exploring the inherent video data structure itself.
Meanwhile, some multimodal contrastive learning methods [37, 43, 63] have been developed to learn video representation from noisy text supervision.
For self-supervised learning, the prior knowledge of temporal information has been widely exploited to design pretext tasks [79, 45, 83, 5] for SSVP.
Recently, contrastive learning [29, 46, 30, 53, 25, 28] is popular to learn better visual representation.
However these methods heavily rely on strong data augmentation and large batch size [24].
Predicting the video clip with autoencoders in pixel space has been explored for representation learning by using CNN or LSTM backbones [49, 62], or conducting video generation with autoregressive GPT [84].
Instead, our VideoMAE aims to use the simple masked autoencoder with recent ViT backbones to perform data-efficient SSVP.
Masked visual modeling.
Masked visual modeling has been proposed to learn effective visual representations based on the simple pipeline of masking and reconstruction.
These works mainly focus on the image domain.
The early work [73] treated the masking as a noise type in denoised autoencoders [72] or inpainted missing regions with context [48] by using convolutions.
iGPT [11] followed the success of GPT [7, 56] in NLP and operated a sequence of pixels for prediction.
The original ViT [21] investigated the masked token prediction for self-supervised pre-training.
More recently, the success of vision transformer has led to investigation of Transformer-based architectures for masked visual modeling [4, 20, 31, 80, 82, 90].
BEiT [4], BEVT [77] and VIMPAC [65] followed
BERT [18] and proposed to learn visual representations from images and videos by predicting the discrete tokens [57].
MAE [31] introduced an asymmetric encoder-decoder architecture for masked image modeling.
MaskFeat [80] proposed to reconstruct the HOG features of masked tokens to perform self-supervised pre-training in videos.
VideoMAE is inspired by the ImageMAE and introduces specific design in implementation for SSVP.
In particular, compared with previous masked video modeling [31, 77, 65], we present a simpler yet more effective video masked autoencoder by directly reconstructing the pixels.
Our VideoMAE is the first masked video pre-training framework of simply using plain ViT backbones.
Proposed Method
In this section, we first revisit ImageMAE [31].
Then we analyze the characteristics of video data.
Finally, we show how we explore MAE in the video data by presenting our VideoMAE.
3.1 Revisiting Image Masked Autoencoders
ImageMAE [31] performs the masking and reconstruction task with an asymmetric encoder-decoder architecture.
The input image I ∈ R 3 × H × W is first divided into regular non-overlapping patches of size 16 × 16, and each patch is represented with token embedding .
Then a subset of tokens are randomly masked with a high masking ratio (75%), and only the remaining ones are fed into the transformer encoder Φ enc .
Finally, a shallow decoder Φ dec is placed on top of the visible tokens from the encoder and learnable mask tokens to reconstruct the image.
The loss function is mean squared error (MSE) loss between the normalized masked tokens and reconstructed ones in the pixel space:
[Equation]
where p is the token index, Ω is the set of masked tokens, I
[Figure]
3.2 Characteristics of Video Data
Compared with static images, video data contain temporal relations.
We show the motivation of our
VideoMAE by analyzing video characteristics.
Temporal redundancy.
There are frequently captured frames in a video.
The semantics vary slowly in the temporal dimension [88].
We observe that consecutive frames are highly redundant, as shown in Figure 2.
This property leads to two critical issues in masked video autoencoding.
First, it would be less efficient to keep the original temporal frame rate for pre-training.
This would draw us to focus more on static or slow motions in our masked modeling.
Second, temporal redundancy greatly dilutes motion representations.
This would make the task of reconstructing missing pixels not difficult under the normal masking ratio (e.g., 50% to 75%).
The encoder backbone is not effective in capturing motion representations.
Temporal correlation.
Videos could be viewed as the temporal extension of static appearance, and therefore there exists an inherent correspondence between adjacent frames.
This temporal correlation could increase the risk of information leakage in the masking and reconstruction pipeline.
In this sense, as shown in Figure 2, we can reconstruct the masked patches by finding the spatiotemporal corresponding unmasked patches in the adjacent frames under plain random masking or frame masking.
In this case, it might guide the VideoMAE to learn low-level temporal correspondence rather than high-level information such as spatiotemporal reasoning over the content.
To alleviate this behavior, we need to propose a new masking strategy to make the reconstruction more challenging and encourage effective learning of spatiotemporal structure representations.
3.3 VideoMAE
To relieve the above issues in video masked modeling, we make the customized design in our
VideoMAE, and the overall pipeline is shown in Figure 1.
Our VideoMAE takes the downsampled frames as inputs and uses the cube embedding to obtain video tokens.
Then, we propose a simple design of tube masking with high ratio to perform MAE pre-training with an asymmetric encoderdecoder architecture.
Our backbone uses the vanilla ViT with joint space-time attention .
Temporal downsampling.
According to the above analysis on temporal redundancy over consecutive frames, we propose to use the strided temporal sampling strategy to perform more efficient video pre-training.
Formally, one video clip consisting of t consecutive frames is first randomly sampled from the original video V .
We then use temporal sampling to compress the clip to T frames, each of which contains H × W × 3 pixels.
In experiments, the stride τ is set to 4 and 2 on Kinetics and
Something-Something, respectively.
Cube embedding.
We adopt the joint space-time cube embedding [3, 22, 39] in our VideoMAE, where we treat each cube of size 2 × 16 × 16 as one token embedding.
Thus, the cube embedding layer obtains T 2 × H 16 × W 16 3D tokens and maps each token to the channel dimension D .
This design can decrease the spatial and temporal dimension of input, which helps to alleviate the spatiotemporal redundancy in videos.
Tube masking with extremely high ratios.
First, temporal redundancy is a factor affecting VideoMAE design.
We find that VideoMAE is in favor of extremely high masking ratios (e.g. 90% to 95%) compared with the ImageMAE.
Video information density is much lower than images, and we expect a high ratio to increase the reconstruction difficulty.
This high masking ratio is helpful to mitigate the information leakage during masked modeling and make masked video reconstruction a meaningful self-supervised pre-training task.
Second, temporal correlation is another factor in our VideoMAE design.
We find even under the extremely high masking ratio, we can still improve the masking efficiency by proposing the temporal tube masking mechanism.
Temporal tube masking enforces a mask to expand over the whole temporal axis, namely, different frames sharing the same masking map.
Mathematically, the tube mask mechanism can be expressed as I [ p x,y, · ∈ Ω] ∼ Bernoulli( ρ mask ) and different time t shares the same value.
With this mechanism, temporal neighbors of masked cubes are always masked.
So for some cubes with no or small motion (e.g., finger cube in 4th row of Figure 2 (d)), we can not find the spatiotemporal corresponding content in all frames.
In this way, it would encourage our
VideoMAE to reason over high-level semantics to recover these totally missing cubes.
This simple strategy can alleviate the information leakage for cubes with no or negligible motion, and turns out to be effective in practice for masked video pre-training.
Backbone: joint space-time attention.
Due to the high proportion of masking ratio mentioned above, only a few tokens are left as the input for the encoder.
To better capture high-level spatio-temporal information in the remaining tokens, we use the vanilla ViT backbone [21] and adopt the joint space-time attention [3, 39].
Thus, all pair tokens could interact with each other in the multi-head self-attention layer [71].
The specific architecture design for the encoder and decoder is shown in supplementary materials.
The quadratic complexity of the joint space-time attention mechanism is a computational bottleneck, while our design of an extremely high masking ratio alleviates this issue by only putting the unmasked tokens (e.g., 10%) into the encoder during the pre-training phase.
Experiments 4.1 Datasets
We evaluate our VideoMAE on five common video datasets: Kinetics-400 [34], Something-Something
V2 [26], UCF101 [61], HMDB51 [35], and AVA [27].
The Kinetics-400 contains around 240k training videos and 20k validation videos of 10s from 400 classes.
The Something-Something V is another large-scale video dataset, having around 169k videos for training and 20k videos for validation.
In contrast to Kinetics-400, this dataset contains 174 motion-centric action classes.
These two large-scale video datasets focus on different visual cues for action recognition.
UCF101 and
HMDB51 are two relatively small video datasets, which contain around 9.5k/3.5k train/val videos and 3.5k/1.5k train/val videos, respectively.
Compared with those large-scale video datasets, these two small datasets are more suitable for verifying the effectiveness of VideoMAE, as training large
ViT models is more challenging on small datasets.
Moreover, we also transfer the learned ViT models by VideoMAE to downstream action detection task.
We work on AVA, a dataset for spatiotemporal localization of human actions with 211k training and 57k validation video segments.
In experiments of downstream tasks, we fine-tune the pre-trained VideoMAE models on the training set and report the results on the validation set.
The implementation details are described in Appendix § B.
4.2 Ablation Studies
In this subsection, we perform in-depth ablation studies on VideoMAE design with the default backbone of 16-frame ViT-B on Something-Something V2 (SSV2) and Kinetics-400 (K400).
The specific architectures for the encoder and decoder are shown in Appendix § A. For fine-tuning, we perform TSN [76] uniform sampling on SSV2 and dense sampling [78, 23] on K400.
All models share the same inference protocol, i.e., 2 clips × 3 crops on SSV2 and 5 clips × 3 crops on K400.
Decoder design.
The lightweight decoder is one key component of our VideoMAE.
We conduct experiments with the different depths in Table 1a.
Unlike in ImageMAE, a deep decoder here is important for better performance, while a shallow decoder could reduce the GPU memory consumption.
We take 4 blocks for the decoder by default.
The decoder width is set to half channel of the encoder (e.g., 384-d for ViT-B), following the design in the image domain.
[Table]
Masking strategy.
We compare different masking strategies in Table 1b.
When increasing the masking ratio from 75% to 90% for tube masking, the performance on SSV2 boosts from 68.0% to 69.6%.
Then, with an extremely high ratio, we find tube masking also achieves better performance than plain random masking and frame masking.
We attribute these interesting observations to the redundancy and temporal correlation in videos.
The conclusion on K400 is in accord with one on
SSV2.
One may note that the performance gap on K400 is lower than one on SSV2.
We argue that the Kinetics videos are mostly stationary and scene-related.
The effect of temporal modeling is not obvious.
Overall, we argue that our default designs enforce the networks to capture more useful spatiotemporal structures and therefore make VideoMAE a more challenging task, which a good self-supervised learner hunger for
Reconstruction target.
First, if we only employ the center frame as the target, the results would decrease greatly as shown in Table 1c.
The sampling stride is also sensitive.
The result of small sampling stride τ 2 is lower than default sampling stride τ (68.9% vs. 69.6% on SSV2).
We also try to reconstruct 2 T frames from the downsampled T frames, but it obtains slightly worse results on
SSV2.
For simplicity, we use the input downsampled clip as our default reconstruction target.
Pre-training strategy.
We compare different pre-training strategies in Table 1d.
Similar to previous trials [3, 6], training video transformers from scratch yields unsatisfied results on video datasets.
When pre-trained on the large-scale ImageNet-21K dataset, the video transformer obtains better accuracy from 32.6% to 61.8% on SSV2 and 68.8% to 78.9% on K400.
Using the models pre-trained on both ImageNet-21K and Kinetics further increases accuracy to 65.2% on SSV2.
Our VideoMAE can effectively train a video transformer on the video dataset itself without using any extra data and achieve the best performance (69.6% on SSV2 and 80.0% on K400).
Pre-training dataset.
First, we pre-train the ViT-B on ImageNet-1K for 1600 epochs, following the recipes in [31].
Then we inflate the 2D patch embedding layer to our cube embedding layer following [10] and fine-tune the model on the target video datasets.
The results surpass the model trained from scratch as shown in Table 1e.
We also compare the ImageMAE pre-trained model with
VideoMAE models pre-trained on video datasets.
We see that our VideoMAE models can achieve better performance than ImageMAE.
However, when we try to transfer the pre-trained VideoMAE models to the other video datasets (e.g. from Kinetics to Something-Something), the results are slightly worse than their counterpart, which is directly pre-trained on its own target video datasets.
We argue that domain shift between pre-training and target datasets could be an important issue.
[Table]
[Table] [Table]
Loss function.
Table 1f contains an ablation study of loss function.
We find that the MSE loss could achieve a higher result compared with the L1 loss and smooth L1 loss.
Therefore, we employ the
MSE loss by default.
4.3 Main Results and Analysis
VideoMAE: data-efficient learner.
The self-supervised video pre-training (SSVP) has been extensively studied in previous works, but they mainly use the CNN-based backbones.
Few works have investigated transformer-based backbone in SSVP.
Therefore, to demonstrate the effectiveness of
VideoMAE for transformer-based SSVP, we compare two methods implemented by ourselves: (1) training from scratch and (2) pre-training with contrastive learning (MoCo v3 [15]).
For training from scratch, we carefully tune these hyper-parameters to successfully pre-train ViT-Base from the training set of the dataset.
For pre-training with MoCo v3, we strictly follow the training practice in its image counterpart and carefully avoid the collapse issue.
The recognition accuracy is reported in Table 2.
We see that our VideoMAE significantly outperforms other two training settings.
For instance, on the largest dataset of Kinetics-400, our VideoMAE outperforms training from scratch by around 10% and MoCo v3 pre-training by around 5%.
This superior performance demonstrates that masked autoencoder provides an effective pre-training mechanism for video transformers.
We also see that the performance gap between our VideoMAE and the other two methods becomes larger as the training set becomes smaller.
Notably, even with only 3.5k training clips on HMDB51, our VideoMAE pre-training can still obtain a satisfying accuracy (around 61%).
This new result demonstrates that VideoMAE is a more data-efficient learner for SSVP.
This property is particularly important for scenarios with limited data available and different with contrastive learning methods.
We compare the efficiency of VideoMAE pre-training and MoCo v3 pre-training in Table 3.
The task of masked autoencoding with a high ratio is more challenging and thereby requires more training epochs (800 vs. 300).
Thanks to the asymmetric encoder-decoder in our VideoMAE and extremely high masking ratio, our pre-training time is much shorter than MoCo v3 (19.5 vs. 61.7 hours).
High masking ratio.
In VideoMAE, one core design is the extremely high masking ratio.
We perform an investigation of this design on the Kinetics-400 and Something-Something V2 datasets.
The results are shown in Figure 3.
We see that the best masking ratio is extremely high, and even 95% can achieve good performance for both datasets.
This result is difference from BERT [18] in
NLP and MAE [31] in images.
We analyze the temporal redundancy and correlation in videos makes it possible for our VideoMAE to learn plausible outputs with such a high masking ratio.
[Figure]
an extremely high masking ratio (
%) achieves the best efficiency and effectiveness trade-off on
[Figure]
Our default backbone is 16-frame vanilla ViT-B. • denotes that all models are trained for the same 132k iterations, and • denotes that all models are trained for the sameepochs.
Note that it takes 132k iterations to pre-train the model for 800 epochs on the full training set of SomethingSomething V2.
We also visualize the reconstructed examples in Appendix
§ E.
We see that even under an extremely high masking ratio, VideoMAE can produce satisfying reconstructed results.
This implies VideoMAE is able to learn useful representations that capture the holistic spatiotemporal structure in videos.
Transfer learning: quality vs. quantity.
To further investigate the generalization ability of
VideoMAE in representation learning, we transfer the learned VideoMAE from Kinetics-400 to
Something-Something V2, UCF101, and HMDB51.
The results are shown in Table 4, and we compare them with MoCo v3 pre-training.
The models pre-trained by VideoMAE are better than those pre-trained by MoCo v3, demonstrating that our VideoMAE learns more transferable representations.
Comparing Table 2 and Table 4, the transferred representation outperforms the original VideoMAE models trained from its own dataset on UCF101 and HMDB51.
In contrast, the transferred representation is worse on Something-Something V2.
To figure out whether this inconsistent result is caused by the large scale of Something-Something V2, we further perform a detailed investigation by decreasing the pre-training video numbers.
In this study, we run two experiments: (1) pre-training with the same epochs and (2) pre-training with the same time budget.
The result is shown in Figure 4.
We see that more training iterations could contribute to better performance when we decrease the size of the pre-training set.
Surprisingly, even with only 42k pre-training videos, we can still obtain better accuracy than the Kinetics pre-trained models with 240k videos (68.7% vs. 68.5%).
This result implies that domain shift is another important factor, and data quality is more important than data quantity in SSVP when there exists a difference between pre-training and target datasets.
It also demonstrates that VideoMAE is a data-efficient learner for SSVP.
Transfer learning: downstream action detection.
We also transfer the learned VideoMAE on
Kinetics-400 to downstream action detection dataset AVA.
Following the standard setting [27], we evaluate on top 60 common classes with mean Average Precision (mAP) as the metric under IoU threshold of 0.5.
The results are shown in the Table 5.
After self-supervised pre-training on Kinetics400, our VideoMAE with the vanilla ViT-B can achieve 26.7 mAP on AVA, which demonstrates the strong transferability of our VideoMAE.
If the pre-trained ViT-B is additionally fine-tuned on
Kinetics-400 with labels, the transfer learning performance can further increase about 5 mAP (from 26.7 to 31.8).
More remarkably, when we scale up the pre-training configurations with larger video datasets (e.g. Kinetics-700) or more powerful backbones (e.g. ViT-Large and ViT-Huge), VideoMAE can finally obtain better performance.
For example, our ViT-L VideoMAE pre-trained on Kineticsachieves 39.3 mAP and ViT-H VideoMAE pre-trained on Kinetics-400 has 39.5 mAP.
These results demonstrate that the self-supervised pre-trained models transfer well not only on action classification task but on more complex action detection task.
[Table] [Table]
4.4 Comparison with the state of the art
We compare with the previous state-of-the-art performance on the Kinetics-400 and SomethingSomething V2 datasets.
The results are reported in Table 6 and Table 7.
Our VideoMAE can easily scale up with more powerful backbones (e.g. ViT-Large and ViT-Huge) and more frames (e.g. 32).
Our VideoMAE achieves the top-1 accuracy of 75.4% on Something-Something V2 and 87.4% on
Kinetics-400 without using any extra data.
We see that the existing state-of-the-art methods all depend on the external data for pre-training on the Something-Something V2 dataset.
On the contrary, our VideoMAE without any external data significantly outperforms previous methods with the same input resolution by around 5%.
Our ViT-H VideoMAE also achieves very competitive performance on the Kinetics-400 dataset without using any extra data, which is even better than ViViT-H with on
[Table]
JFT-300M pre-training (86.6% v.s. 84.9%).
When fine-tuned with larger spatial resolutions and input video frames, the performance of our ViT-H VideoMAE can further boost from 86.6% to 87.4%.
Conclusion
In this paper
[Table]
we have presented a simple and data-efficient self-supervised learning method (VideoMAE) for video transformer pre-training.
Our VideoMAE introduces two critical designs of extremely high masking ratio and tube masking strategy to make the video reconstruction task more challenging.
This harder task would encourage VideoMAE to learn more representative features and relieve the information leakage issue.
Empirical results demonstrate this simple algorithm works well for video datasets of different scales.
In particular, we are able to learn effective VideoMAE only with thousands of video clips, which has significant practical value for scenarios with limited data available.
Future work VideoMAE could be further improved by using larger webly datasets, larger models (e.g., ViT-G) and larger spatial resolutions of input video (e.g., 384 2 ).
VideoMAE only leverages the
RGB video stream without using additional audio or text stream.
We expect that audio and text from the video data can provide more information for self-supervised pre-training.
Broader impact Potential negative societal impacts of VideoMAE are mainly concerned with energy consumption.
The pre-training phase may lead to a large amount of carbon emission.
Though the pre-training is energy-consuming, we only need to pre-train the model once.
Different downstream tasks can then share the same pre-trained model via additional fine-tuning.
Our VideoMAE unleashes the great potential of vanilla vision transformer for video analysis, which could increase the risk of video understanding model or its outputs being used incorrectly, such as for unauthorized surveillance.
Acknowledgements and disclosure of funding Thanks to Ziteng Gao, Lei Chen and Chongjian
Ge for their help.
This work is supported by National Natural Science Foundation of China (No.
62076119, No. 61921006), the Fundamental Research Funds for the Central Universities (No.
020214380091), Tencent AI Lab Rhino-Bird Focused Research Program (No. JR202125), and
Collaborative Innovation Center of Novel Software Technology and Industrialization.
Appendix
In this appendix
[Table]
we provide more details of VideoMAE from the following aspects:
• The detailed architecture illustration is in § A. • The implementation details are in § B. • Experimental results are in § C where there are ablation studies on the Something-Something V2 and Kinectics-400 datasets, and a downstream evaluation task (i.e., action detection).
More comparisons with state-of-the-art methods on the UCF101 and HMDB51 datasets are included as well.
• Results analysis is in § D. • Visualization of reconstructed samples is in § E. • License of the datasets is in § F.
A Architectures
We use an asymmetric encoder-decoder architecture for video self-supervised pre-training and discard the decoder during the fine-tuning phase.
We take the 16-frame vanilla ViT-Base for example
[Table]
and the specific architectural design for the encoder and decoder is shown in Table 8.
We adopt the joint space-time attention [3, 39] to better capture the high-level spatio-temporal information in the remaining tokens
[Table]
16-frame vanilla ViT-Base for example.
“MHA” here denotes the joint space-time self-attention.
The output sizes are denoted by { C × T × S } for channel, temporal and spatial sizes.
B Implementation Details
We conduct the experiments with 64 GPUs for both pre-training and fine-tuning on the SomethingSomething V2 and Kinetics-400 datasets.
The experiments on the smaller UCF101 and HMDB datasets are trained with 8 GPUs.
The experiments on the AVA dataset are conducted with 32 GPUs.
We linearly scale the base learning rate w.r.t.
the overall batch size, lr = base learning rate × batch size / 256 .
We adopt the PyTorch [47] and DeepSpeed 2 frameworks for faster training.
We have made the code 3 and pre-trained models 4 public to facilitate future research in self-supervised video pre-training.
Something-Something V2.
Our VideoMAE is pre-trained for 800 epochs on Something-Something
V2 by default.
During the fine-tuning phase, we perform the uniform sampling following TSN [76].
For evaluation, all models share the same inference protocol, i.e., 2 clips × 3 crops.
The default
[Table] [Table]
[Table]
settings of pre-training, fine-tuning, and linear probing are shown in Table 9, Table 12, and Table 11.
For supervised training, we follow the recipe in [22] and train from scratch for 100 epochs.
Note that we use no flip augmentation during both the pre-training and fine-tuning phase.
We additionally adopt the repeated augmentation [32] during the fine-tuning phase in Table 6, which can further increase the Top-1 accuracy by 0.1% - 0.3%.
Kinetics-400.
Our VideoMAE is pre-trained for 800 epochs on Kinetics-400 by default.
During the fine-tuning phase, we perform the dense sampling following Slowfast [23].
For evaluation, all models share the same inference protocol, i.e., 5 clips × 3 crops.
The default settings of pre-training and fine-tuning are shown in Table 9 and Table 12.
For supervised training from scratch, we follow the recipe in [22] and train the model for 200 epochs.
Note that we adopt the repeated augmentation [32] during the fine-tuning phase in Table 7, which can further increase the Top-1 accuracy by 0.8% - 1.0%.
[Table]
AVA v2.
[Table]
UCF101.
We follow a similar recipe on Kinetics for pre-training.
Our VideoMAE is pre-trained with a masking ratio of 75% for 3200 epochs.
The batch size and base learning rate are set toand 3e-4, respectively.
Here, 16 frames with a temporal stride of 4 are sampled.
For fine-tuning, the model is trained with repeated augmentation [32] and a batch size of 128 for 100 epochs.
The base learning rate, layer decay and drop path are set to 5e-4, 0.7 and 0.2, respectively.
For evaluation, we adopt the inference protocol of 5 clips × 3 crops.
HMDB51.
Our VideoMAE is pre-trained with a masking ratio of 75% for 4800 epochs.
The batch size and base learning rate are set to 192 and 3e-4, respectively.
Here, 16 frames with a temporal stride of 2 are sampled.
For fine-tuning, the model is trained with repeated augmentation [32] and a batch size of 128 for 50 epochs.
The base learning rate, layer decay and drop path are set to 1e-3, 0.
and 0.2, respectively.
For evaluation, we adopt the inference protocol of 10 clips × 3 crops.
AVA.
We follow the action detection architecture in Slowfast [23] and use the detected person boxes from AIA [66].
The default settings of fine-tuning are shown in Table 12.
For data augmentations, we resize the short side of the input frames to 256 pixels.
We apply a random crop of the input frames to 224 × 224 pixels and random flip during training.
We use only ground-truth person boxes for training and the detected boxes with confidence ≥ 0.8 for inference.
C Additional Results C.1 Training schedule
Figure 5 shows the influence of the longer pre-training schedule on the Something-Something V2 and
Kinetics-400 datasets.
We find that a longer pre-training schedule brings slight gains to both datasets.
In the main paper, our VideoMAE is pre-trained for 800 epochs by default.
C.2 Comparison with the state-of-the-art methods
We present the detailed comparison with the state-of-the-art on UCF101 and HMDB51 in Table 13.
Figure 6 additionally shows that our VideoMAE is a data-efficient learner that allows us to effectively train video transformers only from limited video data (e.g., 9.5k clips in UCF101, and 3.5k clips in
HMDB51) without any ImageNet pre-training.
VideoMAE significantly outperforms training from scratch, MoCo v3 pre-training [15], and the previous best performance from Vi 2 CLR [19] without extra data on these small-scale video datasets.
Compared with those large-scale video datasets, these two small datasets are more proper to verify the effectiveness of VideoMAE, as training large ViT models is more challenging on small datasets.
D Model result analysis
In this section, we add the analysis of model results.
As shown in Figure 7 and Figure 8, our
VideoMAE bring significant gain for most categories on SSV2, which implies that our VideoMAE
[Figure]
Here each point is a full training schedule.
Our default ViT-B backbone is described in Table
[Figure]
can capture more spatiotemporal structure representations than ImageMAE and ImageNet-21k supervised pre-trained model.
On the other hand, we also notice that our VideoMAE performs slightly worse than other two models on some categories.
To better understand how the model works, we select several examples from validation set.
The examples are shown in Figure 9.
For the example in the 1st row, we find our VideoMAE might not capture the motion information from very small object.
We suspect that tokens containing the small motion might all be masked due to our extremely high masking ratio, so our VideoMAE could hardly reconstruct the masked small motion pattern.
For the example in the 2nd row, we find our VideoMAE could capture the deformation of objects and movement from the squeeze of the hand, while this cannot be discriminated by image pre-training.
We leave more detailed analysis of our VideoMAE for future work.
E Visualization
We show several examples of reconstruction in Figure 10 and Figure 11.
Videos are all randomly chosen from the validation set.
We can see that even under an extremely high masking ratio,
VideoMAE can produce satisfying reconstructed results.
These examples imply that our VideoMAE
[Table]
Slow-R152 (2 × )
Kinetics328M
V 94.
CORP f [33]
SlowOnly-R
Kinetics32M
V 93.
ρ SimCLR ρ =2 [24] SlowOnly-R
Kinetics32M
V 88.
N/A ρ SwAV ρ =2 [24]
SlowOnly-R
Kinetics32M
V 87.
N/A ρ MoCo ρ =2 [24]
SlowOnly-R
Kinetics32M
V 91.
N/A ρ BYOL ρ =2 [24]
SlowOnly-R
Kinetics32M
V 92.
N/A ρ BYOL ρ =4 [24]
SlowOnly-R
Kinetics32M
V 94.
MIL-NCE [44]
S3D
HowTo100M 9M
V+T 91.
MMV [1]
S3D-G
AS+HTM 9M
V+A+T 92.
CPD [37]
ResNet
IG300k
N/A
V+T 92.
ELO [52]
R(2+1)D
Youtube8MN/A
N/A
V+A 93.
XDC [2]
R(2+1)D
Kinetics15M
V+A 84.
XDC [2]
R(2+1)D
IG65M 15M
V+A 94.
GDT [50]
R(2+1)D
Kinetics15M
V+A 89.
GDT [50]
R(2+
[Table]
is able to learn more representative features that capture the holistic spatiotemporal structure in videos.
F License of Data
All the datasets we used are commonly used datasets for academic purpose.
The license of the
Something-Something V2 5 and UCF101 6 datasets is custom.
The license of the Kinetics-400 7 ,
HMDB51 8 and AVA 9 datasets is CC BY-NC 4.0 10 .
B e n d i n g s o m e t h
i n g s o t h a t i t d e f o r m s
B e n d i n g s o m e t h
i n g un t il
i
t
b r e a k s
D r
o
pp i n g s o m e t h
i n g b e h i n d s o m e t h i n g
H i tt i n g s o m e t h i n g w i t h s o m e t h
i n g
H o l d i n g s o m e t h i n g b e h i n d s o m e t h i n g
L e tt i n g s o m e t h i n g r o ll u p a s l a n t e d s u r f a c e , s o i t r o ll s b a c k d o w n
L i f t i n g a s u r f a c
e
w
i
t h s o m
e t h
i n g o n i t b u t n o t e n o u g h
f
o r
i t t o s
li d e d o w n
M o v i n g s o m e t
h
i n g a c r o ss a s u r f a c e un t il i t
f a ll s d o w n
M o v i n g s o m e t
h
i n g a c r o ss a s u r f a c
e
w
i
t
h o u t
i t f a lli n g d
o w n
M o v i n g s o m e t
h
i n g a n d s o m e t h i n g s o t
h e y p a ss e a c h o t
h e r
M o v i n g s o m e t
h
i n g t
o w a r d s t
h e c a m e r a
P i
c
k
i n g s o m e t h i n g u p
P o k i n g a h o l e i n t
o s o m e t h i n g s o f t
P o k i n g a s t a c k o f s o m e t
h
i n g s o t h e s t a c k c o ll a p s e s
P o k i n g s o m e t
h
i n g s o
i t s li g h t
l y m o v e s
P o u r i n g s o m e t
h
i n g o n t o s o m e t h i n g
P r e t e n d i n g o r f a ili n g t
o w i
p e s o m e t
h i n g o ff o f s o m
e t
h
i n g
P r e t e n d i n g o r t r y i n g a n d f a ili n g t
o t w i s t s o m e t h
i n g
P r e t e n d i n g t o c l o s e s o m e t
h
i n g w i t
h o u t a c t u a ll y c
l o s i n g i t
P r e t e n d i n g t o p o k e s o m e t
h
i n g
P r e t e n d i n g t o p o u r s o m e t h i n g o u t
o f s o m e t h
i n g , b u t s o m e t h i n g
i s e m p t y
P r e t e n d i n g t o p u t s o m e t h i n g b e h i n d s o m e t
h
i n g
P r e t e n d i n g t o p u t s o m e t h i n g i n t o s o m e t h
i n g
P r e t e n d i n g t o p u t s o m e t h i n g n e x t t o s o m e t h
i n g
P r e t e n d i n g t o s p
r
e a d a i r o n t
o s o m e t h
i n g
P r e t e n d i n g t o s q u ee
z e
s o m e t
h
i n g
P r e t e n d i n g t o t a k e s o m e t h
i n g o u t
o f s o m e t h
i n g
P r e t e n d i n g t
o t u r n s o m e t h
i n g u p s i d e d o w n
P u lli n g t
w
o
e n d s o f s o m
e t h
i n g b u t n o t
h
i n g h a pp e n s
P u s h
i n g s o m e t h i n g o n t o s o m e t h i n g
P u tt i n g s o m e t h i n g o n t h e e dg e o f s o m e t h
i n g s o
i t i s n o t
s u pp o r t
e d a n d f a ll s d o w n
P u tt i n g s o m e t h i n g o n t
o s o m e t h
i n g e l s e t h a t c a nn o t s u pp o r t
i t s o
i t
f a ll s d o w n
P u tt i n g s o m e t h i n g t h a t c a nn o t a c t u a ll y s t a n d u p r
i g h t u p r
i g h t o n t h e t a b l e , s o i t
f a ll s o n i t s s i d e
R e m o v i n g s o m e t
h
i n g , r e v e a li n g s o m e t h i n g b e h i n d
S c oo p i n g s o m e t
h
i n g u p w i t
h s o m e t
h i n g
S h
o w i n g a p
h
o t
o o f s o m e t h
i n g t o t h e c a m e r a
Sp illi n g s o m e t h
i n g b e h i n d s o m e t h i n g
Sp illi n g s o m e t h i n g n e x t t
o s o m e t h
i n g
Sp illi n g s o m e t h
i n g o n t o s o m e t h i n g
Sp r
e
a
d
i n g s o m e t h i n g o n t
o s o m e t h
i n g
Sq u ee z i n g s o m e t h i n g
T h r o w i n g s o m e t
h
i n g i n t h e a i r a n d c a t c h i n g i t
T r y i n g b u t f a ili n g t
o a tt
a c h s o m e t h
i n g t o s o m e t h i n g b e c a u s e i t d o e s n ' t s t
i c k
T r y i n g t
o b e n d s o m e t
h
i n g un b e n d a b l e s o n o t h i n g h a pp e n s
T r y i n g t
o p o u r
s o m e t h
i n g i n t o s o m e t h i n g , b
u t m i ss i n g s o
i t s p ill s n e x t t o i t
T w i s t
i n g s o m e t h i n g
T o p - 1 A cc u r a c
y
( % ) o n SS
V+12.
+11.7+20.5+14.
+11.
+13.
2.4x 2.3x 2.8x +17.
+10.
+12.
+21.
+17.
+13.
+11.8 3.1x 3.0x +16.
+12.
+10.
+26.1 +16.
+15.
+11.1 +21.
+10.
+17.3 +11.
+11. +11.
+22.
+15.
+14.8+12.
+13.
3.0x 2.1x +12.
+17.
+12.
+14.
+13.2+12.
+15.
+12.
ImageMAE
VideoMAE
A pp r o a c h i n g s o m e t h i n g w i t
h
y
o u r c a m e r a
L e tt i n g s o m e t h i n g r o ll d o w n a s l a n t e d s u r f a c e
M o v i n g s o m e t
h
i n g d o w n
M o v i n g s o m e t
h
i n g u p
P o k i n g a s t a c k o f s o m e t
h
i n g w i t
h o u t t h e s t a c k c o ll a p s i n g
P r e t e n d i n g t o t a k e s o m e t h
i n g f r o m s o m e w h e
r
e
P u s h
i n g s o m e t h i n g s o t h a t i t
f a ll s o f f t h e t a b l e
P u tt i n g s o m e t h i n g t h a t c a n ' t r o ll o n t o a s l a n t e d s u r f a c e , s o i t s t a y s w h e
r
e
i t i s
P u tt i n g s o m e t h i n g u p r
i g h t o n t h e t a b l e
S h
o w i n g s o m e t h
i n g n e x t t o s o m e t h i n g
S h
o w i n g s o m e t h i n g o n t
o p o f s o m e t
h
i n g
S t a c k i n g nu m b e r o f s o m
e t
h
i n g
T a k i n g s o m e t h i n g f r o m s o m e w h e
r e
T u r n i n g t h e c a m e r a r
i g h t w h il e f
[Figure]
m i n g s
[Figure] [Figure]
[Figure] [Figure]
D r
o
pp i n g s o m e t h
i n g b e h i n d s o m e t h i n g
D r
o
pp i n g s o m
e t
h
i n g i n t o s o m e t h i n g
D r
o
pp
i n g s o m e t h i n g n e x t t o s o m e t h i n g
H i tt i n g s o m e t h i n g w i t h s o m e t h
i n g
H o l d i n g s o m e t h i n g b e h i n d s o m e t h i n g H o l d i n g s o m e t h i n g i n f r o n t
o f s o m
e t h
i n g
H o l d i n g s o m e t h i n g n e x t t o s o m e t h i n g
L e tt i n g s o m e t h i n g r o ll u p a s l a n t e d s u r f a c e , s o i t r o ll s b a c k d o w n
L i f t i n g a s u r f a c
e
w
i
t h s o m
e t h
i n g o n i t b u t n o t e n o u g h
f
o r
i t t o s
li d e d o w n
L i f t i n g u p o n e e n d o f s o m e t h i n g , t h e n l e tt i n g i t
d r o p d o w n
M o v i n g s o m e t
h
i n g a n d s o m e t h i n g a w a y f r o m e a c h o t h e r
M o v i n g s o m e t
h
i n g a n d s o m e t h i n g c l o s e r t
o e a c h o t h e r
M o v i n g s o m e t
h
i n g a n d s o m e t h i n g s o t
h e
y c o lli d e w i t
h e a c h o t h e r
M o v i n g s o m e t
h
i n g a n d s o m e t h i n g s o t
h e y p a ss e a c h o t
h e r
M o v i n g s o m e t
h
i n g a w a y f r o m s o m e t h i n g
M o v i n g s o m e t
h
i n g c l o s e r t
o s o m e t h
i n g
M o v i n g s o m e t
h
i n g t
o w a r d s t
h e c a m e r a
P o k i n g a h o l e i n t
o s o m e t h i n g s o f t
P o k i n g a s t a c k o f s o m e t
h
i n g s o t h e s t a c k c o ll a p s e s
P o k i n g s o m e t
h
i n g s o
i t s li g h t
l y m o v e s
P o k i n g s o m e t
h
i n g s o li g h t
l y t
h a t i t
d o e s n ' t
o r a l m o s t
d o e s n ' t m o v e
P r e t e n d i n g o r f a ili n g t
o w i
p e s o m e t
h i n g o ff o f s o m
e t
h
i n g
P r e t e n d i n g t o c l o s e s o m e t
h
i n g w i t
h o u t a c t u a ll y c
l o s i n g i t
P r e t e n d i n g t o o p e n s o m e t h i n g w i t h o u t a c t u a ll y o p e n i n g i t
P r e t e n d i n g t
o p i c k s o m e t h
i n g u p
P r e t e n d i n g t o p o u r s o m e t h i n g o u t
o f s o m e t h
i n g , b u t s o m e t h i n g
i s e m p t y
P r e t e n d i n g t o p u t s o m e t h i n g b e h i n d s o m e t
h
i n g
P r e t e n d i n g t o p u t s o m e t h i n g n e x t t o s o m e t h
i n g
P r e t e n d i n g t o p u t s o m e t h i n g o n a s u r f a c e
P r e t e n d i n g t o s q u ee
z e
s o m e t
h
i n g
P r e t e n d i n g t
o t h r o w s o m
e t h
i n g
P u lli n g s o m e t
h
i n g f r o m b e h i n d o f s o m e t h
i n g
P u lli n g s o m e t
h
i n g f r o m l e f t t
o r i g h t
P u lli n g s o m e t
h
i n g o n t o s o m e t h i n g
P u lli n g t
w
o
e n d s o f s o m
e t h
i n g b u t n o t
h
i n g h a p p e n s
P u lli n g t
w
o
e n d s o f s o m
e t h
i n g s o t h a t i t s e p a r a t
e s i n t o t w o p
i
e
c e s
P u s h
i n g s o m e t h i n g f r o m l e f t t
o r i g h t
P u s h
i n g s o m e t h i n g s o t h a t i t a l m o s t f a ll s o ff b u t d
o e s n ' t
P u s h
i n g s o m e t h i n g s o t h a t i t s li g h t l y m o v e s
P u s h
i n g s o m e t h i n g w i t h s o m e t h
i n g
P u tt i n g s o m e t h i n g a n d s o m e t
h
i n g o n t h e t a b l e
P u tt i n g s o m e t h i n g b e h i n d s o m e t h i n g P u tt i n g s o m e t h i n g t h a t c a nn o t a c t u a ll y s t a n d u p r
i g h t u p r
i g h t o n t h e t a b l e , s o i t
f a ll s o n i t s s i d e
S c oo p i n g s o m e t
h
i n g u p w i t
h s o m e t
h i n g
S h
o w i n g s o m e t h
i n g b e h i n d s o m e t h i n g
S h
o w i n g s o m e t h
i n g n e x t t o s o m e t h i n g
S o m e t
h
i n g c o lli d i n g w i t h s o m e t h
i n g a n d b o t h a r e b e i n g d e f l e c t e d
Sp illi n g s o m e t h i n g n e x t t
o s o m e t h
i n g
T e a r i n g s o m e t h i n g j u s t a li tt l e b
i
t
T h r o w i n g s o m e t
h
i n g a g a i n s t s o m e t h i n g
T h r o w i n g s o m e t
h
i n g i n t h e a i r a n d l e tt i n g i t
f a ll
T o u c h i n g ( w i t
h o u t m o v i n g ) p a r t o f s o m e t
h
i n g
T r y i n g b u t f a ili n g t
o a tt
a c h s o m e t h
i n g t o s o m e t h i n g b e c a u s e i t d o e s n ' t s t
i c k
T r y i n g t
o b e n d s o m e t
h
i n g un b e n d a b l e s o n o t h i n g h a pp e n s
T r y i n g t
o p o u r
s o m e t h
i n g i n t o s o m e t h i n g , b
u t m i ss i n g s o
i t s p ill s n e x t t o i t
T u r n i n g t h e c a m e r a u p w a r d s w h il e f il m i n g s o m e t h
i n g
T o p - 1 A cc u r a c
y
( % ) o n SS V+26.0 +23.6+20.4+31.
+25.
+20.
+31.8 +22.
+21.
+16.2 +23.5+23.
6.0x 2.9x +29.
+28.
+21. +21.
+31.
+20.
+24.
2.4x +16.
+16.
+15.
+17.
4.0x +15.4+17.
2.3x +27.7 +23.
+16.
3.5x +31.
+19.
+16.
+20.4 +25.
+20.9 +23.
+15.7 +15.
+16.9+18.
2.5x +20.
2.4x +15.
+15.
+18.
2.5x +19.7+18.
2.8x +15.8 ImageNet-21k sup.
VideoMAE
B u r y i n g s o m e t h i n g i n s o m e t h i n g
P ili n g s o m e t h i n g u p
P o k i n g a s t a c k o f s o m e t
h
i n g w i t
h o u t t h e s t a c k c o ll a p s i n g
P o k i n g s o m e t
h
i n g s o t h a t i t s p i n s a r o un d
P o u r i n g s o m e t
h
i n g i n t o s o m e t h
i n g un t il
i
t
o v e r f l o w s
P o u r i n g s o m e t
h
i n g o n t o s o m e t h i n g
P r e t e n d i n g o r t r y i n g a n d f a ili n g t
o t w i s t s o m e t h
i n g
P r e t e n d i n g t o p u t s o m e t h i n g un d e r n e a t h s o m e t
h i n g
P r e t e n d i n g t o s p
r
e a d a i r o n t
o s o m e t h
i n g
P u tt i n g s o m e t h i n g o n a f l a t s u r f a c e w i t
h o u t l e t t i n g i t
r o ll
P u tt i n g s o m e t h i n g o n t
o a s l a n t e d s u r f a c e b u t i t
d o e s n ' t g li d e d o w n
P u tt i n g s o m e t h i n g t h a t c a n ' t r o ll o n t o a s l a n t e d s u r f a c e , s o i t s t a y s w h e
r
e
i t i s
P u tt i n g s o
[Figure]
e t h i n g u p r
i g h
[Figure]
t
[Figure]
e t a
[Figure]
l
[Figure]
P u
[Figure]
i n g s o
[Figure]
e t h i n g
[Figure]
s o
[Figure]
e t h i n
[Figure]
a
[Figure]
e t
[Figure]
e t a
[Figure]
o
[Figure]
n
[Figure]
e
[Figure] [Figure]
V
[Figure]
[Figure] [Figure]