English

Fixed-sized clusters $k$-Means

Machine Learning 2025-01-28 v1

Abstract

We present a kk-means-based clustering algorithm, which optimizes the mean square error, for given cluster sizes. A straightforward application is balanced clustering, where the sizes of each cluster are equal. In the kk-means assignment phase, the algorithm solves an assignment problem using the Hungarian algorithm. This makes the assignment phase time complexity O(n3)O(n^3). This enables clustering of datasets of size more than 5000 points.

Keywords

Cite

@article{arxiv.2501.16113,
  title  = {Fixed-sized clusters $k$-Means},
  author = {Mikko I. Malinen and Pasi Fränti},
  journal= {arXiv preprint arXiv:2501.16113},
  year   = {2025}
}

Comments

7 pages, 2 figures

R2 v1 2026-06-28T21:19:46.851Z