Goto

Collaborating Authors

 improved input reprogramming


Improved Input Reprogramming for GAN Conditioning

Dinh, Tuan, Seo, Daewon, Du, Zhixu, Shang, Liang, Lee, Kangwook

arXiv.org Artificial Intelligence

We study the GAN conditioning problem, whose goal is to convert a pretrained unconditional GAN into a conditional GAN using labeled data. We first identify and analyze three approaches to this problem -- conditional GAN training from scratch, fine-tuning, and input reprogramming. Our analysis reveals that when the amount of labeled data is small, input reprogramming performs the best. Motivated by real-world scenarios with scarce labeled data, we focus on the input reprogramming approach and carefully analyze the existing algorithm. After identifying a few critical issues of the previous input reprogramming approach, we propose a new algorithm called InRep+. Our algorithm InRep+ addresses the existing issues with the novel uses of invertible neural networks and Positive-Unlabeled (PU) learning. Via extensive experiments, we show that InRep+ outperforms all existing methods, particularly when label information is scarce, noisy, and/or imbalanced. For instance, for the task of conditioning a CIFAR10 GAN with 1% labeled data, InRep+ achieves an average Intra-FID of 82.13, whereas the second-best method achieves 114.51.

  dataset, generator, improved input reprogramming, (12 more...)
2201.02692
  Country:
  Genre: Research Report (1.00)
  Industry: Information Technology (0.46)