Lawrence Stewart

Differentiable Clustering with Perturbed Spanning Forests

2023 · arXiv (Cornell University)

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 …