Aug, 2024

加速扩散模型取样的简单早期退出框架

TL;DR该研究解决了扩散模型在取样速度上存在的瓶颈,通过提出一种新的框架,能够自适应地分配计算资源以估计得分,从而减少整体取样时间。研究发现,得分估计所需的计算量会随着时间步变化,提出的早期退出策略能够在不影响图像质量的前提下显著提高取样吞吐量。