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

Fluid: Scaling Autoregressive Text-to-image Generative Models with Continuous Tokens

About

Scaling up autoregressive models in vision has not proven as beneficial as in large language models. In this work, we investigate this scaling problem in the context of text-to-image generation, focusing on two critical factors: whether models use discrete or continuous tokens, and whether tokens are generated in a random or fixed raster order using BERT- or GPT-like transformer architectures. Our empirical results show that, while all models scale effectively in terms of validation loss, their evaluation performance -- measured by FID, GenEval score, and visual quality -- follows different trends. Models based on continuous tokens achieve significantly better visual quality than those using discrete tokens. Furthermore, the generation order and attention mechanisms significantly affect the GenEval score: random-order models achieve notably better GenEval scores compared to raster-order models. Inspired by these findings, we train Fluid, a random-order autoregressive model on continuous tokens. Fluid 10.5B model achieves a new state-of-the-art zero-shot FID of 6.16 on MS-COCO 30K, and 0.69 overall score on the GenEval benchmark. We hope our findings and results will encourage future efforts to further bridge the scaling gap between vision and language models.

Lijie Fan, Tianhong Li, Siyang Qin, Yuanzhen Li, Chen Sun, Michael Rubinstein, Deqing Sun, Kaiming He, Yonglong Tian• 2024

Related benchmarks

TaskDatasetResultRank
Semantic segmentationADE20K (val)
mIoU22.1
3069
Image ClassificationImageNet-1K
Top-1 Acc80.3
1239
Text-to-Image GenerationGenEval
Overall Score69
704
Image ClassificationImageNet A
Top-1 Acc20.2
698
Text-to-Image GenerationGenEval
Overall Score70
517
Image ClassificationRESISC45
Accuracy74.4
472
Depth EstimationNYU v2 (test)--
435
Image ClassificationObjectNet--
251
Image ClassificationImageNet-R
Accuracy44.2
217
Text-to-Image GenerationMS-COCO
FID9.62
145
Showing 10 of 29 rows

Other info

Follow for update