HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

MAGVIT: Masked Generative Video Transformer

MAGVIT: Masked Generative Video Transformer

Abstract

We introduce the MAsked Generative VIdeo Transformer, MAGVIT, to tackle various video synthesis tasks with a single model. We introduce a 3D tokenizer to quantize a video into spatial-temporal visual tokens and propose an embedding method for masked video token modeling to facilitate multi-task learning. We conduct extensive experiments to demonstrate the quality, efficiency, and flexibility of MAGVIT. Our experiments show that (i) MAGVIT performs favorably against state-of-the-art approaches and establishes the best-published FVD on three video generation benchmarks, including the challenging Kinetics-600. (ii) MAGVIT outperforms existing methods in inference time by two orders of magnitude against diffusion models and by 60x against autoregressive models. (iii) A single MAGVIT model supports ten diverse generation tasks and generalizes across videos from different visual domains. The source code and trained models will be released to the public at https://magvit.cs.cmu.edu.

Code Repositories

google-research/magvit
Official
jax
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
text-to-video-generation-on-somethingMAGVIT
FVD: 79.1
video-generation-on-bair-robot-pushingMAGVIT
Cond: 1
FVD score: 62
Pred: 15
Train: 15
video-generation-on-kinetics-600-12-framesMAGVIT
FVD: 9.9
video-generation-on-ucf-101MAGVIT (AR)
FVD16: 265
video-generation-on-ucf-101MAGVIT (-L-CG, 128x128, class-conditional)
FVD16: 76±2
Inception Score: 89.27±0.15
video-generation-on-ucf-101MAGVIT (-B-CG, 128x128, class-conditional)
FVD16: 159±2
Inception Score: 83.55±0.14
video-prediction-on-bair-robot-pushing-1MAGVIT (-B-FP)
FVD: 76±0.1
video-prediction-on-bair-robot-pushing-1MAGVIT (-L-FP)
FVD: 62±0.1
video-prediction-on-kinetics-600-12-framesMAGVIT (-L-FP)
Cond: 5
FVD: 9.9±0.3
Pred: 11
video-prediction-on-kinetics-600-12-framesMAGVIT (-B-FP)
Cond: 5
FVD: 24.5±0.9
Pred: 11
video-prediction-on-something-something-v2MAGVIT
FVD: 28.5

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