Bayesian Mixture Models For Semi-Supervised Clustering - Archive ouverte HAL
Pré-Publication, Document De Travail Année : 2019

Bayesian Mixture Models For Semi-Supervised Clustering

Joachim Flocon-Cholet
  • Fonction : Auteur
  • PersonId : 772934
  • IdRef : 196443776
Stéphane Gosselin
Sandrine Vaton

Résumé

In most real-world applications of clustering, data is partially labeled by an expert. Classical clustering approaches have been extensively studied in the presence of partial labels, however little work has been done to treat the general case of Bayesian mixture models. In this paper, we propose a new approach to perform semi-supervised clustering using parametric and non parametric mixture models. We show how our approach generalizes mixture models with different types of emission distributions and priors under the same theoretical framework for semi-supervised clustering. The partial labels intervene in the clustering in the form of a Hidden Markov Random Field (HMRF) that introduces a penalty if the partial labels are not respected. We demonstrate how to perform inference in both the finite and infinite case with priors on the mixture components and the parameters using variational inference. Our experimental evaluations on synthetic data show how the method can leverage the partial labels to choose the correct clustering and the correct number of clusters. We also show that by introducing a small fraction of partial labels our method improves the clustering accuracy and outperforms a strong baseline in the literature on benchmark datasets.
Fichier principal
Vignette du fichier
Bayesian_Mixture_Models_For_SemiSupervised_Clustering.pdf (3.39 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-02372337 , version 1 (20-11-2019)

Identifiants

  • HAL Id : hal-02372337 , version 1

Citer

Amine Echraibi, Joachim Flocon-Cholet, Stéphane Gosselin, Sandrine Vaton. Bayesian Mixture Models For Semi-Supervised Clustering. 2019. ⟨hal-02372337⟩
212 Consultations
346 Téléchargements

Partager

More