Least Squares Generative Adversarial Networks

被引:3562
作者
Mao, Xudong [1 ]
Li, Qing [1 ]
Xie, Haoran [2 ]
Lau, Raymond Y. K. [3 ]
Wang, Zhen [4 ]
Smolley, Stephen Paul [5 ]
机构
[1] City Univ Hong Kong, Dept Comp Sci, Hong Kong, Hong Kong, Peoples R China
[2] Educ Univ Hong Kong, Dept Math & Informat Technol, Hong Kong, Hong Kong, Peoples R China
[3] City Univ Hong Kong, Dept Informat Syst, Hong Kong, Hong Kong, Peoples R China
[4] Northwestern Polytech Univ, Ctr Opt Imagery Anal & Learning, Fremont, CA USA
[5] CodeHatch Corp, Edmonton, AB, Canada
来源
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV) | 2017年
关键词
D O I
10.1109/ICCV.2017.304
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Unsupervised learning with generative adversarial networks (GANs) has proven hugely successful. Regular GANs hypothesize the discriminator as a classifier with the sigmoid cross entropy loss function. However, we found that this loss function may lead to the vanishing gradients problem during the learning process. To overcome such a problem, we propose in this paper the Least Squares Generative Adversarial Networks (LSGANs) which adopt the least squares loss function for the discriminator. We show that minimizing the objective function of LSGAN yields minimizing the Pearson chi(2) divergence. There are two benefits of LSGANs over regular GANs. First, LSGANs are able to generate higher quality images than regular GANs. Second, LSGANs perform more stable during the learning process. We evaluate LSGANs on LSUN and CIFAR-10 datasets and the experimental results show that the images generated by LSGANs are of better quality than the ones generated by regular GANs. We also conduct two comparison experiments between LSGANs and regular GANs to illustrate the stability of LSGANs.
引用
收藏
页码:2813 / 2821
页数:9
相关论文
共 32 条
[1]  
ABADI M, 2015, TENSORFLOW LARGE SCA, DOI DOI 10.48550/ARXIV.1605.08695
[2]  
[Anonymous], ARXIV161202136
[3]  
[Anonymous], 2016, Unrolled generative adversarial networks
[4]  
[Anonymous], 2016, ARXIV161200005
[5]  
[Anonymous], 2017, ARXIV170106264
[6]  
Arjovsky M., 2017, ARXIV170107875
[7]  
Chen X., 2016, P ADV NEUR INF PROC
[8]  
Denton E. L., 2015, ADV NEURAL INFORM PR, P1486, DOI DOI 10.5555/2969239.2969405
[9]   Generative Adversarial Networks [J].
Goodfellow, Ian ;
Pouget-Abadie, Jean ;
Mirza, Mehdi ;
Xu, Bing ;
Warde-Farley, David ;
Ozair, Sherjil ;
Courville, Aaron ;
Bengio, Yoshua .
COMMUNICATIONS OF THE ACM, 2020, 63 (11) :139-144
[10]  
He K., 2016, PROC CVPR IEEE, P630, DOI [10.1007/978-3-319-46493-0_38, DOI 10.1007/978-3-319-46493-0_38, DOI 10.1109/CVPR.2016.90]