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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Few-shot Image Generation | Sunglasses 10-shot | FID46.95 | 36 | |
| Few-shot Image Generation | Babies 10-shot | FID96.25 | 35 | |
| Few-shot Image Generation | AFHQ-Cat 10-shot | FID63.6 | 34 | |
| Few-shot Image Generation | AFHQ-Wild 10-shot | FID77.18 | 34 | |
| Few-shot Image Generation | AFHQ-Dog 10-shot | FID158 | 34 | |
| Few-shot Image Generation | MetFaces 10-shot | FID73.33 | 34 | |
| Image Generation | Obama 100-shot (train) | FID41.87 | 28 | |
| Image Generation | Grumpy cat 100-shot (train) | FID31.22 | 28 | |
| Image Generation | Panda 100-shot (train) | FID17.95 | 28 | |
| Few-shot Image Generation | Obama 100-shot | FID41.87 | 26 |