arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.03420cs.LGstat.ML
收藏

Towards Generalized Implementation of Wasserstein Distance in GANs

University of Montreal(蒙特利尔大学);Mila - Quebec AI Institute(Mila - 魁北克人工智能研究所);Shanghai University of Finance and Economics(上海财经大学);Shanghai Jiao Tong University(上海交通大学);HEC Montreal(蒙特利尔高等商学院);CIFAR AI Research Chair(CIFAR人工智能研究讲席)

Minkai Xu, Zhiming Zhou, Guansong Lu, Jian Tang, Weinan Zhang, Yong Yu

英文摘要:

Wasserstein GANs (WGANs), built upon the Kantorovich-Rubinstein (KR) duality of Wasserstein distance, is one of the most theoretically sound GAN models. However, in practice it does not always outperform other variants of GANs. This is mostly due to the imperfect implementation of the Lipschitz condition required by the KR duality. Extensive work has been done in the community with different implementations of the Lipschitz constraint, which, however, is still hard to satisfy the restriction perfectly in practice. In this paper, we argue that the strong Lipschitz constraint might be unnecessary for optimization. Instead, we take a step back and try to relax the Lipschitz constraint. Theoretically, we first demonstrate a more general dual form of the Wasserstein distance called the Sobolev duality, which relaxes the Lipschitz constraint but still maintains the favorable gradient property of the Wasserstein distance. Moreover, we show that the KR duality is actually a special case of the Sobolev duality. Based on the relaxed duality, we further propose a generalized WGAN training scheme named Sobolev Wasserstein GAN (SWGAN), and empirically demonstrate the improvement of SWGAN over existing methods with extensive experiments.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑