HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

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

BenchmarkMethodologyMetrics
image-generation-on-afhq-catCLR-GAN
FID: 4.45
image-generation-on-celeba-64x64CLR-GAN
FID: 13.63
image-generation-on-cifar-10CLR-GAN
FID: 23.3
image-generation-on-ffhq-256-x-256CLR-GAN
FID: 3.37
Precision: 0.71
Recall: 0.44
image-generation-on-imagenet-64x64CLR-GAN
FID: 20.27
image-generation-on-lsun-churches-256-x-256CLR-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.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp