Generalised Mutual Information: a Framework for Discriminative Clustering - Inria - Institut national de recherche en sciences et technologies du numérique Accéder directement au contenu
Pré-Publication, Document De Travail Année : 2023

Generalised Mutual Information: a Framework for Discriminative Clustering

Résumé

In the last decade, recent successes in deep clustering majorly involved the Mutual Information (MI) as an unsupervised objective for training neural networks with increasing regularisations. While the quality of the regularisations have been largely discussed for improvements, little attention has been dedicated to the relevance of MI as a clustering objective. In this paper, we first highlight how the maximisation of MI does not lead to satisfying clusters. We identified the Kullback-Leibler divergence as the main reason of this behaviour. Hence, we generalise the mutual information by changing its core distance, introducing the Generalised Mutual Information (GEMINI): a set of metrics for unsupervised neural network training. Unlike MI, some GEMINIs do not require regularisations when training as they are geometry-aware thanks to distances or kernels in the data space. Finally, we highlight that GEMINIs can automatically select a relevant number of clusters, a property that has been little studied in deep discriminative clustering context where the number of clusters is a priori unknown.
Fichier principal
Vignette du fichier
Generalised Mutual Information a Framework for Discriminative Clustering.pdf (5.55 Mo) Télécharger le fichier
Origine : Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04198445 , version 1 (07-09-2023)

Identifiants

Citer

Louis Ohl, Pierre-Alexandre Mattei, Charles Bouveyron, Warith Harchaoui, Mickaël Leclercq, et al.. Generalised Mutual Information: a Framework for Discriminative Clustering. 2023. ⟨hal-04198445⟩
53 Consultations
26 Téléchargements

Altmetric

Partager

Gmail Facebook X LinkedIn More