arXiv:2410.16257cs.CV2024-10ICML被引 6

探索语言模型生成图像的设计空间,发现大模型更擅长捕捉全局信息。

Elucidating the design space of language models for image generation

  • 用自回归建模处理图像令牌,突破随机性带来的训练难题。
  • 大模型显著提升全局上下文理解能力,解释了规模扩展的性能优势。
  • 提出ELM模型,在ImageNet上达到顶尖水平,适合视觉生成研究者参考。

自回归语言模型在文本生成中的成功激发了计算机视觉领域将大型语言模型(LLMs)用于图像生成的兴趣。然而,由于文本与图像模态的本质差异,语言模型在图像生成中的设计空间仍待深入探索。我们观察到图像令牌比文本令牌具有更高的随机性,这给基于令牌预测的训练带来了挑战。尽管如此,自回归模型仍能有效学习模式,展现出从看似次优优化问题中提取规律的潜力。分析还表明,所有模型均能掌握图像生成中的局部信息重要性,但小模型难以捕捉全局上下文;而大模型则表现出更强的全局建模能力,有助于解释模型规模扩展带来的性能提升。通过大量对比实验,我们系统阐明了语言模型在视觉生成中的设计空间,包括分词器选择、模型架构、可扩展性、词汇设计及采样策略。本工作首次分析了语言模型在视觉生成中的优化行为,有望为将语言模型应用于其他领域提供有效设计启发。最终提出的语言模型ELM在ImageNet 256*256基准上达到当前最优性能。代码已公开于https://github.com/Pepperlll/LMforImageGeneration.git。

原文摘要 · Abstract (English)

The success of autoregressive (AR) language models in text generation has inspired the computer vision community to adopt Large Language Models (LLMs) for image generation. However, considering the essential differences between text and image modalities, the design space of language models for image generation remains underexplored. We observe that image tokens exhibit greater randomness compared to text tokens, which presents challenges when training with token prediction. Nevertheless, AR models demonstrate their potential by effectively learning patterns even from a seemingly suboptimal optimization problem. Our analysis also reveals that while all models successfully grasp the importance of local information in image generation, smaller models struggle to capture the global context. In contrast, larger models showcase improved capabilities in this area, helping to explain the performance gains achieved when scaling up model size. We further elucidate the design space of language models for vision generation, including tokenizer choice, model choice, model scalability, vocabulary design, and sampling strategy through extensive comparative experiments. Our work is the first to analyze the optimization behavior of language models in vision generation, and we believe it can inspire more effective designs when applying LMs to other domains. Finally, our elucidated language model for image generation, termed as ELM, achieves state-of-the-art performance on the ImageNet 256*256 benchmark. The code is available at https://github.com/Pepperlll/LMforImageGeneration.git.

图像生成语言模型视觉建模模型设计

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。