Combining Mixture Components for Clustering

Jean-Patrick Baudry, Adrian E. Raftery, Gilles Celeux, Kenneth Lo, Raphaël Gottardo

Journal of Computational and Graphical Statistics · 2010 · 339 citations · 21 references

DOIFull text

Open access

Concepts

TL;DR

Model‑based clustering fits a mixture of Gaussian components to data, but BIC often overestimates cluster numbers because it selects components that best approximate the density rather than true clusters. The authors propose selecting the total number of Gaussian components with BIC and then hierarchically combining them using an entropy criterion to produce a unique soft clustering for each cluster count up to K, with an automatic selection rule via piecewise linear regression on the entropy plot. Their method selects K via BIC, merges components hierarchically by entropy, yields soft clusterings for all cluster counts ≤K, and is illustrated on simulated data and a flow cytometry dataset. Supplemental materials are available on the journal website.

Abstract

Model-based clustering consists of fitting a mixture model to data and identifying each cluster with one of its components. Multivariate normal distributions are typically used. The number of clusters is usually determined from the data, often using BIC. In practice, however, individual clusters can be poorly fitted by Gaussian distributions, and in that case model-based clustering tends to represent one non-Gaussian cluster by a mixture of two or more Gaussian distributions. If the number of mixture components is interpreted as the number of clusters, this can lead to overestimation of the number of clusters. This is because BIC selects the number of mixture components needed to provide a good approximation to the density, rather than the number of clusters as such. We propose first selecting the total number of Gaussian mixture components, K, using BIC and then combining them hierarchically according to an entropy criterion. This yields a unique soft clustering for each number of clusters less than or equal to K. These clusterings can be compared on substantive grounds, and we also describe an automatic way of selecting the number of clusters via a piecewise linear regression fit to the rescaled entropy plot. We illustrate the method with simulated data and a flow cytometry dataset. Supplemental Materials are available on the journal Web site and described at the end of the paper.

References

21