提出新离散分词器,让图像自回归模型性能超越扩散模型。
Stabilize the Latent Space for Image Autoregressive Modeling: A Unified Perspective
- 用自监督学习特征做K-Means聚类,构建稳定离散图像分词器。
- 基于该分词器的自回归模型首次在图像生成上超过扩散模型。
- 适合研究图像生成、GPT风格模型及潜在空间优化的学者。
基于潜在空间的图像生成模型(如LDM和MIM)已取得显著成果,通常通过VQGAN或VAE等重建自编码器将像素映射到紧凑的潜在空间,并在该空间学习数据分布。然而,这一做法是否最优仍存疑问。本文观察到:尽管共享同一潜在空间,自回归模型在图像生成上明显落后于LDM和MIM,与自然语言处理中GPT的主导地位形成鲜明对比。为此,我们提出统一视角,强调潜在空间稳定性在图像生成中的关键作用。进一步,我们设计了一种简单有效的离散图像分词器(DiGIT),通过在自监督学习特征上应用K-Means实现。实验表明,使用DiGIT的自回归模型在图像理解与生成上均表现优异,其基于下一个词预测的原则天然适配GPT类模型,却对其他生成模型具有挑战性。令人振奋的是,首次实现一个类GPT的图像自回归模型在生成质量上超越了LDM,且在扩大模型规模时表现出类似GPT的显著提升。结果凸显了优化潜在空间与离散分词融合在推进图像生成模型能力上的潜力。代码已开源。
原文摘要 · Abstract (English)
Latent-based image generative models, such as Latent Diffusion Models (LDMs) and Mask Image Models (MIMs), have achieved notable success in image generation tasks. These models typically leverage reconstructive autoencoders like VQGAN or VAE to encode pixels into a more compact latent space and learn the data distribution in the latent space instead of directly from pixels. However, this practice raises a pertinent question: Is it truly the optimal choice? In response, we begin with an intriguing observation: despite sharing the same latent space, autoregressive models significantly lag behind LDMs and MIMs in image generation. This finding contrasts sharply with the field of NLP, where the autoregressive model GPT has established a commanding presence. To address this discrepancy, we introduce a unified perspective on the relationship between latent space and generative models, emphasizing the stability of latent space in image generative modeling. Furthermore, we propose a simple but effective discrete image tokenizer to stabilize the latent space for image generative modeling by applying K-Means on the latent features of self-supervised learning models. Experimental results show that image autoregressive modeling with our tokenizer (DiGIT) benefits both image understanding and image generation with the next token prediction principle, which is inherently straightforward for GPT models but challenging for other generative models. Remarkably, for the first time, a GPT-style autoregressive model for images outperforms LDMs, which also exhibits substantial improvement akin to GPT when scaling up model size. Our findings underscore the potential of an optimized latent space and the integration of discrete tokenization in advancing the capabilities of image generative models. The code is available at \url{https://github.com/DAMO-NLP-SG/DiGIT}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。