用随机条件+蒸馏,让小模型学会生成训练中没出现过的图像。
Random Conditioning with Distillation for Data-Efficient Diffusion Model Compression
- 用噪声图配随机文本做蒸馏,无需真实图像。
- 学生模型能生成训练时未见的概念,效果更好。
- 适合资源有限但需高质量生成的场景。
扩散模型通过逐步去噪生成高质量图像,但因模型庞大且需反复采样而计算成本高。知识蒸馏将复杂教师模型的知识迁移到更简单的学生模型,在识别任务中广泛应用,尤其擅长传递学生训练中未见的概念。然而,其在扩散模型中的应用仍不充分,尤其是如何让学生生成训练图像中未包含的概念。本文提出随机条件(Random Conditioning)方法,将噪声图像与随机选择的文本条件配对,实现高效的无图像知识蒸馏。通过该方法,学生模型可生成训练中未出现的概念。应用于条件扩散模型蒸馏时,该方法使学生在不生成特定条件图像的情况下探索条件空间,显著提升生成质量与效率。这推动了生成式扩散模型的轻量化部署,扩大其在科研与实际应用中的可及性。代码、模型与数据集见 https://dohyun-as.github.io/Random-Conditioning。
原文摘要 · Abstract (English)
Diffusion models generate high-quality images through progressive denoising but are computationally intensive due to large model sizes and repeated sampling. Knowledge distillation, which transfers knowledge from a complex teacher to a simpler student model, has been widely studied in recognition tasks, particularly for transferring concepts unseen during student training. However, its application to diffusion models remains underexplored, especially in enabling student models to generate concepts not covered by the training images. In this work, we propose Random Conditioning, a novel approach that pairs noised images with randomly selected text conditions to enable efficient, image-free knowledge distillation. By leveraging this technique, we show that the student can generate concepts unseen in the training images. When applied to conditional diffusion model distillation, our method allows the student to explore the condition space without generating condition-specific images, resulting in notable improvements in both generation quality and efficiency. This promotes resource-efficient deployment of generative diffusion models, broadening their accessibility for both research and real-world applications. Code, models, and datasets are available at https://dohyun-as.github.io/Random-Conditioning .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。