视觉自回归建模:通过下一尺度预测实现可扩展的图像生成

Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Prediction

田柯宇 Keyu Tian · ByteDance Seed · 2024-04-03 · arXiv:2404.02905 ↗ · 被引 1060

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

我们提出了视觉自回归建模(VAR),这是一种新的生成范式,它将图像上的自回归学习重新定义为从粗到细的“下一尺度预测”或“下一分辨率预测”,与标准的逐光栅扫描“下一标记预测”不同。这种简单直观的方法使得自回归变换器能够快速学习视觉分布并具有良好的泛化能力:VAR 首次使得类似 GPT 的自回归模型在图像生成上超越了扩散变换器。在 ImageNet 256x256 基准测试中,VAR 显著改进了自回归基线,将 FID 从 18.65 降至 1.73,IS 从 80.4 提升至 350.2,推理速度提升约 20 倍。实验还验证了 VAR 在图像质量、推理速度、数据效率和可扩展性等多个维度上优于扩散变换器(DiT)。扩展 VAR 模型展现出与 LLM 中观察到的类似的清晰幂律缩放定律,线性相关系数接近-0.998,这是有力的证据。VAR 还在下游任务(如图像修复、外推和编辑)中展示了零样本泛化能力。这些结果表明 VAR 初步模拟了 LLM 的两个重要特性:缩放定律和零样本任务泛化。我们已发布所有模型和代码,以促进 AR/VAR 模型在视觉生成和统一学习中的探索。

We present Visual AutoRegressive modeling (VAR), a new generation paradigm that redefines the autoregressive learning on images as coarse-to-fine "next-scale prediction" or "next-resolution prediction", diverging from the standard raster-scan "next-token prediction". This simple, intuitive methodology allows autoregressive (AR) transformers to learn visual distributions fast and generalize well: VAR, for the first time, makes GPT-like AR models surpass diffusion transformers in image generation. On ImageNet 256x256 benchmark, VAR significantly improve AR baseline by improving Frechet inception distance (FID) from 18.65 to 1.73, inception score (IS) from 80.4 to 350.2, with around 20x faster inference speed. It is also empirically verified that VAR outperforms the Diffusion Transformer (DiT) in multiple dimensions including image quality, inference speed, data efficiency, and scalability. Scaling up VAR models exhibits clear power-law scaling laws similar to those observed in LLMs, with linear correlation coefficients near -0.998 as solid evidence. VAR further showcases zero-shot generalization ability in downstream tasks including image in-painting, out-painting, and editing. These results suggest VAR has initially emulated the two important properties of LLMs: Scaling Laws and zero-shot task generalization. We have released all models and codes to promote the exploration of AR/VAR models for visual generation and unified learning.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 16)

阅读逐段中英对照全文 →