Computational efficient Variational Bayesian Gaussian Mixture Models via Coreset

Abstract

Variational Bayesian Gaussian Mixture Model is a popular clustering algorithm with a reliable performance. However, it is noted that the model fitting process takes long time, especially when dealing with large scale data, since it utilizes the whole dataset. To address this issue, in paper we propose a new algorithm termed a weighted VBGMM via Coreset. Specifically, a new coreset construction method is first proposed to sample the data which is used to fit the model. To evaluate the algorithm, two datasets are used: 1) six rat kidney images datasets 2) three human kidney images datasets. The results show that our proposed algorithm is much faster (~ 20 times) comparing to classic VBGMM while maintaining the similar performance on whole dataset.