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

Wasserstein-Wasserstein Auto-Encoders

Xi’an Jiaotong University(西安交通大学);Zhongnan University of Economics and Law(中南财经政法大学);The Hong Kong University of Science and Technology(香港科技大学);Duke-NUS Medical School(杜克-新加坡国立大学医学院)

Shunkang Zhang, Yuan Gao, Yuling Jiao, Jin Liu, Yang Wang, Can Yang

英文摘要:

To address the challenges in learning deep generative models (e.g.,the blurriness of variational auto-encoder and the instability of training generative adversarial networks, we propose a novel deep generative model, named Wasserstein-Wasserstein auto-encoders (WWAE). We formulate WWAE as minimization of the penalized optimal transport between the target distribution and the generated distribution. By noticing that both the prior $P_Z$ and the aggregated posterior $Q_Z$ of the latent code Z can be well captured by Gaussians, the proposed WWAE utilizes the closed-form of the squared Wasserstein-2 distance for two Gaussians in the optimization process. As a result, WWAE does not suffer from the sampling burden and it is computationally efficient by leveraging the reparameterization trick. Numerical results evaluated on multiple benchmark datasets including MNIST, fashion- MNIST and CelebA show that WWAE learns better latent structures than VAEs and generates samples of better visual quality and higher FID scores than VAEs and GANs.

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

↑