atlas

Research map / Image generation / Autoregressive and latent generation

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

NeurIPS 2024 · Image generation › Autoregressive and latent generation · approach: Next-scale and token autoregressive images · 111 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes Visual AutoRegressive (VAR) modeling, which replaces raster-scan next-token prediction with coarse-to-fine next-scale prediction over multi-scale VQ token maps, letting GPT-style transformers surpass DiT on ImageNet generation and show power-law scaling.

Problem

Autoregressive image generation with raster-scan next-token prediction lags diffusion models in quality, is computationally expensive, and breaks the unidirectional dependency assumption on 2D image tokens.

Approach

Define the autoregressive unit as an entire token map: encode an image into multi-scale token maps and predict each higher-resolution map conditioned on all coarser ones, with a block-wise causal attention mask in training and a multi-scale residual quantizer with shared codebook.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
replace AR next-token prediction with VAR next-scale prediction (ablation)AR transformer · ImageNet 256×256FID: 18.65 -> 5.22Helped
replace AR next-token prediction with VAR next-scale prediction (ablation)AR transformer · ImageNet 256×256inference cost relative to baseline: 1 -> 0.013Helped
add AdaLN (ablation)VAR-d16 · ImageNet 256×256FID: 5.22 -> 4.95Helped
add Top-k sampling (k=600) (ablation)VAR-d16 + AdaLN · ImageNet 256×256FID: 4.95 -> 4.64Helped
add CFG with ratio 2.0 (ablation)VAR-d16 + AdaLN + Top-k · ImageNet 256×256FID: 4.64 -> 3.60Helped
normalize q and k to unit vectors before attention (ablation)VAR-d16 + AdaLN + Top-k + CFG · ImageNet 256×256FID: 3.60 -> 3.30Helped
scale model up to 2.0B parameters (ablation)VAR · ImageNet 256×256FID: 3.30 -> 1.73Helped
use VAR instead of the AR baselineImage generation models · ImageNet 256×256FID: 2.10 -> 1.73Helped
use VAR instead of DiT-XL/2Image generation models · ImageNet 256×256IS: 278.2 -> 350.2Helped
use VAR instead of DiT-XL/2Image generation models · ImageNet 512×512FID: 3.04 -> 2.63Helped
increase VAR transformer sizeVAR · ImageNet validation settest cross-entropy lossHelped
increase VAR model size and training computeVAR · ImageNet 256×256 generated samplesvisual fidelity and soundnessHelped

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet OpenImages

Benchmarks: ImageNet 256x256 class-conditional generation ImageNet 512x512 class-conditional generation

Metrics: FID IS precision recall wall-clock inference time test cross-entropy loss token error rate

Baselines: VQGAN DiT-XL/2 L-DiT-3B L-DiT-7B MaskGIT ADM BigGAN StyleGAN-XL RQTransformer ViT-VQGAN

Builds on: VQGAN GPT-2 DiT MaskGIT

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy