Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Autoregressive Image Generation without Vector Quantization

About

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.

Tianhong Li, Yonglong Tian, He Li, Mingyang Deng, Kaiming He• 2024

Related benchmarks

TaskDatasetResultRank
Semantic segmentationADE20K (val)
mIoU23.4
3089
Image ClassificationImageNet-1K
Top-1 Acc80.4
1239
Class-conditional Image GenerationImageNet 256x256
Inception Score (IS)304.1
1021
Image ClassificationImageNet A
Top-1 Acc20.6
723
Image GenerationImageNet 256x256
IS303.7
606
Image ClassificationRESISC45
Accuracy73.8
539
Class-conditional Image GenerationImageNet 256x256 (val)
Inception Score (IS)303.7
535
Image GenerationImageNet 256x256 (val)
FID1.55
461
Text-to-Image GenerationGenEval
GenEval Score75.75
459
Depth EstimationNYU v2 (test)--
438
Showing 10 of 80 rows
...

Other info

Code

Follow for update