Alert button

Collapse by Conditioning: Training Class-conditional GANs with Limited Data

Jan 17, 2022
Mohamad Shahbazi, Martin Danelljan, Danda Pani Paudel, Luc Van Gool

Figure 1 for Collapse by Conditioning: Training Class-conditional GANs with Limited Data
Figure 2 for Collapse by Conditioning: Training Class-conditional GANs with Limited Data
Figure 3 for Collapse by Conditioning: Training Class-conditional GANs with Limited Data
Figure 4 for Collapse by Conditioning: Training Class-conditional GANs with Limited Data

Share this with someone who'll enjoy it:

Class-conditioning offers a direct means of controlling a Generative Adversarial Network (GAN) based on a discrete input variable. While necessary in many applications, the additional information provided by the class labels could even be expected to benefit the training of the GAN itself. Contrary to this belief, we observe that class-conditioning causes mode collapse in limited data settings, where unconditional learning leads to satisfactory generative ability. Motivated by this observation, we propose a training strategy for conditional GANs (cGANs) that effectively prevents the observed mode-collapse by leveraging unconditional learning. Our training strategy starts with an unconditional GAN and gradually injects conditional information into the generator and the objective function. The proposed method for training cGANs with limited data results not only in stable training but also in generating high-quality images, thanks to the early-stage exploitation of the shared information across classes. We analyze the aforementioned mode collapse problem in comprehensive experiments on four datasets. Our approach demonstrates outstanding results compared with state-of-the-art methods and established baselines. The code is available at: https://github.com/mshahbazi72/transitional-cGAN

View paper onarxiv iconopen_review iconOpenReview

Share this with someone who'll enjoy it: