2023
Differentiable Clustering with Perturbed Spanning Forests
NeurIPS 2023poster
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…