title: Classifying highly imbalanced ICU data.
abstract: Highly imbalanced data sets are those where the class of interest is rare. In this paper, we compare the performance of several common data mining methods, logistic regression, discriminant analysis, Classification and Regression Tree (CART) models, C5, and Support Vector Machines (SVM) in predicting the discharge status (alive or deceased, with "deceased" being the class of interest) of patients from an Intensive Care Unit (ICU). Using a variety of misclassification cost ratio (MCR) values and using specificity, recall, precision, the F-measure, and confusion entropy (CEN) as criteria for evaluating each method's performance, C5 and SVM performed better than the other methods. At a MCR of 100, C5 had the highest recall and SVM the highest specificity and lowest CEN. We also used Hand's measure to compare the five methods. According to Hand's measure, logistic regression performed the best. This article makes several contributions. We show how the use of MCR for analyzing imbalanced medical data significantly improves the method's classification performance. We also found that the F-measure and precision did not improve as the MCR was increased.
authors.affiliation: University of Pittsburgh, Pittsburgh, PA 15260, USA. yfr1@pitt.edu
authors.forename: Yazan F
authors.initials: YF
authors.lastname: Roumani
authors.forename: Jerrold H
authors.initials: JH
authors.lastname: May
authors.forename: David P
authors.initials: DP
authors.lastname: Strum
authors.forename: Luis G
authors.initials: LG
authors.lastname: Vargas
mesh_terms: D003625:Data Collection / Q000379:methods*; D057225:Data Mining / Q000379:methods*; D003663:Decision Trees; D016002:Discriminant Analysis; D005260:Female; D017052:Hospital Mortality; D006801:Humans; D007362:Intensive Care Units / Q000706:statistics & numerical data*; D016015:Logistic Models; D008297:Male; D008875:Middle Aged; D015233:Models, Statistical; D010351:Patient Discharge / Q000706:statistics & numerical data*; D060388:Support Vector Machine; D014481:United States