Hierarchical clustering beyond the worst-case

Conference item

Hiererachical clustering, that is computing a recursive partitioning of a dataset to obtain clusters at increasingly finer granularity is a fundamental problem in data analysis. Although hierarchical clustering has mostly been studied through procedures such as linkage algorithms, or top-down heuristics, rather than as optimization problems, recently Dasgupta proposed an objective function for hierarchical clustering and initiated a line of work developing algorithms that explicitly optimize an objective. In this paper, we consider a fairly general random graph model for hierarchical clustering, called the hierarchical stochastic block model (HSBM), and show that in certain regimes the SVD approach of McSherry combined with specific linkage methods results in a clustering that give an Op1q approximation to Dasgupta’s cost function. We also show that an approach based on SDP relaxations for balanced cuts based on the work of Makarychev et al., combined with the recursive sparsest cut algorithm of Dasgupta, yields an Op1q approximation in slightly larger regimes and also in the semi-random setting, where an adversary may remove edges from the random graph generated according to an HSBM. Finally, we report empirical evaluation on synthetic and real-world data showing that our proposed SVD-based method does indeed achieve a better cost than other widely-used heurstics and also results in a better classification accuracy when the underlying problem was that of multi-class classification.

Authors: Cohen-Addad, V; Kanade, V; Mallmann-Trenn, F

• Publication status: Published
• Peer review status: Peer reviewed
• Publisher: Neural Information Processing Systems
• Host title: 31st Conference on Neural Information Processing Systems (NIPS 2017)
• Journal: 31st Conference on Neural Information Processing Systems (NIPS 2017)
• Publication date: 2018-07-01
• Acceptance date: 2017-09-04