Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

HART: Efficient Visual Generation with Hybrid Autoregressive Transformer

About

We introduce Hybrid Autoregressive Transformer (HART), an autoregressive (AR) visual generation model capable of directly generating 1024x1024 images, rivaling diffusion models in image generation quality. Existing AR models face limitations due to the poor image reconstruction quality of their discrete tokenizers and the prohibitive training costs associated with generating 1024px images. To address these challenges, we present the hybrid tokenizer, which decomposes the continuous latents from the autoencoder into two components: discrete tokens representing the big picture and continuous tokens representing the residual components that cannot be represented by the discrete tokens. The discrete component is modeled by a scalable-resolution discrete AR model, while the continuous component is learned with a lightweight residual diffusion module with only 37M parameters. Compared with the discrete-only VAR tokenizer, our hybrid approach improves reconstruction FID from 2.11 to 0.30 on MJHQ-30K, leading to a 31% generation FID improvement from 7.85 to 5.38. HART also outperforms state-of-the-art diffusion models in both FID and CLIP score, with 4.5-7.7x higher throughput and 6.9-13.4x lower MACs. Our code is open sourced at https://github.com/mit-han-lab/hart.

Haotian Tang, Yecheng Wu, Shang Yang, Enze Xie, Junsong Chen, Junyu Chen, Zhuoyang Zhang, Han Cai, Yao Lu, Song Han• 2024

Related benchmarks

TaskDatasetResultRank
Text-to-Image GenerationGenEval
Overall Score56.59
704
Text-to-Image GenerationDPG-Bench
Overall Score80.89
451
Text-to-Image GenerationGenEval
GenEval Score51
442
Text-to-Image GenerationGenEval
Overall Score0.56
277
Text-to-Image GenerationDPG
Overall Score80.89
256
Text-to-Image GenerationMJHQ-30K
Overall FID11
239
Text-to-Image GenerationGenEval
Overall Score51
218
Text-to-Image GenerationHPS v2.1
Overall Score29.07
96
Text-to-Image GenerationImageReward
ImageReward Score0.661
69
Image GenerationGenEval
Overall Score50.9
69
Showing 10 of 14 rows

Other info

Follow for update