摘要
arXiv:2505.03983v1 交叉类型: cross
摘要:去噪扩散概率模型(DDPMs)已成为生成建模的强大力量。然而,它们的顺序计算需求导致了显著的推理时间瓶颈。在本文中,我们利用DDPMs与随机局部化的连接,证明在适当的重参数化下,DDPM的增量满足交换性性质。这一一般性的见解使得可以将自回归模型中的各种性能优化技术近乎无损地适应到扩散设置中。为了证明这一点,我们引入了\emph{自动推测解码}(ASD),这是一种将广泛使用的推测解码算法扩展到DDPM中,并且不需要任何辅助草稿模型的方法。我们的理论分析表明,ASD 在 $K$ 步顺序DDPM上的并行运行时间加速达到了 $\tilde{O}(K^{\frac{1}{3}})$。我们还展示了自动推测解码的实际实现可以显著加速DDPM推理在各种领域的运行。