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

arXiv:2510.18457v3 Announce Type: replace Abstract: 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 10imes 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.

Related

• ClusterMark: Towards Robust Watermarking for Autoregressive Image Generators with Visual Token Clustering
• Sampling-Aware Quantization for Diffusion Models
• Adversarial Concept Distillation for One-Step Diffusion Personalization
• Diagnosing and Improving Diffusion Models by Estimating the Optimal Loss Value
• MixFlow: Mixed Source Distributions Improve Rectified Flows

Source: arXiv cs.CV | 2026-04-24