arXiv:2503.11056cs.CV2025-03ICCV被引 43

无需对抗训练,用扩散模型实现图像令牌化新纪录

Flow to the Mode: Mode-Seeking Diffusion Autoencoders for State-of-the-Art Image Tokenization

  • 分两阶段训练:先匹配数据模式,再主动寻找模式
  • 在ImageNet-1K上多压缩率下刷新令牌化性能
  • 纯变压器架构,不依赖卷积或蒸馏其他模型

自VQGAN和潜在扩散模型兴起以来,主流图像生成系统普遍采用两阶段流程:先将视觉数据压缩为低维潜在空间,再学习生成模型。现有令牌化训练通常采用标准方案,通过均方误差、感知损失和对抗损失联合优化图像压缩与重建。尽管先前研究提出扩散自编码器以实现端到端感知导向的图像压缩,但尚未在ImageNet-1K重建任务中达到顶尖性能。本文提出FlowMo,一种基于变压器的扩散自编码器,在不使用卷积、对抗损失、二维空间对齐潜在码或从其他令牌化器蒸馏的情况下,实现了多个压缩率下的新性能纪录。核心洞见是将训练分为模式匹配预训练和模式搜索后训练两个阶段。我们还进行了大量分析,并探索了在FlowMo令牌化器之上构建生成模型的可行性。代码与模型将公开于http://kylesargent.github.io/flowmo。

原文摘要 · Abstract (English)

Since the advent of popular visual generation frameworks like VQGAN and latent diffusion models, state-of-the-art image generation systems have generally been two-stage systems that first tokenize or compress visual data into a lower-dimensional latent space before learning a generative model. Tokenizer training typically follows a standard recipe in which images are compressed and reconstructed subject to a combination of MSE, perceptual, and adversarial losses. Diffusion autoencoders have been proposed in prior work as a way to learn end-to-end perceptually-oriented image compression, but have not yet shown state-of-the-art performance on the competitive task of ImageNet-1K reconstruction. We propose FlowMo, a transformer-based diffusion autoencoder that achieves a new state-of-the-art for image tokenization at multiple compression rates without using convolutions, adversarial losses, spatially-aligned two-dimensional latent codes, or distilling from other tokenizers. Our key insight is that FlowMo training should be broken into a mode-matching pre-training stage and a mode-seeking post-training stage. In addition, we conduct extensive analyses and explore the training of generative models atop the FlowMo tokenizer. Our code and models will be available at http://kylesargent.github.io/flowmo .

扩散模型图像压缩令牌化自编码器

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