James-Stein shrinkage to improve k-means cluster analysis

We study a general algorithm to improve accuracy in cluster analysis that employs the James-Stein shrinkage effect in k-means clustering. We shrink the centroids of clusters toward the overall mean of all data using a James-Stein-type adjustment, and then the James-Stein shrinkage estimators act as the new centroids in the next clustering iteration until convergence. We compare the shrinkage results to the traditional k-means method. Monte Carlo simulation shows that the magnitude of the improvement depends on the within-cluster variance and especially on the effective dimension of the covariance matrix. Using the Rand index, we demonstrate that accuracy increases significantly in simulated data and in a real data example.

• Type: Journal
• Year: 2010
• Where: CSDA
• Authors: Jinxin Gao, David B. Hitchcock
• Keywords: CSDA 2010, James-Stein Shrinkage, James-Stein Shrinkage Effect, James-Stein Shrinkage Estimators