arXiv:2506.21022cs.CV2025-06被引 6

用128个二进制向量表示1024x1024图像,速度更快、效率更高。

Instella-T2I: Pushing the Limits of 1D Discrete Latent Space Image Generation

  • 将图像转为1D二进制向量序列,替代传统一热编码,更紧凑
  • 仅用128个离散令牌生成1024x1024图像,比标准VQ-VAE少32倍
  • 单卡可训练4096全局批量,200 GPU天完成训练,适合大规模部署

图像分词在降低高分辨率图像建模的计算负担方面至关重要,显著提升了图像与多模态理解及生成的效率。近期1D潜在空间进展通过去除2D网格结构,减少了所需标记数。本文进一步推进紧凑离散图像表示,提出1D二进制图像潜在空间。通过将每幅图像表示为二进制向量序列,而非传统一热码本标记,该方法在保持1D潜在空间紧凑性的同时保留了高分辨率细节。据我们所知,所提出的文本到图像模型是首个仅使用128个离散标记即可在扩散与自回归生成中实现竞争力表现的模型,适用于高达1024x1024的图像,相比标准VQ-VAEs减少多达32倍的标记数量。结合简单的模型架构,该方法在训练与推理速度上均有显著提升。文本到图像模型可在单个GPU节点(含8块AMD MI300X)上支持4096的全局批量,训练可在200 GPU天内完成。模型性能媲美现代图像生成模型,且无需内部私有训练数据或后训练优化,为传统分词方法提供了一种可扩展、高效的替代方案。

原文摘要 · Abstract (English)

Image tokenization plays a critical role in reducing the computational demands of modeling high-resolution images, significantly improving the efficiency of image and multimodal understanding and generation. Recent advances in 1D latent spaces have reduced the number of tokens required by eliminating the need for a 2D grid structure. In this paper, we further advance compact discrete image representation by introducing 1D binary image latents. By representing each image as a sequence of binary vectors, rather than using traditional one-hot codebook tokens, our approach preserves high-resolution details while maintaining the compactness of 1D latents. To the best of our knowledge, our text-to-image models are the first to achieve competitive performance in both diffusion and auto-regressive generation using just 128 discrete tokens for images up to 1024x1024, demonstrating up to a 32-fold reduction in token numbers compared to standard VQ-VAEs. The proposed 1D binary latent space, coupled with simple model architectures, achieves marked improvements in speed training and inference speed. Our text-to-image models allow for a global batch size of 4096 on a single GPU node with 8 AMD MI300X GPUs, and the training can be completed within 200 GPU days. Our models achieve competitive performance compared to modern image generation models without any in-house private training data or post-training refinements, offering a scalable and efficient alternative to conventional tokenization methods.

图像生成潜空间高效建模扩散模型

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