Journal of the Operations Research Society of China ›› 2019, Vol. 7 ›› Issue (4): 579-597.doi: 10.1007/s40305-019-00248-x

Special Issue: Continuous Optimization

Previous Articles     Next Articles

Generative Adversarial Networks with Joint Distribution Moment Matching

Yi-Ying Zhang1, Chao-Min Shen1,2,3, Hao Feng4, Preston Thomas Fletcher5, Gui-Xu Zhang1   

  1. 1 Department of Computer Science, East China Normal University, Shanghai 200062, China;
    2 Shanghai Key Laboratory of Multidimensional Information Processing, East China Normal University, Shanghai 200062, China;
    3 Westlake Institute for Brain-Like Science and Technology, Hangzhou 310027, China;
    4 Didi Chuxing Science and Technology Co. Ltd., Beijing 100193, China;
    5 Department of Electrical and Computer Engineering, University of Virginia, Charlottesville, VA 22904, USA
  • Received:2018-11-02 Revised:2019-03-27 Online:2019-11-30 Published:2019-11-28
  • Contact: Chao-Min Shen, Yi-Ying Zhang, Hao Feng, Preston Thomas Fletcher, Gui-Xu Zhang E-mail:cmshen@cs.ecnu.edu.cn;yyzhangcs@outlook.com;fengxhao@outlook.com;ptf8v@virginia.edu;gxzhang@cs.ecnu.edu.cn
  • Supported by:
    This work is supported by the National Natural Science Foundation of China (Nos. 11771276, 11471208, 61731009), and the Foundation of Science and Technology Commission of Shanghai Municipality (No. 14DZ2260800).

Abstract: Generative adversarial networks (GANs) have shown impressive power in the field of machine learning. Traditional GANs have focused on unsupervised learning tasks. In recent years, conditional GANs that can generate data with labels have been proposed in semi-supervised learning and have achieved better image quality than traditional GANs. Conditional GANs, however, generally only minimize the difference between marginal distributions of real and generated data, neglecting the difference with respect to each class of the data. To address this challenge, we propose the GAN with joint dis tribution moment matching (JDMM-GAN) for matching the joint distribution based on maximum mean discrepancy, which minimizes the differences of both the marginal and conditional distributions. The learning procedure is iteratively conducted by the stochastic gradient descent and back-propagation. We evaluate JDMM-GAN on several benchmark datasets, including MNIST, CIFAR-10 and the Extended Yale Face. Compared with the state-of-the-art GANs, JDMM-GAN generates more realistic images and achieves the best inception score for CIFAR-10 dataset.

Key words: Generative Adversarial Networks, Joint Distribution Moment Matching, Maximum mean discrepancy

CLC Number: