Implications of imbalanced datasets for empirical ROC-AUC estimation in binary classification tasks

• Published in: Journal of statistical computation and simulation 2024-01, Vol.94 (1), p.183-203
• Main Authors: Liu, Yujian, Li, Yazhe, Xie, Dejun
• Format: Article
• Language: English
• Subjects: Area under receiver operating characteristic curve; Classifiers; Datasets; Empirical analysis; empirical AUC estimator; imbalanced dataset
• ISSN: 0094-9655; 1563-5163
• DOI: 10.1080/00949655.2023.2238235

The area under the curve (AUC) is the most popular measure for summarizing a binary classifier's receiver operating characteristic (ROC) curve. Therefore, it is essential to ensure that the AUC estimation is accurate. One straightforward and popular estimation approach is to calculate the empirical AUC from the data. However, one must look closely at the behaviour of this point estimator, particularly its variance. This study demonstrates both analytically and empirically that the empirical AUC estimation could be highly volatile in many circumstances when applied to an imbalanced dataset. To be more specific, we have proved that under some frequently encountered circumstances, variances of the empirical AUC estimator increase with the imbalanced level of the dataset. Hence, under the imbalanced setting, variances could be high. Furthermore, we conduct several simulations and experiments to solidify our findings. Therefore, extra attention must be paid when the empirical ROC-AUC is used to summarize the classifier's performance, especially when the dataset presents high class imbalance.