Back to Search Start Over

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

Authors :
Yu, Lijun
Lezama, José
Gundavarapu, Nitesh B.
Versari, Luca
Sohn, Kihyuk
Minnen, David
Cheng, Yong
Birodkar, Vighnesh
Gupta, Agrim
Gu, Xiuye
Hauptmann, Alexander G.
Gong, Boqing
Yang, Ming-Hsuan
Essa, Irfan
Ross, David A.
Jiang, Lu
Publication Year :
2023

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.<br />Comment: ICLR 2024

Details

Database :
arXiv
Publication Type :
Report
Accession number :
edsarx.2310.05737
Document Type :
Working Paper