HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Unmasked Teacher: Towards Training-Efficient Video Foundation Models

Kunchang Li; Yali Wang; Yizhuo Li; Yi Wang; Yinan He; Limin Wang; Yu Qiao

Unmasked Teacher: Towards Training-Efficient Video Foundation Models

Abstract

Video Foundation Models (VFMs) have received limited exploration due to high computational costs and data scarcity. Previous VFMs rely on Image Foundation Models (IFMs), which face challenges in transferring to the video domain. Although VideoMAE has trained a robust ViT from limited data, its low-level reconstruction poses convergence difficulties and conflicts with high-level cross-modal alignment. This paper proposes a training-efficient method for temporal-sensitive VFMs that integrates the benefits of existing methods. To increase data efficiency, we mask out most of the low-semantics video tokens, but selectively align the unmasked tokens with IFM, which serves as the UnMasked Teacher (UMT). By providing semantic guidance, our method enables faster convergence and multimodal friendliness. With a progressive pre-training framework, our model can handle various tasks including scene-related, temporal-related, and complex video-language understanding. Using only public sources for pre-training in 6 days on 32 A100 GPUs, our scratch-built ViT-L/16 achieves state-of-the-art performances on various video tasks. The code and models will be released at https://github.com/OpenGVLab/unmasked_teacher.

Code Repositories

opengvlab/unmasked_teacher
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
action-classification-on-kinetics-400UMT-L (ViT-L/16)
Acc@1: 90.6
Acc@5: 98.7
action-classification-on-kinetics-400Unmasked Teacher (ViT-L)
Acc@1: 90.6
Acc@5: 98.7
FLOPs (G) x views: 1434×3×4
Parameters (M): 304
action-classification-on-kinetics-600UMT-L (ViT-L/16)
Top-1 Accuracy: 90.5
Top-5 Accuracy: 98.8
action-classification-on-kinetics-700UMT-L (ViT-L/16)
Top-1 Accuracy: 83.6
Top-5 Accuracy: 96.7
action-classification-on-moments-in-timeUMT-L (ViT-L/16)
Top 1 Accuracy: 48.7
Top 5 Accuracy: 78.2
action-recognition-on-ava-v2-2UMT-L (ViT-L/16)
mAP: 39.8
video-question-answering-on-activitynet-qaUMT-L (ViT-L/16)
Accuracy: 47.9
video-retrieval-on-activitynetUMT-L (ViT-L/16)
text-to-video R@1: 66.8
text-to-video R@10: 94.9
text-to-video R@5: 89.1
video-to-text R@1: 64.4
video-to-text R@10: 94.8
video-to-text R@5: 89.1
video-retrieval-on-didemoUMT-L (ViT-L/16)
text-to-video R@1: 70.4
text-to-video R@10: 93.5
text-to-video R@5: 90.1
video-to-text R@1: 65.7
video-to-text R@10: 93.3
video-to-text R@5: 89.6
video-retrieval-on-lsmdcUMT-L (ViT-L/16)
text-to-video R@1: 43.0
text-to-video R@10: 73.0
text-to-video R@5: 65.5
video-to-text R@1: 41.4
video-to-text R@10: 71.5
video-to-text R@5: 64.3
video-retrieval-on-msr-vttUMT-L (ViT-L/16)
text-to-video R@1: 58.8
text-to-video R@10: 87.1
text-to-video R@5: 81.0
video-to-text R@1: 58.6
video-to-text R@10: 86.5
video-to-text R@5: 81.6
video-retrieval-on-ssv2-label-retrievalUMT-L (ViT-L/16)
text-to-video R@1: 73.3
text-to-video R@10: 96.6
text-to-video R@5: 92.7
video-retrieval-on-ssv2-template-retrievalUMT-L (ViT-L/16)
text-to-video R@1: 90.8
text-to-video R@10: 100.0
text-to-video R@5: 100.0
video-retrieval-on-vatexUnmasked Teacher
text-to-video R@1: 72
text-to-video R@10: 97.8
text-to-video R@5: 95.1
video-to-text R@1: 86.0
video-to-text R@10: 99.6
visual-question-answering-on-msrvtt-qa-1UMT-L (ViT-L/16)
Accuracy: 0.471
visual-question-answering-on-msvd-qa-1UMT-L (ViT-L/16)
Accuracy: 0.552
zero-shot-video-retrieval-on-activitynetUMT-L (ViT-L/16)
text-to-video R@1: 42.8
text-to-video R@10: 79.8
text-to-video R@5: 69.6
video-to-text R@1: 40.7
video-to-text R@10: 78.6
video-to-text R@5: 67.6
zero-shot-video-retrieval-on-didemoUMT-L (ViT-L/16)
text-to-video R@1: 48.6
text-to-video R@10: 79.0
text-to-video R@5: 72.9
video-to-text R@1: 49.9
video-to-text R@10: 81.4
video-to-text R@5: 74.8
zero-shot-video-retrieval-on-lsmdcUMT-L (ViT-L/16)
text-to-video R@1: 25.2
text-to-video R@10: 50.5
text-to-video R@5: 43.0
video-to-text R@1: 23.2
video-to-text R@10: 44.2
video-to-text R@5: 37.7
zero-shot-video-retrieval-on-msr-vttUMT-L (ViT-L/16)
text-to-video R@1: 42.6
text-to-video R@10: 73.1
text-to-video R@5: 64.4
video-to-text R@1: 38.6
video-to-text R@10: 69.6
video-to-text R@5: 59.8
zero-shot-video-retrieval-on-msvdUMT-L (ViT-L/16)
text-to-video R@1: 49.0
text-to-video R@10: 84.7
text-to-video R@5: 76.9
video-to-text R@1: 74.5
video-to-text R@10: 92.8
video-to-text R@5: 89.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
Unmasked Teacher: Towards Training-Efficient Video Foundation Models | Papers | HyperAI