Apr, 2024

基于分值的生成模型的概率流动常微分方程的收敛分析

TL;DR基于分数的生成模型中,我们从理论和数值的角度研究了基于概率流 ODE 的确定性采样器的收敛性质,并证明了目标和生成数据分布之间的总变差可以在连续时间层面上通过 d√δ(其中 d 表示数据维度,δ 表示 L2 - 评分匹配误差)被上界限制,并针对使用 p 阶 Runge-Kutta 积分器进行具体实现的情况,建立了离散层面上的误差界限为 O (d (√δ + (dh)^p))。最后,我们进行了高达 128 个维度的问题的数值研究,验证了我们的理论,结果表明更好的评分匹配误差和维度依赖性。