HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Masked Feature Prediction for Self-Supervised Visual Pre-Training

Chen Wei Haoqi Fan Saining Xie Chao-Yuan Wu Alan Yuille Christoph Feichtenhofer

Masked Feature Prediction for Self-Supervised Visual Pre-Training

Abstract

We present Masked Feature Prediction (MaskFeat) for self-supervised pre-training of video models. Our approach first randomly masks out a portion of the input sequence and then predicts the feature of the masked regions. We study five different types of features and find Histograms of Oriented Gradients (HOG), a hand-crafted feature descriptor, works particularly well in terms of both performance and efficiency. We observe that the local contrast normalization in HOG is essential for good results, which is in line with earlier work using HOG for visual recognition. Our approach can learn abundant visual knowledge and drive large-scale Transformer-based models. Without using extra model weights or supervision, MaskFeat pre-trained on unlabeled videos achieves unprecedented results of 86.7% with MViT-L on Kinetics-400, 88.3% on Kinetics-600, 80.4% on Kinetics-700, 39.8 mAP on AVA, and 75.0% on SSv2. MaskFeat further generalizes to image input, which can be interpreted as a video with a single frame and obtains competitive results on ImageNet.

Code Repositories

yyk-wew/semanticmim
pytorch
Mentioned in GitHub
facebookresearch/SlowFast
Official
pytorch
Mentioned in GitHub
Westlake-AI/openmixup
pytorch
Mentioned in GitHub
mx-mark/dmjd
pytorch
Mentioned in GitHub
mx-mark/videotransformer-pytorch
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
action-classification-on-kinetics-400MaskFeat (K600, MViT-L)
Acc@1: 87.0
Acc@5: 97.4
action-classification-on-kinetics-400MaskFeat (no extra data, MViT-L)
Acc@1: 86.7
Acc@5: 97.3
action-classification-on-kinetics-600MaskFeat (no extra data, MViT-L)
Top-1 Accuracy: 88.3
Top-5 Accuracy: 98.0
action-classification-on-kinetics-700MaskFeat (no extra data, MViT-L)
Top-1 Accuracy: 80.4
Top-5 Accuracy: 95.7
action-recognition-in-videos-on-somethingMaskFeat (Kinetics600 pretrain, MViT-L)
GFLOPs: 2828*3
Parameters: 218
Top-1 Accuracy: 75.0
Top-5 Accuracy: 95.0
action-recognition-on-ava-v2-2MaskFeat (Kinetics-600 pretrain, MViT-L)
mAP: 39.8
self-supervised-image-classification-on-1MaskFeat (ViT-L)
Number of Params: 307M
Top 1 Accuracy: 85.7%

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp