ImageNet classification with deep convolutional neural networks
Alex Krizhevsky, Ilya Sutskever et al.
4.8k
Citations
249
Influential Citations
PLoS ONE
Venue
2015
Year
Binary classifiers are routinely evaluated with performance measures such as sensitivity and specificity, and performance is frequently illustrated with Receiver Operating Characteristics (ROC) plots. Alternative measures such as positive predictive value (PPV) and the associated Precision/Recall (PRC) plots are used less frequently. Many bioinformatics studies develop and evaluate classifiers that are to be applied to strongly imbalanced datasets in which the number of negatives outweighs the number of positives significantly. While ROC plots are visually appealing and provide an overview of a classifier's performance across a wide range of specificities, one can ask whether ROC plots could be misleading when applied in imbalanced classification scenarios. We show here that the visual interpretability of ROC plots in the context of imbalanced datasets can be deceptive with respect to conclusions about the reliability of classification performance, owing to an intuitive but wrong interpretation of specificity. PRC plots, on the other hand, can provide the viewer with an accurate prediction of future classification performance due to the fact that they evaluate the fraction of true positives among positive predictions. Our findings have potential implications for the interpretation of a large number of studies that use ROC plots on imbalanced datasets.
This paper addresses a critical issue in machine learning evaluation: the widespread use of ROC plots for imbalanced datasets can lead to overly optimistic performance estimates. Many practitioners default to ROC curves because they are visually appealing and provide a single metric (AUC), but the authors show that this can be deceptive when negatives vastly outnumber positives. The paper is highly cited (over 4,700 citations), indicating its significant impact on how researchers evaluate classifiers in domains like bioinformatics, fraud detection, and medical diagnosis.
The paper does not present new experimental results but uses illustrative examples to show that ROC plots can indicate high performance (e.g., AUC near 1) even when the classifier performs poorly on the positive class. In contrast, PRC plots correctly reflect poor performance in such cases. The authors argue that for imbalanced datasets, PRC plots should be the default evaluation tool.
This paper has had a lasting impact on the machine learning community, leading to a shift in evaluation practices for imbalanced classification. It is now common to see PRC plots alongside or instead of ROC plots in many application papers. The work also underscores the importance of choosing evaluation metrics that align with the problem's class distribution and practical goals.
Alex Krizhevsky, Ilya Sutskever et al.
Ashish Vaswani, Noam Shazeer et al.
Douglas M. Bates, Martin Mächler et al.
Diederik P. Kingma, Jimmy Ba