Variational Information Bottleneck for Unsupervised Clustering: Deep Gaussian Mixture Embedding
Authors: Yigit Ugur, George Arvanitakis, Abdellatif Zaidi
Keywords: clustering, Variational Information Bottleneck, Gaussian Mixture Model
Abstract: In this paper, we develop an unsupervised generative clustering framework that combines variational information bottleneck and the Gaussian Mixture Model. Specifically, in our approach we use the variational information bottleneck method and model the latent space as a mixture of Gaussians. We derive a bound on the cost function of our model that generalizes the evidence lower bound (ELBO); and provide a variational inference type algorithm that allows to compute it. In the algorithm, the coders’ mappings are parametrized using neural networks and the bound is approximated by Markov sampling and optimized with stochastic gradient descent. Numerical results on real datasets are provided to support the efficiency of our method.