The use of the area under the roc curve in the evaluation of machine learning algorithms

被引:4663
|
作者
Bradley, AP [1 ]
机构
[1] UNIV QUEENSLAND,DEPT ELECT & COMP ENGN,COOPERAT RES CTR SENSOR SIGNAL & INFORMAT PROC,ST LUCIA,QLD 4072,AUSTRALIA
关键词
the ROC curve; the area under the ROC curve (AUC); accuracy measures; cross-validation; Wilcoxon statistic; standard error;
D O I
10.1016/S0031-3203(96)00142-2
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In this paper we investigate the use of the area under the receiver operating characteristic (ROC) curve (AUC) as a performance measure for machine learning algorithms. As a case study we evaluate six machine learning algorithms (C4.5, Multiscale Classifier, Perceptron, Multi-layer Perceptron, k-Nearest Neighbours, and a Quadratic Discriminant Function) on six ''real world'' medical diagnostics data sets. We compare and discuss the use of AUC to the more conventional overall accuracy and find that AUC exhibits a number of desirable properties when compared to overall accuracy: increased sensitivity in Analysis of Variance (ANOVA) tests; a standard error that decreased as both AUC and the number of test samples increased; decision threshold independent; and it is invariant to a priori class probabilities. The paper concludes with the recommendation that AUC be used in preference to overall accuracy for ''single number'' evaluation of machine learning algorithms. (C) 1997 Pattern Recognition Society.
引用
收藏
页码:1145 / 1159
页数:15
相关论文
共 50 条
  • [1] Use of the area under the ROC curve in the evaluation of machine learning algorithms
    Bradley, Andrew P.
    1997, Elsevier Science Ltd, Oxford, United Kingdom (30)
  • [2] On use of partial area under the ROC curve for evaluation of diagnostic performance
    Ma, Hua
    Bandos, Andriy I.
    Rockette, Howard E.
    Gur, David
    STATISTICS IN MEDICINE, 2013, 32 (20) : 3449 - 3458
  • [3] Area Under the ROC Curve Maximization for Metric Learning
    Gajic, Bojana
    Amato, Ariel
    Baldrich, Ramon
    van de Weijer, Joost
    Gatta, Carlo
    2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS, CVPRW 2022, 2022, : 2806 - 2815
  • [4] Active learning to maximize area under the ROC curve
    Culver, Matt
    Kun, Deng
    Scott, Stephen
    ICDM 2006: SIXTH INTERNATIONAL CONFERENCE ON DATA MINING, PROCEEDINGS, 2006, : 149 - +
  • [5] The area under the ROC curve
    Cantarero, GG
    Jarabo, RM
    MEDICINA CLINICA, 1996, 106 (09): : 355 - 356
  • [6] Support Vector Algorithms for Optimizing the Partial Area under the ROC Curve
    Narasimhan, Harikrishna
    Agarwal, Shivani
    NEURAL COMPUTATION, 2017, 29 (07) : 1919 - 1963
  • [7] Bounds on the area under the ROC curve
    Shapiro, JH
    JOURNAL OF THE OPTICAL SOCIETY OF AMERICA A-OPTICS IMAGE SCIENCE AND VISION, 1999, 16 (01): : 53 - 57
  • [8] LIMITATIONS OF AREA UNDER THE ROC CURVE
    Oden, Anders
    Johansson, Helena
    McCloskey, Eugene
    Kanis, John A.
    OSTEOPOROSIS INTERNATIONAL, 2012, 23 : S196 - S196
  • [9] Estimation of the area under the ROC curve
    Faraggi, D
    Reiser, B
    STATISTICS IN MEDICINE, 2002, 21 (20) : 3093 - 3106
  • [10] Bounds on the area under the ROC curve
    J Opt Soc Am A, 1 (53):