$(q,p)$-Wasserstein GANs:Wasserstein GANs 的底度量比较
机器学习
2019-02-12 v1 机器学习
摘要
生成对抗网络(GANs)作为生成模型在计算机视觉和机器学习领域产生了重大影响。Wasserstein GANs(WGANs)通过将模型分布与数据分布之间的 -Wasserstein 距离最小化作为目标函数,将最优传输(OT)理论引入了 GANs。此后,由于其稳定性和理论框架,WGANs 引起了广泛关注。我们通过引入 -Wasserstein GANs 家族为 WGAN 文献做出了贡献,该家族允许在 GAN 学习过程中使用更一般的 -Wasserstein 度量()。虽然该方法能够纳入任何代价函数作为底度量,但我们重点研究 的 度量。这是一个值得注意的推广,因为在 WGAN 文献中,OT 距离通常基于 底度量。我们在两个简单示例中展示了不同 -Wasserstein 距离的效果。此外,通过在 MNIST 和 CIFAR-10 数据集上比较不同的 对,我们证明了底度量确实会产生影响。我们的实验表明,改变底度量和 可以显著改进常见的 情形。
引用
@article{arxiv.1902.03642,
title = {(q,p)-Wasserstein GANs: Comparing Ground Metrics for Wasserstein GANs},
author = {Anton Mallasto and Jes Frellsen and Wouter Boomsma and Aasa Feragen},
journal= {arXiv preprint arXiv:1902.03642},
year = {2019}
}