HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Language Model Beats Diffusion -- Tokenizer is Key to Visual Generation

Language Model Beats Diffusion -- Tokenizer is Key to Visual Generation

Abstract

While Large Language Models (LLMs) are the dominant models for generative tasks in language, they do not perform as well as diffusion models on image and video generation. To effectively use LLMs for visual generation, one crucial component is the visual tokenizer that maps pixel-space inputs to discrete tokens appropriate for LLM learning. In this paper, we introduce MAGVIT-v2, a video tokenizer designed to generate concise and expressive tokens for both videos and images using a common token vocabulary. Equipped with this new tokenizer, we show that LLMs outperform diffusion models on standard image and video generation benchmarks including ImageNet and Kinetics. In addition, we demonstrate that our tokenizer surpasses the previously top-performing video tokenizer on two more tasks: (1) video compression comparable to the next-generation video codec (VCC) according to human evaluations, and (2) learning effective representations for action recognition tasks.

Code Repositories

jy0205/Pyramid-Flow
pytorch
Mentioned in GitHub
lucidrains/magvit2-pytorch
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
image-generation-on-imagenet-256x256MAGVIT-v2 (w/o guidance)
FID: 3.65
image-generation-on-imagenet-256x256MAGVIT-v2
FID: 1.78
image-generation-on-imagenet-512x512MAGVIT-v2 (w/o guidance)
FID: 3.07
Inception score: 213.1
image-generation-on-imagenet-512x512MAGVIT-v2
FID: 1.91
Inception score: 324.3
video-generation-on-kinetics-600-12-framesMAGVIT-v2
FVD: 4.3±0.1
video-generation-on-ucf-101MAGVIT-v2
FVD16: 58±3
video-generation-on-ucf-101MAGVIT-v2 (AR)
FVD16: 109
video-prediction-on-kinetics-600-12-framesMAGVIT-v2
FVD: 4.3±0.1

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
Language Model Beats Diffusion -- Tokenizer is Key to Visual Generation | Papers | HyperAI