Improved Generative Adversarial Networks with Reconstruction Loss

Yanchun Li,Nanfeng Xiao,Wanli Ouyang
DOI: https://doi.org/10.1016/j.neucom.2018.10.014
IF: 6
2018-01-01
Neurocomputing
Abstract:We propose a simple regularization scheme to handle the problem of mode missing and unstable training in the generative adversarial networks (GAN). The key idea is to utilize the visual features learned by the discriminator. We reconstruct the real data by feeding the generator with the real data features extracted by the discriminator. A reconstruction loss is added to the GAN’s objective function to enforce the generator can reconstruct from the features of the discriminator, which helps to explicitly guide the generator towards to near the probable configurations of real data. The proposed reconstruction loss improves the performance of GAN, produces higher quality images on different dataset, and can be easily combined with other regularization loss functions such as gradient penalty to improve the performance of various GANs. We conducted experiments on the widespread adopted architecture DCGAN and the complicated ResNet architecture across different datasets, the results of which show the effectiveness and robustness of our proposed method.
What problem does this paper attempt to address?