Evaluating and tuning predictive data mining models using receiver operating characteristic curves

• 成果类型: Article
• 署名作者: Sinha, AP; May, JH
• 署名单位: University of Wisconsin System; University of Wisconsin Madison; Pennsylvania Commonwealth System of Higher Education (PCSHE); University of Pittsburgh; Pennsylvania Commonwealth System of Higher Education (PCSHE); University of Pittsburgh
• 刊物名称: JOURNAL OF MANAGEMENT INFORMATION SYSTEMS
• ISSN/ISSBN: 0742-1222
• DOI: 10.1080/07421222.2004.11045815
• 发表日期: 2004
• 页码: 249-280

摘要

In this study, we conduct an empirical analysis of the performance of five popular data mining methods-neural networks, logistic rearession, linear discriminant analysis, decision trees, and nearest neighbor-on two binary classification problems from the credit evaluation domain. Whereas most studies comparing data mining methods have employed accuracy as a performance measure, we argue that, for problems such as credit evaluation, the focus should be on minimizing misclassification cost. We first generate receiver operating characteristic (ROC) curves for the classifiers and use the area under the curve (AUC) measure to compare aggregate performance of the five methods over the spectrum of decision thresholds. Next. using the ROC results, we propose a method for tuning the classifiers by identifying optimal decision thresholds. We compare the methods based on expected costs across a range of cost-probability ratios. In addition to expected cost and AUC, we evaluate the models on the basis of their generalizability to unseen data, their scalability to other problems in the domain, and their robustness against changes in class distributions. We found that the performance of logistic regression and neural network models was superior under most conditions. In contrast. decision tree and nearest neighbor models yielded higher costs, and were much less generalizable and robust than the other models. An important finding, of this research is that the models can be effectively tuned post hoc to make them cost sensitive, even though they were built without incorporating misclassification costs.