Autoregressive Image Generation without Vector Quantization

By Tianhong Li, Yonglong Tian, He Li, Mingyang Deng, Kaiming He. Neural Information Processing Systems, 2024

691 citations recorded by the source corpus.

Conventional wisdom holds that autoregressive models for image generation are typically accompanied by vector-quantized tokens. We observe that while a discrete-valued space can facilitate representing a categorical distribution, it is not a necessity for autoregressive modeling. In this work, we propose to model the per-token probability distribution using a diffusion procedure, which allows us to apply autoregressive models in a continuous-valued space. Rather than using categorical cross-entropy loss, we define a Diffusion Loss function to model the per-token probability. This approach eliminates the need for discrete-valued tokenizers. We evaluate its effectiveness across a wide range of cases, including standard autoregressive models and generalized masked autoregressive (MAR) variants. By removing vector quantization, our image generator achieves strong results while enjoying the speed advantage of sequence modeling. We hope this work will motivate the use of autoregressive generation in other continuous-valued domains and applications. Code is available at: https://github.com/LTH14/mar.

Subfields: Computer Vision and Pattern Recognition, Media Technology

View Paper: https://www.semanticscholar.org/paper/e46a79cf7c2c9192639f9e1d12280650e06b9d56

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Ming-Hsuan Yang
  3. Xinggang Wang
  4. Xiaogang Wang
  5. Dacheng Tao
  6. Jitendra Malik
  7. Trevor Darrell
  8. Andrew Zisserman

In Media Technology

  1. Richard Zhang
  2. William T. Freeman
  3. Eli Shechtman
  4. Frédo Durand
  5. Xinggang Wang
  6. Ming-Hsuan Yang
  7. Tomer Michaeli
  8. Jun-Yan Zhu

Citations in the corpus