Hierarchical clustering with dot products recovers hidden tree structure
Annie Gray, Alexander Modell, Patrick Rubin-Delanchy, Nick Whiteley
Abstract
In this paper we offer a new perspective on the well established agglomerative clustering algorithm, focusing on recovery of hierarchical structure. We recommend a simple variant of the standard algorithm, in which clusters are merged by maximum average dot product and not, for example, by minimum distance or within-cluster variance. We demonstrate that the tree output by this algorithm provides a bona fide estimate of generative hierarchical structure in data, under a generic probabilistic graphical model. The key technical innovations are to understand how hierarchical information in this model translates into tree geometry which can be recovered from data, and to characterise the benefits of simultaneously growing sample size and data dimension. We demonstrate superior tree recovery performance with real data over existing approaches such as UPGMA, Ward's method, and HDBSCAN.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 1ea61c15-77cb-4ec4-8392-8a065a765453Cited by top-tier papers1
Ask how each one uses itBuilds on2
Related papers
- Hierarchical Agglomerative Graph Clustering in Nearly-Linear TimeLaxman Dhulipala, David Eisenstat, Jakub Lacki, Vahab S. Mirrokni et al.ICML 2021 · 30 citations
- Expected Probabilistic HierarchiesMarcel Kollovieh, Bertrand Charpentier, Daniel Zügner, Stephan GünnemannNeurIPS 2024 · 4 citations
- Fast Parallel Algorithms for Euclidean Minimum Spanning Tree and Hierarchical Spatial ClusteringYiqiu Wang, Shangdi Yu, Yan Gu, Julian ShunSIGMOD 2021 · 34 citations
- Top-Down Deep Clustering with Multi-Generator GANsDaniel P. M. de Mello, Renato M. Assunção, Fabricio MuraiAAAI 2022 · 22 citations
- End-to-End Learning of Probabilistic Hierarchies on GraphsDaniel Zügner, Bertrand Charpentier, Morgane Ayle, Sascha Geringer et al.ICLR 2022 · 4 citations
