arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.02692cs.LGcs.AI

Improved Input Reprogramming for GAN Conditioning

University of Wisconsin-Madison(威斯康星大学麦迪逊分校);Daegu Gyeongbuk Institute of Science and Technology(大邱庆北科学技术院);University of Hong Kong(香港大学)

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

英文摘要:

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 76.24, whereas the second-best method achieves 114.51.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑