让扩散模型生成更稳定,减少采样时的随机波动干扰
Taming Sampling Perturbations with Variance Expansion Loss for Latent Diffusion Models
- 引入方差扩展损失,防止潜在空间过紧导致的敏感性
- 在保持重建精度的同时,显著提升对采样扰动的鲁棒性
- 适用于各类潜在扩散模型,尤其适合追求高质量生成的场景
潜在扩散模型因其在紧凑潜在空间中学习扩散过程的能力,已成为高保真、高效图像生成的主流框架。然而,现有研究主要关注潜在空间的重建精度与语义对齐,我们发现另一个关键因素——对采样扰动的鲁棒性——同样影响生成质量。通过实证与理论分析,我们发现常用的基于β-VAE的编码器会产生过于紧凑的潜在流形,在扩散采样过程中对随机扰动极为敏感,导致视觉退化。为此,我们提出一种简单而有效的方法:引入方差扩展损失,对抗方差坍缩,利用重建与方差扩展之间的对抗性协同作用,实现重建精度与采样鲁棒性的自适应平衡。大量实验表明,该方法在不同潜在扩散架构下均能持续提升生成质量,证实了潜在空间鲁棒性是稳定、忠实扩散采样的关键缺失要素。
原文摘要 · Abstract (English)
Latent diffusion models have emerged as the dominant framework for high-fidelity and efficient image generation, owing to their ability to learn diffusion processes in compact latent spaces. However, while previous research has focused primarily on reconstruction accuracy and semantic alignment of the latent space, we observe that another critical factor, robustness to sampling perturbations, also plays a crucial role in determining generation quality. Through empirical and theoretical analyses, we show that the commonly used $β$-VAE-based tokenizers in latent diffusion models, tend to produce overly compact latent manifolds that are highly sensitive to stochastic perturbations during diffusion sampling, leading to visual degradation. To address this issue, we propose a simple yet effective solution that constructs a latent space robust to sampling perturbations while maintaining strong reconstruction fidelity. This is achieved by introducing a Variance Expansion loss that counteracts variance collapse and leverages the adversarial interplay between reconstruction and variance expansion to achieve an adaptive balance that preserves reconstruction accuracy while improving robustness to stochastic sampling. Extensive experiments demonstrate that our approach consistently enhances generation quality across different latent diffusion architectures, confirming that robustness in latent space is a key missing ingredient for stable and faithful diffusion sampling.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。