通过显式对齐潜在分布,提升图像生成质量与效率
Distribution Matching Variational AutoEncoder
- 用分布匹配约束让编码器潜空间贴合任意参考分布
- 在ImageNet上仅用64轮训练即达gFID 3.2,优于传统VAE
- 适合关注生成质量与潜空间设计的研究者
多数视觉生成模型在应用扩散或自回归建模前将图像压缩至潜空间。然而,现有方法如标准VAE和对齐基础模型的编码器隐含地限制潜空间分布,却未明确塑造其分布形态,导致最优分布类型不清晰。本文提出分布匹配变分自编码器(DMVAE),通过分布匹配约束显式将编码器的潜分布与任意参考分布对齐。该方法超越传统VAE的高斯先验,可对齐来自自监督学习特征、扩散噪声或其他先验分布的分布。借助DMVAE,我们系统研究了不同潜分布对建模效果的影响,发现自监督学习导出的分布能在重建保真度与建模效率间取得优异平衡,在ImageNet上仅用64个训练周期即达到gFID 3.2。结果表明,通过分布级对齐选择合适的潜分布结构,比依赖固定先验更关键,是缩小易建模潜空间与高保真图像合成之间差距的核心。代码已公开于https://github.com/sen-ye/dmvae。
原文摘要 · Abstract (English)
Most visual generative models compress images into a latent space before applying diffusion or autoregressive modelling. Yet, existing approaches such as VAEs and foundation model aligned encoders implicitly constrain the latent space without explicitly shaping its distribution, making it unclear which types of distributions are optimal for modeling. We introduce \textbf{Distribution-Matching VAE} (\textbf{DMVAE}), which explicitly aligns the encoder's latent distribution with an arbitrary reference distribution via a distribution matching constraint. This generalizes beyond the Gaussian prior of conventional VAEs, enabling alignment with distributions derived from self-supervised features, diffusion noise, or other prior distributions. With DMVAE, we can systematically investigate which latent distributions are more conducive to modeling, and we find that SSL-derived distributions provide an excellent balance between reconstruction fidelity and modeling efficiency, reaching gFID equals 3.2 on ImageNet with only 64 training epochs. Our results suggest that choosing a suitable latent distribution structure (achieved via distribution-level alignment), rather than relying on fixed priors, is key to bridging the gap between easy-to-model latents and high-fidelity image synthesis. Code is avaliable at https://github.com/sen-ye/dmvae.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。