Oct, 2024

多学生扩散蒸馏提升一次性生成器性能

TL;DR本文解决了扩散模型在多步骤推理中生成高质量样本的问题,提出了一种多学生蒸馏框架(MSD),该框架可以将条件教师扩散模型蒸馏为多个单步生成器。通过为每个学生生成器分配条件数据的子集,MSD在相同能力下实现了更高的生成质量,同时显著加快了推理速度,并在一次性图像生成上获得了新的最优成绩。