The area under the ROC curve as a measure of clustering quality

Pablo A. Jaskowiak*, Ivan G. Costa, Ricardo J.G.B. Campello

*Kontaktforfatter

Publikation: Bidrag til tidsskriftTidsskriftartikelForskningpeer review

Abstract

The area under the receiver operating characteristics (ROC) Curve, referred to as AUC, is a well-known performance measure in the supervised learning domain. Due to its compelling features, it has been employed in a number of studies to evaluate and compare the performance of different classifiers. In this work, we explore AUC as a performance measure in the unsupervised learning domain, more specifically, in the context of cluster analysis. In particular, we elaborate on the use of AUC as an internal/relative measure of clustering quality, which we refer to as Area Under the Curve for Clustering (AUCC). We show that the AUCC of a given candidate clustering solution has an expected value under a null model of random clustering solutions, regardless of the size of the dataset and, more importantly, regardless of the number or the (im)balance of clusters under evaluation. In addition, we elaborate on the fact that, in the context of internal/relative clustering validation as we consider, AUCC is actually a linear transformation of the Gamma criterion from Baker and Hubert (1975), for which we also formally derive a theoretical expected value for chance clusterings. We also discuss the computational complexity of these criteria and show that, while an ordinary implementation of Gamma can be computationally prohibitive and impractical for most real applications of cluster analysis, its equivalence with AUCC actually unveils a much more efficient algorithmic procedure. Our theoretical findings are supported by experimental results. These results show that, in addition to an effective and robust quantitative evaluation provided by AUCC, visual inspection of the ROC curves themselves can be useful to further assess a candidate clustering solution from a broader, qualitative perspective as well.

OriginalsprogEngelsk
TidsskriftData Mining and Knowledge Discovery
Vol/bind36
Udgave nummer3
Sider (fra-til)1219-1245
ISSN1384-5810
DOI
StatusUdgivet - maj 2022
Udgivet eksterntJa

Bibliografisk note

Funding Information:
This project was partially funded by Brazilian research agencies FAPESP (Process #2011/04247-5) and CNPq (#302161/2017-1). Ivan G. Costa was supported by the Interdisciplinary Center for Clinical Research (IZKF) Faculty of Medicine at the RWTH Aachen.

Publisher Copyright:
© 2022, The Author(s), under exclusive licence to Springer Science+Business Media LLC, part of Springer Nature.

Fingeraftryk

Dyk ned i forskningsemnerne om 'The area under the ROC curve as a measure of clustering quality'. Sammen danner de et unikt fingeraftryk.

Citationsformater