Searching Towards Class-Aware Generators for Conditional Generative Adversarial Networks

Peng Zhou,Lingxi Xie,Bingbing Ni,Qi Tian
DOI: https://doi.org/10.1109/lsp.2022.3193589
2022-01-01
IEEE Signal Processing Letters
Abstract:Conditional generative adversarial networks (cGANs) are designed to generate images based on the provided conditions, e.g ., class-level distributions, semantic label maps, etc . Existing methods have used the same generator architecture for all classes. This paper presents an idea that adopts neural architecture search (NAS) to find a class-aware architecture for each class. The search space contains regular and class-modulated convolutions, where the latter is designed to introduce class-specific information while avoiding the reduction of training data for each class generator. The search algorithm follows a weight-sharing pipeline with mixed-architecture optimization so that the search cost does not grow with the number of classes. To learn the sampling policy, a Markov decision process is embedded into the search algorithm, and a moving average is applied for better stability. Class-aware generators show advantages over class-agnostic architectures experimentally. Moreover, we discover two intriguing phenomena that are inspirational to craft cGANs by hand.
What problem does this paper attempt to address?