Skip to main content

Dimension-Grouped Mixed Membership Models for Multivariate Categorical Data.

Publication ,  Journal Article
Gu, Y; Erosheva, EA; Xu, G; Dunson, DB
Published in: Journal of machine learning research : JMLR
February 2023

Mixed Membership Models (MMMs) are a popular family of latent structure models for complex multivariate data. Instead of forcing each subject to belong to a single cluster, MMMs incorporate a vector of subject-specific weights characterizing partial membership across clusters. With this flexibility come challenges in uniquely identifying, estimating, and interpreting the parameters. In this article, we propose a new class of Dimension-Grouped MMMs ( Gro-M3s ) for multivariate categorical data, which improve parsimony and interpretability. In Gro-M3s , observed variables are partitioned into groups such that the latent membership is constant for variables within a group but can differ across groups. Traditional latent class models are obtained when all variables are in one group, while traditional MMMs are obtained when each variable is in its own group. The new model corresponds to a novel decomposition of probability tensors. Theoretically, we derive transparent identifiability conditions for both the unknown grouping structure and model parameters in general settings. Methodologically, we propose a Bayesian approach for Dirichlet Gro-M3s to inferring the variable grouping structure and estimating model parameters. Simulation results demonstrate good computational performance and empirically confirm the identifiability results. We illustrate the new methodology through applications to a functional disability survey dataset and a personality test dataset.

Duke Scholars

Published In

Journal of machine learning research : JMLR

DOI

EISSN

1533-7928

ISSN

1532-4435

Publication Date

February 2023

Volume

24

Start / End Page

88

Related Subject Headings

  • Artificial Intelligence & Image Processing
  • 4905 Statistics
  • 4611 Machine learning
  • 17 Psychology and Cognitive Sciences
  • 08 Information and Computing Sciences
 

Citation

APA
Chicago
ICMJE
MLA
NLM
Gu, Y., Erosheva, E. A., Xu, G., & Dunson, D. B. (2023). Dimension-Grouped Mixed Membership Models for Multivariate Categorical Data. Journal of Machine Learning Research : JMLR, 24, 88. https://doi.org/10.48550/arxiv.2109.11705
Gu, Yuqi, Elena A. Erosheva, Gongjun Xu, and David B. Dunson. “Dimension-Grouped Mixed Membership Models for Multivariate Categorical Data.Journal of Machine Learning Research : JMLR 24 (February 2023): 88. https://doi.org/10.48550/arxiv.2109.11705.
Gu Y, Erosheva EA, Xu G, Dunson DB. Dimension-Grouped Mixed Membership Models for Multivariate Categorical Data. Journal of machine learning research : JMLR. 2023 Feb;24:88.
Gu, Yuqi, et al. “Dimension-Grouped Mixed Membership Models for Multivariate Categorical Data.Journal of Machine Learning Research : JMLR, vol. 24, Feb. 2023, p. 88. Epmc, doi:10.48550/arxiv.2109.11705.
Gu Y, Erosheva EA, Xu G, Dunson DB. Dimension-Grouped Mixed Membership Models for Multivariate Categorical Data. Journal of machine learning research : JMLR. 2023 Feb;24:88.

Published In

Journal of machine learning research : JMLR

DOI

EISSN

1533-7928

ISSN

1532-4435

Publication Date

February 2023

Volume

24

Start / End Page

88

Related Subject Headings

  • Artificial Intelligence & Image Processing
  • 4905 Statistics
  • 4611 Machine learning
  • 17 Psychology and Cognitive Sciences
  • 08 Information and Computing Sciences