Oct, 2023

CADS: 通过条件退火采样释放扩散模型的多样性

TL;DR通过使用一种改进的采样策略,我们提供了一种可以增加扩散模型的生成多样性的方法,尤其是在高指导尺度下,同时最小化样本质量损失。该方法通过在推理过程中向条件向量添加预定的、单调递减的高斯噪声来退火条件信号,以平衡多样性和条件对齐。我们的 Condition-Annealed Diffusion Sampler (CADS) 可以与任何预训练模型和采样算法一起使用,并且我们证明它在各种条件生成任务中提高了扩散模型的多样性。此外,通过使用现有的预训练扩散模型,CADS 在 256×256 和 512×512 像素的类条件 ImageNet 生成任务中取得了新的最优 FID 值,分别为 1.70 和 2.31。