arXiv:2510.18457cs.CVcs.LG2025-10中稿 · CVPR被引 17

用冻结的视觉大模型做扩散模型的分词器,训练速度提升10倍。

VFM-VAE: Vision Foundation Models Can Be Good Tokenizers for Latent Diffusion Models

  • 直接用冻结视觉大模型作分词器,避免微调导致的表征退化。
  • 80轮训练即达gFID 2.22,640轮降至1.62,速度提升10倍。
  • 适合想快速训练高质量图像生成模型的研究者和开发者。

潜空间扩散模型(LDMs)的性能高度依赖于其视觉分词器的质量。尽管近期工作尝试通过知识蒸馏将视觉基础模型(VFMs)融入分词器训练,但实验发现该方法不可避免地削弱了原始VFM学习到的表征鲁棒性。本文提出一种更直接的方法:利用冻结的VFM作为分词器,命名为VFM-变分自编码器(VFM-VAE)。为充分发挥冻结VFM的潜力,设计了新解码器,从语义丰富的VFM表征中重建真实图像。通过系统研究不同分词器对扩散训练中表征学习的影响,实现了分词器与扩散模型间的双向协同优化。该分词器设计与训练策略显著提升性能与效率:仅需80轮训练即达到无条件引导(w/o CFG)下的gFID 2.22(较先前方法提速10倍);持续训练至640轮时,gFID进一步降至1.62。结果充分证明了视觉基础模型作为视觉分词器在加速LDM训练方面的巨大潜力。

原文摘要 · Abstract (English)

The performance of Latent Diffusion Models (LDMs) is critically dependent on the quality of their visual tokenizers. While recent works have explored incorporating Vision Foundation Models (VFMs) into the tokenizers training via distillation, we empirically find this approach inevitably weakens the robustness of learnt representation from original VFM. In this paper, we bypass the distillation by proposing a more direct approach by leveraging the frozen VFM for the LDMs tokenizer, named VFM Variational Autoencoder (VFM-VAE).To fully exploit the potential to leverage frozen VFM for the LDMs tokenizer, we design a new decoder to reconstruct realistic images from the semantic-rich representation of VFM. With the proposed VFM-VAE, we conduct a systematic study on how the representation from different tokenizers impact the representation learning process throughout diffusion training, enabling synergistic benefits of dual-side alignment on both tokenizers and diffusion models. Our effort in tokenizer design and training strategy lead to superior performance and efficiency: our system reaches a gFID (w/o CFG) of 2.22 in merely 80 epochs (a 10$\times$ speedup over prior tokenizers). With continued training to 640 epochs, it further attains a gFID (w/o CFG) of 1.62. These results offer solid evidence for the substantial potential of VFMs to serve as visual tokenizers to accelerate the LDM training progress.

扩散模型视觉大模型分词器

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