用噪声数据训练生成模型,一步生成高质量图像
Denoising Score Distillation: From Noisy Diffusion Pretraining to One-Step High-Quality Generation
- 从噪声样本中预训练扩散模型,再通过得分蒸馏转化为单步生成器
- 在多种噪声水平下均显著提升生成质量,超越原始教师模型
- 适合缺乏干净数据的科研场景,兼具理论深度与实用价值
扩散模型在生成高分辨率、逼真图像方面取得了显著成功,但其性能高度依赖高质量训练数据,难以从有噪声或损坏的样本中学习有意义的分布。这限制了其在科学领域中的应用,因为这些领域往往缺乏清洁数据或获取成本高昂。本文提出去噪得分蒸馏(DSD),一种新颖且高效的方法,可从低质量数据中训练出高质量生成模型。DSD首先仅使用噪声样本对扩散模型进行预训练,随后将其蒸馏为一个可生成清晰输出的一步生成器。尽管得分蒸馏传统上被视为加速扩散模型的手段,我们发现它也能显著提升样本质量,尤其当起始教师模型本身质量较差时。在不同噪声水平和数据集上,DSD始终一致地提升生成性能(见图1)。此外,我们提供了理论分析表明,在线性模型设定下,DSD能够识别出干净数据分布协方差矩阵的特征空间,从而隐式正则化生成器。这一视角将得分蒸馏重新定义为不仅提升效率,更是在低质量数据条件下改进生成模型的有效机制。
原文摘要 · Abstract (English)
Diffusion models have achieved remarkable success in generating high-resolution, realistic images across diverse natural distributions. However, their performance heavily relies on high-quality training data, making it challenging to learn meaningful distributions from corrupted samples. This limitation restricts their applicability in scientific domains where clean data is scarce or costly to obtain. In this work, we introduce denoising score distillation (DSD), a surprisingly effective and novel approach for training high-quality generative models from low-quality data. DSD first pretrains a diffusion model exclusively on noisy, corrupted samples and then distills it into a one-step generator capable of producing refined, clean outputs. While score distillation is traditionally viewed as a method to accelerate diffusion models, we show that it can also significantly enhance sample quality, particularly when starting from a degraded teacher model. Across varying noise levels and datasets, DSD consistently improves generative performancewe summarize our empirical evidence in Fig. 1. Furthermore, we provide theoretical insights showing that, in a linear model setting, DSD identifies the eigenspace of the clean data distributions covariance matrix, implicitly regularizing the generator. This perspective reframes score distillation as not only a tool for efficiency but also a mechanism for improving generative models, particularly in low-quality data settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。