28
0

Does it pay to optimize AUC?

Baojian Zhou
Steven Skiena
Abstract

The Area Under the ROC Curve (AUC) is an important model metric for evaluating binary classifiers, and many algorithms have been proposed to optimize AUC approximately. It raises the question of whether the generally insignificant gains observed by previous studies are due to inherent limitations of the metric or the inadequate quality of optimization. To better understand the value of optimizing for AUC, we present an efficient algorithm, namely AUC-opt, to find the provably optimal AUC linear classifier in R2\mathbb{R}^2, which runs in O(n+nlog(n+n))\mathcal{O}(n_+ n_- \log (n_+ n_-)) where n+n_+ and nn_- are the number of positive and negative samples respectively. Furthermore, it can be naturally extended to Rd\mathbb{R}^d in O((n+n)d1log(n+n))\mathcal{O}((n_+n_-)^{d-1}\log (n_+n_-)) by calling AUC-opt in lower-dimensional spaces recursively. We prove the problem is NP-complete when dd is not fixed, reducing from the \textit{open hemisphere problem}. Experiments show that compared with other methods, AUC-opt achieves statistically significant improvements on between 17 to 40 in R2\mathbb{R}^2 and between 4 to 42 in R3\mathbb{R}^3 of 50 t-SNE training datasets. However, generally the gain proves insignificant on most testing datasets compared to the best standard classifiers. Similar observations are found for nonlinear AUC methods under real-world datasets.

View on arXiv
Comments on this paper