English

$K$-Means and Gaussian Mixture Modeling with a Separation Constraint

Computation 2023-01-24 v2

Abstract

We consider the problem of clustering with KK-means and Gaussian mixture models with a constraint on the separation between the centers in the context of real-valued data. We first propose a dynamic programming approach to solving the KK-means problem with a separation constraint on the centers, building on (Wang and Song, 2011). In the context of fitting a Gaussian mixture model, we then propose an EM algorithm that incorporates such a constraint. A separation constraint can help regularize the output of a clustering algorithm, and we provide both simulated and real data examples to illustrate this point.

Keywords

Cite

@article{arxiv.2007.04586,
  title  = {$K$-Means and Gaussian Mixture Modeling with a Separation Constraint},
  author = {He Jiang and Ery Arias-Castro},
  journal= {arXiv preprint arXiv:2007.04586},
  year   = {2023}
}

Comments

16 pages, 6 tables, 1 figure with 3 subfigures

R2 v1 2026-06-23T16:58:29.028Z