Differentiable Clustering with Perturbed Spanning Forests

Lawrence Stewart, Francis S Bach, Felipe Llinares López +1

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 p…