English

An efficient $k$-means-type algorithm for clustering datasets with incomplete records

Machine Learning 2018-09-11 v2 High Energy Astrophysical Phenomena Machine Learning Computation Methodology

Abstract

The kk-means algorithm is arguably the most popular nonparametric clustering method but cannot generally be applied to datasets with incomplete records. The usual practice then is to either impute missing values under an assumed missing-completely-at-random mechanism or to ignore the incomplete records, and apply the algorithm on the resulting dataset. We develop an efficient version of the kk-means algorithm that allows for clustering in the presence of incomplete records. Our extension is called kmk_m-means and reduces to the kk-means algorithm when all records are complete. We also provide initialization strategies for our algorithm and methods to estimate the number of groups in the dataset. Illustrations and simulations demonstrate the efficacy of our approach in a variety of settings and patterns of missing data. Our methods are also applied to the analysis of activation images obtained from a functional Magnetic Resonance Imaging experiment.

Keywords

Cite

@article{arxiv.1802.08363,
  title  = {An efficient $k$-means-type algorithm for clustering datasets with incomplete records},
  author = {Andrew Lithio and Ranjan Maitra},
  journal= {arXiv preprint arXiv:1802.08363},
  year   = {2018}
}

Comments

21 pages, 12 figures, 3 tables, in press, Statistical Analysis and Data Mining -- The ASA Data Science Journal, 2018

R2 v1 2026-06-23T00:30:56.885Z