Distilling Representations from GAN Generator via Squeeze and Span
Yu Yang, Xiaotian Cheng, Chang Liu, Hakan Bilen, Xiangyang Ji
Abstract
In recent years, generative adversarial networks (GANs) have been an actively studied topic and shown to successfully produce high-quality realistic images in various domains. The controllable synthesis ability of GAN generators suggests that they maintain informative, disentangled, and explainable image representations, but leveraging and transferring their representations to downstream tasks is largely unexplored. In this paper, we propose to distill knowledge from GAN generators by squeezing and spanning their representations. We squeeze the generator features into representations that are invariant to semantic-preserving transformations through a network before they are distilled into the student network. We span the distilled representation of the synthetic domain to the real domain by also using real training data to remedy the mode collapse of GANs and boost the student network performance in a real domain. Experiments justify the efficacy of our method and reveal its great significance in self-supervised representation learning. Code is available at https://github.com/yangyu12/squeeze-and-span . * Equal Contribution 2 Throughout the paper, two terms"representation network" and "student network" are used interchangeably, as are the "generator network" and "teacher network". 36th Conference on Neural Information Processing Systems (NeurIPS 2022).
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on31
- A Simple Framework for Contrastive Learning of Visual RepresentationsTing Chen, Simon Kornblith, Mohammad Norouzi, Geoffrey E. HintonICML 2020 · 24,064 citations
- Bootstrap Your Own Latent - A New Approach to Self-Supervised LearningJean-Bastien Grill, Florian Strub, Florent Altché, Corentin Tallec et al.NeurIPS 2020 · 9,171 citations
- Barlow Twins: Self-Supervised Learning via Redundancy ReductionJure Zbontar, Li Jing, Ishan Misra, Yann LeCun et al.ICML 2021 · 2,942 citations
- Training Generative Adversarial Networks with Limited DataTero Karras, Miika Aittala, Janne Hellsten, Samuli Laine et al.NeurIPS 2020 · 2,345 citations
- An Empirical Study of Training Self-Supervised Vision TransformersXinlei Chen, Saining Xie, Kaiming HeICCV 2021 · 2,340 citations
Related papers
- Distilling GANs with Style-Mixed Triplets for X2I Translation with Limited DataYaxing Wang, Joost van de Weijer, Lu Yu, Shangling JuiICLR 2022 · 2 citations
- Self-Supervised Generative Adversarial CompressionChong Yu, Jeff PoolNeurIPS 2020 · 15 citations
- Transformation GAN for Unsupervised Image Synthesis and Representation LearningJiayu Wang, Wengang Zhou, Guo-Jun Qi, Zhongqian Fu et al.CVPR 2020
- Information-Theoretic GAN Compression with Variational Energy-based ModelMinsoo Kang, Hyewon Yoo, Eunhee Kang, Sehwan Ki et al.NeurIPS 2022 · 5 citations
- Distilling Portable Generative Adversarial Networks for Image TranslationHanting Chen, Yunhe Wang, Han Shu, Changyuan Wen et al.AAAI 2020 · 89 citations
