用扩散模型的得分信息蒸馏出高效生成的恒等映射模型,兼顾速度与质量。
Score-based Idempotent Distillation of Diffusion Models
- 从扩散模型得分中蒸馏出恒等生成网络,无需对抗训练
- 可单步或多步采样,生成速度比传统方法快,质量仍领先
- 支持零样本输入编辑,适合需要快速生成的场景
恒等生成网络(IGNs)是一类基于恒等映射到目标流形的新生成模型,支持单步和多步生成,可在计算成本与样本质量间灵活权衡。但传统IGNs类似生成对抗网络(GANs),需对抗训练,易出现训练不稳定和模式崩溃问题。扩散模型与得分模型通过逐步将样本从高斯分布迁移到目标数据分布实现生成,因其训练稳定且生成质量高而广受欢迎,但代价是计算开销大,需沿完整轨迹迭代推进。为降低采样成本,已有研究提出新采样方法、模型蒸馏和一致性模型,甚至实现扩散模型的一次性采样。本文提出将扩散模型与IGNs结合,通过蒸馏扩散模型的得分信息构建符号化恒等生成模型(SIGN)。所提方法高度稳定,无需对抗损失。我们提供了分数训练方法的理论分析,并实证表明可有效从预训练扩散模型中蒸馏出IGNs,实现比迭代得分模型更快的推理。SIGN支持多步采样,允许用户在质量与效率间权衡。这些模型直接作用于源域,可将噪声或替代分布投影回目标流形,实现零样本输入编辑。我们在多个图像数据集上验证模型,在CIFAR和CelebA数据集上达到当前最优的恒等生成模型性能。
原文摘要 · Abstract (English)
Idempotent generative networks (IGNs) are a new line of generative models based on idempotent mapping to a target manifold. IGNs support both single-and multi-step generation, allowing for a flexible trade-off between computational cost and sample quality. But similar to Generative Adversarial Networks (GANs), conventional IGNs require adversarial training and are prone to training instabilities and mode collapse. Diffusion and score-based models are popular approaches to generative modeling that iteratively transport samples from one distribution, usually a Gaussian, to a target data distribution. These models have gained popularity due to their stable training dynamics and high-fidelity generation quality. However, this stability and quality come at the cost of high computational cost, as the data must be transported incrementally along the entire trajectory. New sampling methods, model distillation, and consistency models have been developed to reduce the sampling cost and even perform one-shot sampling from diffusion models. In this work, we unite diffusion and IGNs by distilling idempotent models from diffusion model scores, called SIGN. Our proposed method is highly stable and does not require adversarial losses. We provide a theoretical analysis of our proposed score-based training methods and empirically show that IGNs can be effectively distilled from a pre-trained diffusion model, enabling faster inference than iterative score-based models. SIGNs can perform multi-step sampling, allowing users to trade off quality for efficiency. These models operate directly on the source domain; they can project corrupted or alternate distributions back onto the target manifold, enabling zero-shot editing of inputs. We validate our models on multiple image datasets, achieving state-of-the-art results for idempotent models on the CIFAR and CelebA datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。