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

Lijun Yu, José Lezama, Nitesh B. Gundavarapu, Luca Versari, Kihyuk Sohn +11 more
10/9/2023
cs.CVcs.AIcs.MM

No official code found. Best implementation: kyegomez/MAGVIT2 (15 stars, MIT, last commit 2024-11-11).

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.

View on arXivView PDF

Code Implementations(1)

kyegomez/MAGVIT2Community66%
MIT

Open source community's implementation of the model from "LANGUAGE MODEL BEATS DIFFUSION — TOKENIZER IS KEY TO VISUAL GENERATION"

151PythonOct 11, 20231 years agoMIT

Similar Papers

Cite this paper

@article{yu2023language,
  title  = {Language Model Beats Diffusion -- Tokenizer is Key to Visual Generation},
  author = {Lijun Yu and José Lezama and Nitesh B. Gundavarapu and Luca Versari and Kihyuk Sohn and David Minnen and Yong Cheng and Vighnesh Birodkar and Agrim Gupta and Xiuye Gu and Alexander G. Hauptmann and Boqing Gong and Ming-Hsuan Yang and Irfan Essa and David A. Ross and Lu Jiang},
  year   = {2023},
  eprint = {2310.05737},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2310.05737v3}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2310.05737.svg)](https://codeofpaper.com/paper/2310.05737)

Discussion