Command Palette
Search for a command to run...
CLR-GAN: Improving GANs Stability and Quality via Consistent Latent Representation and Reconstruction
{Shijie Luo and Shuzhen Han Zhanshan Zhao Ziqian Luan Shengke Sun}
Abstract
Generative Adversarial Networks(GANs) have received considerable attention due to its outstanding ability to generate images. However, training a GAN is hard since the game between the Generator(G) and the Discriminator(D) is unfair. Towards making the competition fairer, we propose a new perspective of training GANs, named Consistent Latent Representation and Reconstruction(CLR-GAN). In this paradigm, we treat the G and D as an inverse process, the discriminator has an additional task to restore the pre-defined latent code while the generator also needs to reconstruct the real input, thus obtaining a relationship between the latent space of G and the out-features of D. Based on this prior, we can put D and G on an equal position during training using a new criterion. Experimental results on various datasets and architectures prove our paradigm can make GANs more stable and generate better quality images(31.22% gain of FID on CIFAR10 and 39.5% on AFHQ-Cat, respectively). We hope that the proposed perspective can inspire researchers to explore different ways of viewing GANs training, rather than being limited to a two-player game. The code is publicly available at https://github.com/Petecheco/CLR-GAN.
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| image-generation-on-afhq-cat | CLR-GAN | FID: 4.45 |
| image-generation-on-celeba-64x64 | CLR-GAN | FID: 13.63 |
| image-generation-on-cifar-10 | CLR-GAN | FID: 23.3 |
| image-generation-on-ffhq-256-x-256 | CLR-GAN | FID: 3.37 Precision: 0.71 Recall: 0.44 |
| image-generation-on-imagenet-64x64 | CLR-GAN | FID: 20.27 |
| image-generation-on-lsun-churches-256-x-256 | CLR-GAN | FID: 3.43 Recall: 0.48 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.