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

MineGAN: effective knowledge transfer from GANs to target domains with few images

About

One of the attractive characteristics of deep neural networks is their ability to transfer knowledge obtained in one domain to other related domains. As a result, high-quality networks can be trained in domains with relatively little training data. This property has been extensively studied for discriminative networks but has received significantly less attention for generative models. Given the often enormous effort required to train GANs, both computationally as well as in the dataset collection, the re-use of pretrained GANs is a desirable objective. We propose a novel knowledge transfer method for generative models based on mining the knowledge that is most beneficial to a specific target domain, either from a single or multiple pretrained GANs. This is done using a miner network that identifies which part of the generative distribution of each pretrained GAN outputs samples closest to the target domain. Mining effectively steers GAN sampling towards suitable regions of the latent space, which facilitates the posterior finetuning and avoids pathologies of other methods such as mode collapse and lack of flexibility. We perform experiments on several complex datasets using various GAN architectures (BigGAN, Progressive GAN) and show that the proposed method, called MineGAN, effectively transfers knowledge to domains with few target images, outperforming existing methods. In addition, MineGAN can successfully transfer knowledge from multiple pretrained GANs. Our code is available at: https://github.com/yaxingwang/MineGAN.

Yaxing Wang, Abel Gonzalez-Garcia, David Berga, Luis Herranz, Fahad Shahbaz Khan, Joost van de Weijer• 2019

Related benchmarks

TaskDatasetResultRank
Image GenerationPanda 100-shot (train)
FID14.84
28
Image GenerationObama 100-shot (train)
FID50.63
28
Image GenerationGrumpy cat 100-shot (train)
FID34.54
28
Few-shot Image GenerationObama 100-shot
FID50.63
26
Few-shot Image GenerationGrumpy Cat 100-shot
FID35.54
26
Few-shot Image GenerationTarget domain (test)
FID58.69
20
Image GenerationAnimalFace Cat standard (train)
FID54.45
20
Image GenerationAnimalFace Dog standard (train)
FID93.03
20
Few-shot GAN adaptationFaces to Van Gogh's paintings (train/test)
Inception Score2.52
10
Few-shot GAN adaptationFaces to Sketches (train test)
IS2.12
10
Showing 10 of 35 rows

Other info

Follow for update