Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Freeze the Discriminator: a Simple Baseline for Fine-Tuning GANs

About

Generative adversarial networks (GANs) have shown outstanding performance on a wide range of problems in computer vision, graphics, and machine learning, but often require numerous training data and heavy computational resources. To tackle this issue, several methods introduce a transfer learning technique in GAN training. They, however, are either prone to overfitting or limited to learning small distribution shifts. In this paper, we show that simple fine-tuning of GANs with frozen lower layers of the discriminator performs surprisingly well. This simple baseline, FreezeD, significantly outperforms previous techniques used in both unconditional and conditional GANs. We demonstrate the consistent effect using StyleGAN and SNGAN-projection architectures on several datasets of Animal Face, Anime Face, Oxford Flower, CUB-200-2011, and Caltech-256 datasets. The code and results are available at https://github.com/sangwoomo/FreezeD.

Sangwoo Mo, Minsu Cho, Jinwoo Shin• 2020

Related benchmarks

TaskDatasetResultRank
Few-shot Image GenerationSunglasses 10-shot
FID46.95
36
Few-shot Image GenerationBabies 10-shot
FID96.25
35
Few-shot Image GenerationAFHQ-Cat 10-shot
FID63.6
34
Few-shot Image GenerationAFHQ-Wild 10-shot
FID77.18
34
Few-shot Image GenerationAFHQ-Dog 10-shot
FID158
34
Few-shot Image GenerationMetFaces 10-shot
FID73.33
34
Image GenerationObama 100-shot (train)
FID41.87
28
Image GenerationGrumpy cat 100-shot (train)
FID31.22
28
Image GenerationPanda 100-shot (train)
FID17.95
28
Few-shot Image GenerationObama 100-shot
FID41.87
26
Showing 10 of 63 rows

Other info

Code

Follow for update