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

Data-Efficient Instance Generation from Instance Discrimination

About

Generative Adversarial Networks (GANs) have significantly advanced image synthesis, however, the synthesis quality drops significantly given a limited amount of training data. To improve the data efficiency of GAN training, prior work typically employs data augmentation to mitigate the overfitting of the discriminator yet still learn the discriminator with a bi-classification (i.e., real vs. fake) task. In this work, we propose a data-efficient Instance Generation (InsGen) method based on instance discrimination. Concretely, besides differentiating the real domain from the fake domain, the discriminator is required to distinguish every individual image, no matter it comes from the training set or from the generator. In this way, the discriminator can benefit from the infinite synthesized samples for training, alleviating the overfitting problem caused by insufficient training data. A noise perturbation strategy is further introduced to improve its discriminative power. Meanwhile, the learned instance discrimination capability from the discriminator is in turn exploited to encourage the generator for diverse generation. Extensive experiments demonstrate the effectiveness of our method on a variety of datasets and training settings. Noticeably, on the setting of 2K training images from the FFHQ dataset, we outperform the state-of-the-art approach with 23.5% FID improvement.

Ceyuan Yang, Yujun Shen, Yinghao Xu, Bolei Zhou• 2021

Related benchmarks

TaskDatasetResultRank
Image GenerationObama 100-shot (train)
FID32.42
28
Image GenerationGrumpy cat 100-shot (train)
FID22.01
28
Image GenerationPanda 100-shot (train)
FID9.85
28
Few-shot Image GenerationObama 100-shot
FID32.42
26
Few-shot Image GenerationGrumpy Cat 100-shot
FID22.01
26
Image GenerationAnimalFace Dog
FID44.93
21
Image GenerationAnimalFace Dog standard (train)
FID44.93
20
Image GenerationAnimalFace Cat standard (train)
FID33.01
20
Image GenerationAFHQ Cat v1 (test)
FID2.6
15
Image GenerationFFHQ 256x256 50k (test)
FID3.31
15
Showing 10 of 32 rows

Other info

Code

Follow for update