$K$-Means and Gaussian Mixture Modeling with a Separation Constraint
Computation
2023-01-24 v2
Abstract
We consider the problem of clustering with -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 -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.
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