DC-AR: Efficient Masked Autoregressive Image Generation with Deep Compression Hybrid Tokenizer

Abstract

We introduce DC-AR, a novel masked autoregressive (AR) text-to-image generation framework that delivers superior image generation quality with exceptional computational efficiency. Due to the tokenizers’ limitations, prior masked AR models have lagged behind diffusion models in terms of quality or efficiency. We overcome this limitation by introducing DC-HT— a deep compression hybrid tokenizer for AR models that achieves a 32x spatial compression ratio while maintaining high reconstruction fidelity and cross-resolution generalization ability. Building upon DC-HT, we extend MaskGIT and create a new hybrid masked autoregressive image generation framework that first produces the structural elements through discrete tokens and then applies refinements via residual tokens. DC-AR achieves state-of-the-art results with a gFID of 5.49 on MJHQ-30K and an overall score of 0.69 on GenEval, while offering 1.5-7.9x higher throughput and 2.0-3.5x lower latency compared to prior leading diffusion and masked autoregressive models. We will release the code and pre-trained models upon publication.

Publication
International Conference on Computer Vision
Han Cai
Han Cai
Senior Research Scientist

Senior Research Scientist at NVIDIA Research.

Enze Xie
Enze Xie
Senior Research Scientist

Senior Research Scientist at NVIDIA Research.

Yao (Jason) Lu
Yao (Jason) Lu
Senior Research Scientist

Senior Research Scientist at NVIDIA Research.

Song Han
Song Han
Associate Professor

Song Han is an associate professor at MIT EECS.