α-EGAN: a-Energy distance GAN with an early stopping rule

被引:7
作者
Ji, Fangting [1 ]
Zhang, Xin [2 ]
Zhao, Junlong [1 ]
机构
[1] Beijing Normal Univ, Sch Stat, Beijing 100875, Peoples R China
[2] Peoples Liberat Army Gen Hosp, Grad Sch, Dept Stat & Epidemiol, Beijing 100853, Peoples R China
基金
中国国家自然科学基金;
关键词
Generative adversarial networks; Energy distance; Early stopping; Hypothetical testing; Evaluation metric; MEAN-SHIFT; NUMBER;
D O I
10.1016/j.cviu.2023.103748
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Generative adversarial networks (GANs) are popular tools for learning the distribution of real samples and generating new ones, where the qualities of the generated images and the degree of preserved variation are two major concerns currently. In view of this, we propose an a-EGAN with energy distance as the loss function of the generator, which is proven to be effective in mitigating mode collapse. Moreover, an early stopping rule is proposed in the frame of hypothesis testing to avoid the unhealthy competition between the generator and the discriminator, thus achieving a trade-off between image qualities and variations. As a byproduct, the energy distance under the Euclidean norm can serve as a novel metric for evaluating generated samples in GAN. Experiments are conducted on simulated manifold datasets, as well as real MNIST and CelebA face datasets, showing that the proposed a-EGAN outperforms several competitors in both training stability and image quality.
引用
收藏
页数:8
相关论文
共 36 条
[1]  
Arjovsky M., 2017, Towards principled methods for training generative adversarial networks
[2]  
Arjovsky M, 2017, PR MACH LEARN RES, V70
[3]  
Bellemare MG, 2017, Arxiv, DOI arXiv:1705.10743
[4]   MEAN SHIFT, MODE SEEKING, AND CLUSTERING [J].
CHENG, YZ .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 1995, 17 (08) :790-799
[5]  
Collins RT, 2003, PROC CVPR IEEE, P234
[6]  
Dai ZH, 2017, 31 ANN C NEURAL INFO, V30
[7]  
Donahue C, 2019, Arxiv, DOI arXiv:1802.04208
[8]  
Esteban C, 2017, Arxiv, DOI arXiv:1706.02633
[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]  
Hartmann KG, 2018, Arxiv, DOI [arXiv:1806.01875, 10.48550/arXiv.1806.01875]