We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

stat.ML

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Statistics > Machine Learning

Title: $k$-means as a variational EM approximation of Gaussian mixture models

Abstract: We show that $k$-means (Lloyd's algorithm) is obtained as a special case when truncated variational EM approximations are applied to Gaussian Mixture Models (GMM) with isotropic Gaussians. In contrast to the standard way to relate $k$-means and GMMs, the provided derivation shows that it is not required to consider Gaussians with small variances or the limit case of zero variances. There are a number of consequences that directly follow from our approach: (A) $k$-means can be shown to increase a free energy associated with truncated distributions and this free energy can directly be reformulated in terms of the $k$-means objective; (B) $k$-means generalizations can directly be derived by considering the 2nd closest, 3rd closest etc. cluster in addition to just the closest one; and (C) the embedding of $k$-means into a free energy framework allows for theoretical interpretations of other $k$-means generalizations in the literature. In general, truncated variational EM provides a natural and rigorous quantitative link between $k$-means-like clustering and GMM clustering algorithms which may be very relevant for future theoretical and empirical studies.
Subjects: Machine Learning (stat.ML)
MSC classes: 62H30
Journal reference: Pattern Recognition Letters, 125: 349-356, 2019
DOI: 10.1016/j.patrec.2019.04.001
Cite as: arXiv:1704.04812 [stat.ML]
  (or arXiv:1704.04812v5 [stat.ML] for this version)

Submission history

From: Dennis Forster [view email]
[v1] Sun, 16 Apr 2017 20:06:30 GMT (171kb,D)
[v2] Fri, 20 Oct 2017 14:30:43 GMT (172kb,D)
[v3] Fri, 20 Jul 2018 11:15:47 GMT (171kb,D)
[v4] Wed, 24 Apr 2019 18:18:27 GMT (858kb,D)
[v5] Thu, 6 Jun 2019 09:37:48 GMT (858kb,D)

Link back to: arXiv, form interface, contact.