Abstract:Original data usually lies on a set of disconnected manifolds rather than a smooth connected manifold. This causes the problem of mode collapse in the training of vanilla Generative Adversarial Network (GAN). There are many existing GAN variants that attempt to address this problem, but they result in limitations. The existing variants either produce simulated instances with low quality or generate identical simulated instances. In this study, we propose a new approach to training GAN utilizing multiple generators, a classifier and a discriminator to address mode collapse. The classifier outputs the statistical probabilities of generated data belonging to a specific category. These probabilities implicitly reflect which manifolds are captured by generators, and the correlation between generators is quantified by mutual information. Our idea views the mutual information values as a constraint to guide generators in learning different manifolds. Specifically, we traverse the generators, calculating the mutual information between each generator and the others. The calculated values are integrated into the generator loss to form a new generator loss and to update the corresponding generator's parameters, using back-propagation. We minimize the mutual information to reduce the correlation between generators while also minimizing the generator loss. This ensures generators capture different manifolds while updating their parameters. A new minimax formula is established to train our approach in a similar spirit to vanilla GAN. We term our approach Mutual Information Multi-generator GAN (MIM-GAN). We conduct extensive experiments utilizing the MNIST, CIFAR10 and CelebA datasets to demonstrate the significant performance improvement of MIM-GAN in both achieving the highest Inception Scores and producing diverse generated data at different resolutions.

STM-GAN: Sequentially Trained Multiple Generators for Mitigating Mode Collapse

Study of Prevention of Mode Collapse in Generative Adversarial Network (GAN)

Soft Generative Adversarial Network: Combating Mode Collapse in Generative Adversarial Network Training Via Dynamic Borderline Softening Mechanism

Overcoming Mode Collapse with Adaptive Multi Adversarial Training

Antimode collapse generative adversarial networks

TL-GAN: Generative Adversarial Networks with Transfer Learning for Mode Collapse (S).

VirtualGAN: Reducing Mode Collapse in Generative Adversarial Networks Using Virtual Mapping

MGGAN: Solving Mode Collapse using Manifold Guided Training

Combating Mode Collapse via Offline Manifold Entropy Estimation

On Mode Collapse in Generative Adversarial Networks

Tackling mode collapse in multi-generator GANs with orthogonal vectors

Text to image synthesis using multi-generator text conditioned generative adversarial networks.

BEGAN v3: Avoiding Mode Collapse in GANs Using Variational Inference

DynGAN: Solving Mode Collapse in GANs with Dynamic Clustering

Combating Mode Collapse in GANs via Manifold Entropy Estimation

Distribution constraining for combating mode collapse in generative adversarial networks

Multi-generator GAN learning disconnected manifolds with mutual information

Training Generative Adversarial Networks in One Stage

Mode Seeking Generative Adversarial Networks for Diverse Image Synthesis