Differentiable Clustering with Perturbed Spanning Forests - Archive ouverte HAL Accéder directement au contenu
Communication Dans Un Congrès Année : 2023

Differentiable Clustering with Perturbed Spanning Forests

Lawrence Stewart
  • Fonction : Auteur
  • PersonId : 1184478
Francis S Bach
  • Fonction : Auteur
  • PersonId : 863086
Quentin Berthet
  • Fonction : Auteur

Résumé

We introduce a differentiable clustering method based on stochastic perturbations of minimum-weight spanning forests. This allows us to include clustering in end-to-end trainable pipelines, with efficient gradients. We show that our method performs well even in difficult settings, such as data sets with high noise and challenging geometries. We also formulate an ad hoc loss to efficiently learn from partial clustering data using this operation. We demonstrate its performance on several data sets for supervised and semi-supervised tasks.
Fichier principal
Vignette du fichier
neurips_2023.pdf (1.31 Mo) Télécharger le fichier
Origine : Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04105310 , version 1 (24-05-2023)
hal-04105310 , version 2 (30-05-2023)
hal-04105310 , version 3 (05-11-2023)

Identifiants

Citer

Lawrence Stewart, Francis S Bach, Felipe Llinares López, Quentin Berthet. Differentiable Clustering with Perturbed Spanning Forests. 37th Conference on Neural Information Processing Systems, Dec 2023, New Orleans, United States. ⟨hal-04105310v3⟩
112 Consultations
46 Téléchargements

Altmetric

Partager

Gmail Facebook X LinkedIn More