UniGen-AR: Unifying Visual Generationwith Auto-Regressive Modeling

1Carnegie Mellon University 2University of Illinois Urbana-Champaign 3Toyota Research Institute
One model · 15+ tasks · one set of weights

A single auto-regressive backbone for unified visual generation

Text-to-image synthesis, classical perception, restoration, and instruction-guided editing — all produced by one MLLM-conditioned VAR model through a shared prompting interface, with no task-specific heads.

Abstract

TL;DR One auto-regressive model handles 15+ visual generation tasks — text-to-image, editing, restoration, and perception — at up to 5× lower latency and >10× fewer FLOPs than diffusion, trained only on public data.

Modern vision pipelines stay fragmented — generation, editing, restoration, and perception each handled by separate models. We study Unified Visual Generation (UVG), where one model produces diverse image-valued outputs through a single multimodal interface.

UniGen-AR pairs a general-purpose multimodal language model encoder with an efficient next-scale visual auto-regressive (VAR) decoder. The MLLM turns free-form instructions and reference images into a unified conditioning sequence that guides the VAR decoder across 15+ tasks in four families.

The result: up to 5× lower inference latency than diffusion baselines while matching or improving quality. Our analysis identifies VQ-VAE tokenizer design — codebook size and hierarchy — as the critical factor for scaling VAR in UVG.

Method

MLLM conditioning meets next-scale VAR decoding

A Qwen2.5-VL encoder reads the instruction and reference image into text embeddings; a learnable [SoS] token and cross-attention steer an Infinity-style VAR decoder that predicts discrete visual tokens, decoded to pixels by a VQ-VAE.

Latency–quality trade-off
≈5× faster than diffusion

Under matched conditioning and hardware, swapping a diffusion decoder for the VAR decoder cuts inference latency dramatically — while improving GenEval quality.

Qwen-2.5-VL conditioning · 512×512 · single H100

Diffusion Qwen + SD35.23 s/img
VAR Qwen + Infinity (Ours)1.05 s/img
Lower is better · GenEval overall: 0.52 → 0.64
>10× fewer FLOPs per generated image than the diffusion counterpart ↓ lower is better · TFLOPs / image
512 × 512
Qwen + SD3
16 T
Qwen + SANA
14 T
UniGen-AR (Ours)
1.9 T 7–10× ↓
1024 × 1024
Qwen + SD3
79 T
UniGen-AR (Ours)
4.1 T 19× ↓
Head-to-head

Compared with specialized & unified models

Input · UniGen-AR (Ours) · task-specialized and unified baselines, across perception, restoration, and editing.

BibTeX

Cite this work

@article{unigenar2026,
  title   = {UniGen-AR: Unifying Visual Generation with Auto-Regressive Modeling},
  author  = {Bao, Zhipeng and Zhu, Zhen and Kumari, Nupur and Bagchi, Anurag
             and Wang, Yu-Xiong and Tokmakov, Pavel and Hebert, Martial},
  journal = {arXiv preprint arXiv:2607.24157},
  year    = {2026}
}