Knowledge distillation methods have recently shown to be a promising direction to speedup the synthesis of large-scale diffusion models by requiring only a few inference steps. While several powerful distillation methods were recently proposed, the overall quality of student samples is typically lower compared to the teacher ones, which hinders their practical usage. In this work, we investigate the relative quality of samples produced by the teacher text-to-image diffusion model and its distilled student version. As our main empirical finding, we discover that a noticeable portion of student samples exhibit superior fidelity compared to the teacher ones, despite the ``approximate'' nature of the student. Based on this finding, we propose an adaptive collaboration between student and teacher diffusion models for effective text-to-image synthesis. Specifically, the distilled model produces the initial sample, and then an oracle decides whether it needs further improvements with a slow teacher model. Extensive experiments demonstrate that the designed pipeline surpasses state-of-the-art text-to-image alternatives for various inference budgets in terms of human preference. Furthermore, the proposed approach can be naturally used in popular applications such as text-guided image editing and controllable generation.
翻译:知识蒸馏方法最近被证明是加速大规模扩散模型合成的有前途方向,仅需少量推理步骤即可完成。尽管近期提出了多种强大的蒸馏方法,但学生模型生成的样本整体质量通常低于教师模型,这限制了其实际应用。在本工作中,我们研究了教师文本到图像扩散模型与其蒸馏学生版本所生成样本的相对质量。主要经验发现是:尽管学生模型具有"近似"特性,但相当一部分学生样本在保真度方面优于教师样本。基于此发现,我们提出了一种学生与教师扩散模型之间的自适应协作方法,用于有效的文本到图像合成。具体而言,蒸馏模型首先生成初始样本,然后由预言机决定是否需要通过慢速教师模型进行进一步优化。大量实验表明,就人类偏好而言,所设计的流水线在各种推理预算下均超越了最先进的文本到图像替代方案。此外,所提方法可自然应用于文本引导图像编辑和可控生成等流行场景。