无需GAN即可用真实数据提升各类生成模型的推理速度
Universal Inverse Distillation for Matching Models with Real-Data Supervision (No GANs)
- 提出通用蒸馏框架RealUID,统一适配扩散、流模型等
- 在不依赖对抗训练的情况下,实现真实数据驱动的单步生成
- 适用于多种生成模型,代码开源,易复现
尽管现代扩散模型、流模型等生成质量优异,但推理速度慢,需多步迭代生成。现有蒸馏方法虽能训练高效单步生成器,但通常仅适用于特定框架(如仅限扩散或流模型),且原为无数据方法。若要引入真实数据,需额外使用复杂的对抗训练和判别器。本文提出RealUID,一种适用于所有匹配模型的通用蒸馏框架,可在无需GAN的前提下,无缝融合真实数据到蒸馏过程。该方法具有简洁的理论基础,涵盖先前针对流匹配和扩散模型的蒸馏方法,并可扩展至桥接匹配与随机插值等变体。代码已开源。
原文摘要 · Abstract (English)
While achieving exceptional generative quality, modern diffusion, flow, and other matching models suffer from slow inference, as they require many steps of iterative generation. Recent distillation methods address this problem by training efficient one-step generators under the guidance of a pre-trained teacher model. However, these methods are often constrained to only one specific framework, e.g., only to diffusion or only to flow models. Furthermore, these methods are originally data-free, and to benefit from the usage of real data, it is required to use an additional complex adversarial training with an extra discriminator model. In this paper, we present RealUID, a universal distillation framework for all matching models that seamlessly incorporates real data into the distillation procedure without GANs. Our RealUID approach offers a simple theoretical foundation that covers previous distillation methods for Flow Matching and Diffusion models, and can be also extended to their modifications, such as Bridge Matching and Stochastic Interpolants. The code can be found in https://github.com/David-cripto/RealUID.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。