Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Training Generative Adversarial Networks with Limited Data

About

Training generative adversarial networks (GAN) using too little data typically leads to discriminator overfitting, causing training to diverge. We propose an adaptive discriminator augmentation mechanism that significantly stabilizes training in limited data regimes. The approach does not require changes to loss functions or network architectures, and is applicable both when training from scratch and when fine-tuning an existing GAN on another dataset. We demonstrate, on several datasets, that good results are now possible using only a few thousand training images, often matching StyleGAN2 results with an order of magnitude fewer images. We expect this to open up new application domains for GANs. We also find that the widely used CIFAR-10 is, in fact, a limited data benchmark, and improve the record FID from 5.59 to 2.42.

Tero Karras, Miika Aittala, Janne Hellsten, Samuli Laine, Jaakko Lehtinen, Timo Aila• 2020

Related benchmarks

TaskDatasetResultRank
Image GenerationCIFAR-10 (test)
FID2.92
536
Unconditional Image GenerationCIFAR-10
FID2.92
280
Unconditional Image GenerationCIFAR-10 (test)
FID2.92
223
Image GenerationCIFAR-10
FID2.42
212
Unconditional Image GenerationCIFAR-10 unconditional
FID2.92
209
Image GenerationCelebA 64 x 64 (test)
FID2.32
208
Image GenerationCIFAR10 32x32 (test)
FID2.92
186
Image GenerationCIFAR-10
Inception Score10.06
178
Unconditional Image GenerationCIFAR-10 32x32 (test)
FID2.92
137
Image GenerationLSUN church
FID5.85
117
Showing 10 of 179 rows
...

Other info

Follow for update