ClusterGAN : Latent Space Clustering in Generative Adversarial Networks

  • 2018-09-10 23:00:37
  • Sudipto Mukherjee, Himanshu Asnani, Eugene Lin, Sreeram Kannan
  • 41

Abstract

Generative Adversarial networks (GANs) have obtained remarkable success inmany unsupervised learning tasks and unarguably, clustering is an importantunsupervised learning problem. While one can potentially exploit thelatent-space back-projection in GANs to cluster, we demonstrate that thecluster structure is not retained in the GAN latent space. In this paper, we propose ClusterGAN as a new mechanism for clustering usingGANs. By sampling latent variables from a mixture of one-hot encoded variablesand continuous latent variables, coupled with an inverse network (whichprojects the data to the latent space) trained jointly with a clusteringspecific loss, we are able to achieve clustering in the latent space. Ourresults show a remarkable phenomenon that GANs can preserve latent spaceinterpolation across categories, even though the discriminator is never exposedto such vectors. We compare our results with various clustering baselines anddemonstrate superior performance on both synthetic and real datasets.

 

Quick Read (beta)

loading the full paper ...