English

Selective inference for k-means clustering

Methodology 2022-03-30 v1 Machine Learning

Abstract

We consider the problem of testing for a difference in means between clusters of observations identified via k-means clustering. In this setting, classical hypothesis tests lead to an inflated Type I error rate. To overcome this problem, we take a selective inference approach. We propose a finite-sample p-value that controls the selective Type I error for a test of the difference in means between a pair of clusters obtained using k-means clustering, and show that it can be efficiently computed. We apply our proposal in simulation, and on hand-written digits data and single-cell RNA-sequencing data.

Keywords

Cite

@article{arxiv.2203.15267,
  title  = {Selective inference for k-means clustering},
  author = {Yiqun T. Chen and Daniela M. Witten},
  journal= {arXiv preprint arXiv:2203.15267},
  year   = {2022}
}
R2 v1 2026-06-24T10:29:31.046Z