Model-based Clustering with Sparse Covariance Matrices

Michael Fop,Thomas Brendan Murphy,Luca Scrucca
DOI: https://doi.org/10.48550/arXiv.1711.07748
2018-09-23
Abstract:Finite Gaussian mixture models are widely used for model-based clustering of continuous data. Nevertheless, since the number of model parameters scales quadratically with the number of variables, these models can be easily over-parameterized. For this reason, parsimonious models have been developed via covariance matrix decompositions or assuming local independence. However, these remedies do not allow for direct estimation of sparse covariance matrices nor do they take into account that the structure of association among the variables can vary from one cluster to the other. To this end, we introduce mixtures of Gaussian covariance graph models for model-based clustering with sparse covariance matrices. A penalized likelihood approach is employed for estimation and a general penalty term on the graph configurations can be used to induce different levels of sparsity and incorporate prior knowledge. Model estimation is carried out using a structural-EM algorithm for parameters and graph structure estimation, where two alternative strategies based on a genetic algorithm and an efficient stepwise search are proposed for inference. With this approach, sparse component covariance matrices are directly obtained. The framework results in a parsimonious model-based clustering of the data via a flexible model for the within-group joint distribution of the variables. Extensive simulated data experiments and application to illustrative datasets show that the method attains good classification performance and model quality.
Methodology,Computation
What problem does this paper attempt to address?
The problem that this paper attempts to solve is how to effectively handle the complex correlation structures of high - dimensional continuous data in model - based clustering, especially when the correlation structures may be different among different clusters. Although the traditional finite Gaussian mixture model is widely used in model - based clustering, the number of its parameters grows quadratically with the increase in the number of variables, which easily leads to over - parameterization. In addition, the existing simplified model methods either do not allow the direct estimation of sparse covariance matrices or do not take into account the differences in variable association structures among different clusters. To solve these problems, the paper introduces a mixture model based on the Gaussian covariance graph model for model - based clustering and is able to directly estimate sparse covariance matrices. This method represents the association structures between variables by using a graphical model within each cluster, thus allowing different clusters to have different association patterns. Specifically, the paper adopts the maximum - likelihood estimation method with a penalty term to estimate the model parameters and the graphical structure, where the penalty term can be used to induce different degrees of sparsity and can incorporate prior knowledge. The model estimation adopts a structured EM algorithm, combined with a genetic algorithm or a step - by - step search strategy to estimate the parameters and the graphical structure. In general, the main contribution of the paper lies in providing a new model - based clustering method, which can not only handle the complex correlation structures in high - dimensional data, but also flexibly adapt to the differences in association patterns among different clusters, thereby improving clustering performance and model quality.