Intelligent Choice of the Number of Clusters in K-Means Clustering: An Experimental Study with Different Cluster Spreads

被引:192
作者
Chiang, Mark Ming-Tso [1 ]
Mirkin, Boris [1 ,2 ]
机构
[1] Birkbeck Univ London, Dept Comp Sci & Informat Syst, London, England
[2] State Univ, Higher Sch Econ, Moscow, Russia
关键词
K-Means clustering; Number of clusters; Anomalous pattern; Hartigan's rule; Gap statistic;
D O I
10.1007/s00357-010-9049-5
中图分类号
O1 [数学];
学科分类号
0701 ; 070101 ;
摘要
The issue of determining "the right number of clusters" in K-Means has attracted considerable interest, especially in the recent years. Cluster intermix appears to be a factor most affecting the clustering results. This paper proposes an experimental setting for comparison of different approaches at data generated from Gaussian clusters with the controlled parameters of between- and within-cluster spread to model cluster intermix. The setting allows for evaluating the centroid recovery on par with conventional evaluation of the cluster recovery. The subjects of our interest are two versions of the "intelligent" K-Means method, ik-Means, that find the "right" number of clusters by extracting "anomalous patterns" from the data one-by-one. We compare them with seven other methods, including Hartigan's rule, averaged Silhouette width and Gap statistic, under different between- and within-cluster spread-shape conditions. There are several consistent patterns in the results of our experiments, such as that the right K is reproduced best by Hartigan's rule - but not clusters or their centroids. This leads us to propose an adjusted version of iK-Means, which performs well in the current experiment setting.
引用
收藏
页码:3 / 40
页数:38
相关论文
共 57 条
[51]   Finding the number of clusters in a dataset: An information-theoretic approach [J].
Sugar, CA ;
James, GM .
JOURNAL OF THE AMERICAN STATISTICAL ASSOCIATION, 2003, 98 (463) :750-763
[52]   Estimating the number of clusters in a data set via the gap statistic [J].
Tibshirani, R ;
Walther, G ;
Hastie, T .
JOURNAL OF THE ROYAL STATISTICAL SOCIETY SERIES B-STATISTICAL METHODOLOGY, 2001, 63 :411-423
[53]   Probabilistic principal component analysis [J].
Tipping, ME ;
Bishop, CM .
JOURNAL OF THE ROYAL STATISTICAL SOCIETY SERIES B-STATISTICAL METHODOLOGY, 1999, 61 :611-622
[54]  
Vapnik V.N., 1982, Estimation of dependences based on empirical data
[55]   Nearest neighbours in least-squares data imputation algorithms with different missing patterns [J].
Wasito, I ;
Mirkin, B .
COMPUTATIONAL STATISTICS & DATA ANALYSIS, 2006, 50 (04) :926-949
[56]   Principal component analysis for clustering gene expression data [J].
Yeung, KY ;
Ruzzo, WL .
BIOINFORMATICS, 2001, 17 (09) :763-774
[57]  
2006, NETLAB NEURAL NETWOR