Autoregressive Image Generation without Vector Quantization

GenAI Diffusion ML Generative Models DM
2024年06月17日
传统观点认为,图像生成的自回归模型通常需要使用向量量化标记。我们发现,虽然离散值空间可以方便地表示分类分布,但它并不是自回归建模的必要条件。在这项工作中,我们提出使用扩散过程对每个标记的概率分布进行建模,这使我们能够在连续值空间中应用自回归模型。我们定义了扩散损失函数来建模每个标记的概率分布,而不是使用分类交叉熵损失。这种方法消除了需要使用离散值标记的必要性。我们在包括标准自回归模型和广义掩码自回归(MAR)变体在内的广泛案例中评估了其有效性。通过消除向量量化,我们的图像生成器在享受序列建模速度优势的同时取得了强大的结果。我们希望这项工作能激发在其他连续值领域和应用中使用自回归生成的动力。
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.