score_sde:通过随机微分方程式基于分数的生成建模的官方代码(ICLR 2021口头) 源码

通过随机微分方程的基于分数的生成建模 此存储库包含论文的正式实现 作者:, , , , 和 我们提出了一个统一的框架,该框架通过随机微分方程(SDE)的角度概括和改进了基于分数的生成模型的先前工作。 特别是,我们可以使用SDE描述的连续时间随机过程将数据转换为简单的噪声分布。 如果我们知道每个中间时间步的边际分布得分,则可以将其转换为样本生成,可以通过得分匹配进行估算。 基本概念如下图所示: 我们的工作可以更好地理解现有方法,新的采样算法,精确似然计算,唯一可识别的编码,潜在代码操作,并为家庭带来新的条件生成能力(包括但不限于类条件生成,修复和着色)基于分数的生成模型。 综合起来,我们在CIFAR-10上无条件生成以及在高保真度生成1024px Celeba-HQ图像时获得的FID为2.20 ,初始得分为9.89 (以下示例)。 另外,我们在均匀反量化的CIFAR-10图像上