English

Hierarchical clustering: visualization, feature importance and model selection

Methodology 2023-01-31 v2 Machine Learning

Abstract

We propose methods for the analysis of hierarchical clustering that fully use the multi-resolution structure provided by a dendrogram. Specifically, we propose a loss for choosing between clustering methods, a feature importance score and a graphical tool for visualizing the segmentation of features in a dendrogram. Current approaches to these tasks lead to loss of information since they require the user to generate a single partition of the instances by cutting the dendrogram at a specified level. Our proposed methods, instead, use the full structure of the dendrogram. The key insight behind the proposed methods is to view a dendrogram as a phylogeny. This analogy permits the assignment of a feature value to each internal node of a tree through an evolutionary model. Real and simulated datasets provide evidence that our proposed framework has desirable outcomes and gives more insights than state-of-art approaches. We provide an R package that implements our methods.

Keywords

Cite

@article{arxiv.2112.01372,
  title  = {Hierarchical clustering: visualization, feature importance and model selection},
  author = {Luben M. C. Cabezas and Rafael Izbicki and Rafael B. Stern},
  journal= {arXiv preprint arXiv:2112.01372},
  year   = {2023}
}

Comments

29 pages, 9 figures, 3 tables

R2 v1 2026-06-24T08:01:54.296Z