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.

Combating Mode Collapse in GANs via Manifold Entropy Estimation

Combating Mode Collapse via Offline Manifold Entropy Estimation

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

Dual Distribution Matching GAN

MGGAN: Solving Mode Collapse using Manifold Guided Training

On Mode Collapse in Generative Adversarial Networks

Distribution Fitting for Combating Mode Collapse in GANs

Antimode collapse generative adversarial networks

BourGAN: Generative Networks with Metric Embeddings

Towards the Gradient Vanishing, Divergence Mismatching and Mode Collapse of Generative Adversarial Nets

GANs with Variational Entropy Regularizers: Applications in Mitigating the Mode-Collapse Issue

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

Alleviating Mode Collapse in GAN via Diversity Penalty Module

BEGAN v3: Avoiding Mode Collapse in GANs Using Variational Inference

Testing Generated Distributions in GANs to Penalize Mode Collapse.

Distribution Fitting for Combating Mode Collapse in Generative Adversarial Networks

Escaping from Collapsing Modes in a Constrained Space

Multi-generator GAN learning disconnected manifolds with mutual information

Mode Collapse and Regularity of Optimal Transportation Maps

Reducing Mode Collapse With Monge–Kantorovich Optimal Transport for Generative Adversarial Networks