用连续令牌和随机生成顺序,让文生图模型实现更好视觉质量
Fluid: Scaling Autoregressive Text-to-image Generative Models with Continuous Tokens
- 采用连续令牌与随机生成顺序的自回归架构
- 在MS-COCO上达6.16零样本FID,GenEval得分0.69
- 为视觉生成模型的可扩展性提供新思路
在视觉领域,自回归模型的扩展效果不如大语言模型显著。本文研究文本到图像生成中的这一扩展难题,重点关注两个关键因素:模型使用离散或连续令牌,以及生成顺序是随机还是固定栅格顺序,基于BERT或GPT类Transformer架构。实验表明,尽管所有模型在验证损失上均有效扩展,但评估性能(以FID、GenEval分数和视觉质量衡量)呈现不同趋势。基于连续令牌的模型显著优于离散令牌模型。此外,生成顺序和注意力机制对GenEval分数影响显著:随机顺序模型明显优于栅格顺序模型。受此启发,我们训练了Fluid——一个基于连续令牌的随机顺序自回归模型。Fluid 10.5B模型在MS-COCO 30K上达到6.16的零样本FID新纪录,在GenEval基准上取得0.69的整体分数。希望本研究结果能推动未来工作进一步缩小视觉与语言模型间的扩展差距。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。