An experimental comparison of cross-validation techniques for estimating the area under the ROC curve

被引:119
作者
Airola, Antti [1 ,2 ]
Pahikkala, Tapio [1 ,2 ]
Waegeman, Willem [3 ]
De Baets, Bernard [3 ]
Salakoski, Tapio [1 ,2 ]
机构
[1] Univ Turku, Dept Informat Technol, Turku 20014, Finland
[2] Turku Ctr Comp Sci TUCS, Turku 20520, Finland
[3] Univ Ghent, Dept Appl Math Biometr & Proc Control, KERMIT, Ghent, Belgium
基金
芬兰科学院;
关键词
Area under the ROC curve; Classifier performance estimation; Conditional AUC estimation; Cross-validation; Leave-pair-out cross-validation; SUPPORT VECTOR MACHINE; REGRESSION; ACCURACY;
D O I
10.1016/j.csda.2010.11.018
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
Reliable estimation of the classification performance of inferred predictive models is difficult when working with small data sets. Cross-validation is in this case a typical strategy for estimating the performance. However, many standard approaches to cross-validation suffer from extensive bias or variance when the area under the ROC curve (AUC) is used as the performance measure. This issue is explored through an extensive simulation study. Leave-pair-out cross-validation is proposed for conditional AUC-estimation, as it is almost unbiased, and its deviation variance is as low as that of the best alternative approaches. When using regularized least-squares based learners, efficient algorithms exist for calculating the leave-pair-out cross-validation estimate. (C) 2010 Elsevier B.V. All rights reserved.
引用
收藏
页码:1828 / 1844
页数:17
相关论文
共 47 条
[1]  
Agarwal S, 2005, J MACH LEARN RES, V6, P393
[2]  
Airola A., 2009, MACHINE LEARNING SYS, P3
[3]   All-paths graph kernel for protein-protein interaction extraction with evaluation of cross-corpus learning [J].
Airola, Antti ;
Pyysalo, Sampo ;
Bjoerne, Jari ;
Pahikkala, Tapio ;
Ginter, Filip ;
Salakoski, Tapio .
BMC BIOINFORMATICS, 2008, 9 (Suppl 11)
[4]   Fast cross-validation algorithms for least squares support vector machine and kernel ridge regression [J].
An, Senjian ;
Liu, Wanquan ;
Venkatesh, Svetha .
PATTERN RECOGNITION, 2007, 40 (08) :2154-2162
[5]  
[Anonymous], P KDD 2001 KNOWL DIS
[6]  
[Anonymous], [No title captured]
[7]  
[Anonymous], P 2 INT INT C AD KNO
[8]  
[Anonymous], 2010, SIGKDD Explor.
[9]   Identifying genes that contribute most to good classification in microarrays [J].
Baker, Stuart G. ;
Kramer, Barnett S. .
BMC BIOINFORMATICS, 2006, 7 (1)
[10]   The use of the area under the roc curve in the evaluation of machine learning algorithms [J].
Bradley, AP .
PATTERN RECOGNITION, 1997, 30 (07) :1145-1159