How to find a good explanation for clustering? - Archive ouverte HAL
Article Dans Une Revue Artificial Intelligence Année : 2023

How to find a good explanation for clustering?

Résumé

k-means and k-median clustering are powerful unsupervised machine learning techniques. However, due to complicated dependencies on all the features, it is challenging to interpret the resulting cluster assignments. Moshkovitz, Dasgupta, Rashtchian, and Frost proposed an elegant model of explainable k-means and k-median clustering in ICML 2020. In this model, a decision tree with k leaves provides a straightforward characterization of the data set into clusters. We study two natural algorithmic questions about explainable clustering. (1) For a given clustering, how to find the “best explanation” by using a decision tree with k leaves? (2) For a given set of points, how to find a decision tree with k leaves minimizing the k-means/median objective of the resulting explainable clustering? To address the first question, we introduce a new model of explainable clustering. Our model, inspired by the notion of outliers in robust statistics, is the following. We are seeking a small number of points (outliers) whose removal makes the existing clustering well-explainable. For addressing the second question, we initiate the study of the model of Moshkovitz et al. from the perspective of multivariate complexity. Our rigorous algorithmic analysis sheds some light on the influence of parameters like the input size, dimension of the data, the number of outliers, the number of clusters, and the approximation ratio, on the computational complexity of explainable clustering.
Fichier principal
Vignette du fichier
1-s2.0-S0004370223000942-main.pdf (678.55 Ko) Télécharger le fichier
Origine Fichiers éditeurs autorisés sur une archive ouverte
Licence

Dates et versions

hal-04288109 , version 1 (17-11-2023)

Licence

Identifiants

Citer

Sayan Bandyapadhyay, Fedor V. Fomin, Petr A. Golovach, William Lochet, Nidhi Purohit, et al.. How to find a good explanation for clustering?. Artificial Intelligence, 2023, 322, pp.103948. ⟨10.1016/j.artint.2023.103948⟩. ⟨hal-04288109⟩
80 Consultations
52 Téléchargements

Altmetric

Partager

More