Feb, 2024

可扩展的 Wasserstein 梯度流用于非平衡最优传输的生成建模

TL;DR本文提出了一种可扩展的基于 Wasserstein 梯度流的生成模型,称为 Semi-dual JKO (S-JKO),通过使用 JKO 步骤的半对偶形式,将训练复杂度降低至 O (K),实验证明该模型在 CIFAR-10 和 CelebA-HQ-256 数据集上取得了显著优于现有 WGF 模型的 FID 分数,与最先进的图像生成模型相当。