Parallel Wasserstein Generative Adversarial Nets with Multiple Discriminators

Parallel Wasserstein Generative Adversarial Nets with Multiple Discriminators

Yuxin Su, Shenglin Zhao, Xixian Chen, Irwin King, Michael Lyu

Proceedings of the Twenty-Eighth International Joint Conference on Artificial Intelligence
Main track. Pages 3483-3489. https://doi.org/10.24963/ijcai.2019/483

Wasserstein Generative Adversarial Nets~(GANs) are newly proposed GAN algorithms and widely used in computer vision, web mining, information retrieval, etc. However, the existing algorithms with approximated Wasserstein loss converge slowly due to heavy computation cost and usually generate unstable results as well. In this paper, we solve the computation cost problem by speeding up the Wasserstein GANs from a well-designed communication efficient parallel architecture. Specifically, we develop a new problem formulation targeting the accurate evaluation of Wasserstein distance and propose an easily parallel optimization algorithm to train the Wasserstein GANs. Compared to traditional parallel architecture, our proposed framework is designed explicitly for the skew parameter updates between the generator network and discriminator network. Rigorous experiments reveal that our proposed framework achieves a significant improvement regarding convergence speed with comparable stability on generating images, compared to the state-of-the-art of Wasserstein GANs algorithms.
Keywords:
Machine Learning: Deep Learning
Machine Learning: Adversarial Machine Learning