Blind method for discovering number of clusters in multidimensional datasets by regression on linkage hierarchies generated from random data

PLoS One. 2020 Jan 23;15(1):e0227788. doi: 10.1371/journal.pone.0227788. eCollection 2020.

Abstract

Determining intrinsic number of clusters in a multidimensional dataset is a commonly encountered problem in exploratory data analysis. Unsupervised clustering algorithms often rely on specification of cluster number as an input parameter. However, this is typically not known a priori. Many methods have been proposed to estimate cluster number, including statistical and information-theoretic approaches such as the gap statistic, but these methods are not always reliable when applied to non-normally distributed datasets containing outliers or noise. In this study, I propose a novel method called hierarchical linkage regression, which uses regression to estimate the intrinsic number of clusters in a multidimensional dataset. The method operates on the hypothesis that the organization of data into clusters can be inferred from the hierarchy generated by partitioning the dataset, and therefore does not directly depend on the specific values of the data or their distribution, but on their relative ranking within the partitioned set. Moreover, the technique does not require empirical data to train on, but can use synthetic data generated from random distributions to fit regression coefficients. The trained hierarchical linkage regression model is able to infer cluster number in test datasets of varying complexity and differing distributions, for image, text and numeric data, using the same regression model without retraining. The method performs favourably against other cluster number estimation techniques, and is also robust to parameter changes, as demonstrated by sensitivity analysis. The apparent robustness and generalizability of hierarchical linkage regression make it a promising tool for unsupervised exploratory data analysis and discovery.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Algorithms*
  • Cluster Analysis
  • Datasets as Topic
  • Humans
  • Pattern Recognition, Automated* / methods
  • Regression Analysis
  • Unsupervised Machine Learning*

Grants and funding

O.Z. received less than $500 CAD from the Queen’s University Department of Oncology Cancer Research and Education Trust Fund as a one time contribution for departmental research. The funders had no role in design, data collection and analysis, decision to publish, or preparation of the manuscript.