HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Projected GANs Converge Faster

Axel Sauer Kashyap Chitta Jens Müller Andreas Geiger

Projected GANs Converge Faster

Abstract

Generative Adversarial Networks (GANs) produce high-quality images but are challenging to train. They need careful regularization, vast amounts of compute, and expensive hyper-parameter sweeps. We make significant headway on these issues by projecting generated and real samples into a fixed, pretrained feature space. Motivated by the finding that the discriminator cannot fully exploit features from deeper layers of the pretrained model, we propose a more effective strategy that mixes features across channels and resolutions. Our Projected GAN improves image quality, sample efficiency, and convergence speed. It is further compatible with resolutions of up to one Megapixel and advances the state-of-the-art Fréchet Inception Distance (FID) on twenty-two benchmark datasets. Importantly, Projected GANs match the previously lowest FIDs up to 40 times faster, cutting the wall-clock time from 5 days to less than 3 hours given the same computational resources.

Code Repositories

tsubota-kouga/ProjectedGAN
pytorch
Mentioned in GitHub
dome272/ProjectedGAN-pytorch
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
conditional-image-generation-on-artbench-10Projected GAN
FID: 11.837
image-generation-on-ade-indoorProjected GAN
FID: 6.7
image-generation-on-afhq-catProjected GAN
FID: 2.16
image-generation-on-afhq-dogProjected GAN
FID: 4.52
image-generation-on-afhq-wildProjected GAN
FID: 2.17
image-generation-on-cityscapesProjected GAN
FID-10k-training-steps: 3.41
image-generation-on-clevrProjected GAN
FID-5k-training-steps: 0.89
image-generation-on-cub-128-x-128Projected GAN
FID: 2.79
image-generation-on-ffhq-256-x-256Projected-GAN (Official)
FID: 3.39
Precision: 0.654
Recall: 0.464
image-generation-on-lsun-bedroom-256-x-256Projected GAN (DINOv2)
FD: 636.35
Precision: 0.80
Recall: 0.23
image-generation-on-lsun-bedroom-256-x-256Projected GAN
FID: 1.52
FID-10k-training-steps: 1.52
image-generation-on-lsun-cat-256-x-256Projected GAN
FID: 3.89
image-generation-on-lsun-churches-256-x-256Projected GAN
FID: 1.59
image-generation-on-lsun-horse-256-x-256Projected GAN
FID: 2.17
image-generation-on-oxford-102-flowers-256-xProjected GAN
FID: 3.86
image-generation-on-pokemon-1024x1024Projected GAN
FID: 33.96
image-generation-on-pokemon-256x256Projected GAN
FID: 26.36
image-generation-on-stanford-carsProjected GANs
FID: 2.09
image-generation-on-stanford-dogsProjected GAN
FID: 11.75

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