LLM2D
增强的扩散采样通过多重ODE解的外推
Enhanced Diffusion Sampling via Extrapolation with Multiple ODE Solutions
作者: Jinyoung Choi, Junoh Kang, Bohyung Han
发布日期: 4/3/2025
arXiv ID: oai:arXiv.org:2504.01855v1

摘要

arXiv:2504.01855v1 公告类型: cross 摘要: 随机扩散模型(DPMs)虽然在生成高质量样本方面非常有效,但由于它们的迭代采样过程,往往面临计算成本高的问题。为了解决这一问题,我们提出了一种受Richardson外推法启发的增强型基于ODE的采样方法,以减少数值误差并提高收敛率。我们的方法RX-DPM利用DPM在中间时间步的多个ODE解来外推去噪预测。这种方法显著提高了最终样本估计的准确性,同时保持了函数评估次数(NFEs)。与标准的Richardson外推法不同,后者假设时间网格的均匀离散,我们开发了一种更通用的公式,适用于任意时间步调度,指导思想是基于基线采样方法推导出的局部截断误差。我们方法的简洁性使得在不显著增加计算开销的情况下可以准确估计数值解,并且可以无缝且方便地集成到各种DPM和求解器中。此外,RX-DPM提供了显式的误差估计,有效地展示了随着主要误差项阶数的增加,收敛速度加快的现象。通过一系列实验,我们证明了所提出的方法在不需要额外采样迭代的情况下提高了生成样本的质量。