Epistemic Generative Adversarial Networks

Abstract

Generative models, particularly Generative Adversarial Networks (GANs), often suffer from a lack of output diversity, frequently generating similar samples rather than a wide range of variations. This paper introduces a novel generalization of the GAN loss function based on Dempster-Shafer theory of evidence, applied to both the generator and discriminator. Additionally, we propose an architectural enhancement to the generator that enables it to predict a mass function for each image pixel. This modification allows the model to quantify uncertainty in its outputs and leverage this uncertainty to produce more diverse and representative generations. Experimental evidence shows that our approach not only improves generation variability but also provides a principled framework for modeling and interpreting uncertainty in generative processes.

Figures (6)

• Figure 1: Belief functions and mass functions
• Figure 2: Representation of a belief function with Borel intervals
• Figure 3: Discriminator architecture comparison.
• Figure 4: Generator Architecture and flow for Epistemic GANs.
• Figure 5: Generations for Standard GAN (left) and Epistemic GAN(right) for Celeb-A dataset.