Estimating the number of clusters using diversity

SK Kingrani, M Levene, D Zhang - Artificial Intelligence Research, 2018 - eprints.bbk.ac.uk
SK Kingrani, M Levene, D Zhang
Artificial Intelligence Research, 2018eprints.bbk.ac.uk
It is an important and challenging problem in unsupervised learning to estimate the number
of clusters in a dataset. Knowing the number of clusters is a prerequisite for many commonly
used clustering algorithms such as k-means. In this paper, we propose a novel diversity
based approach to this problem. Specifically, we show that the difference between the
global diversity of clusters and the sum of each cluster's local diversity of their members can
be used as an effective indicator of the optimality of the number of clusters, where the …
It is an important and challenging problem in unsupervised learning to estimate the number of clusters in a dataset. Knowing the number of clusters is a prerequisite for many commonly used clustering algorithms such as k-means. In this paper, we propose a novel diversity based approach to this problem. Specifically, we show that the difference between the global diversity of clusters and the sum of each cluster's local diversity of their members can be used as an effective indicator of the optimality of the number of clusters, where the diversity is measured by Rao's quadratic entropy. A notable advantage of our proposed method is that it encourages balanced clustering by taking into account both the sizes of clusters and the distances between clusters. In other words, it is less prone to very small "outlier" clusters than existing methods. Our extensive experiments on both synthetic and real-world datasets (with known ground-truth clustering) have demonstrated that our proposed method is robust for clusters of different sizes, variances, and shapes, and it is more accurate than existing methods (including elbow, Calinski-Harabasz, silhouette, and gap-statistic) in terms of finding out the optimal number of clusters.
eprints.bbk.ac.uk
Showing the best result for this search. See all results