Regression trees for predicting mortality in patients with cardiovascular disease: what improvement is achieved by using ensemble-based methods?
In biomedical research, the logistic regression model is the most commonly used method for predicting the probability of a binary outcome. While many clinical researchers have expressed an enthusiasm for regression trees, this method may have limited accuracy for predicting health outcomes. We aimed to evaluate the improvement that is achieved by using ensemble-based methods, including bootstrap aggregation (bagging) of regression trees, random forests, and boosted regression trees. We analyzed 30-day mortality in two large cohorts of patients hospitalized with either acute myocardial infarction (N = 16,230) or congestive heart failure (N = 15,848) in two distinct eras (1999-2001 and 2004-2005). We found that both the in-sample and out-of-sample prediction of ensemble methods offered substantial improvement in predicting cardiovascular mortality compared to conventional regression trees. However, conventional logistic regression models that incorporated restricted cubic smoothing splines had even better performance. We conclude that ensemble methods from the data mining and machine learning literature increase the predictive performance of regression trees, but may not lead to clear advantages over conventional logistic regression models for predicting short-term mortality in population-based samples of subjects with cardiovascular disease.
Institute for Clinical Evaluative Sciences, Toronto, Ontario, Canada. peter.austin@ices.on.ca
Peter C
PC
Austin
Douglas S
DS
Lee
Ewout W
EW
Steyerberg
Jack V
JV
Tu
D000368:Aged; D001699:Biometry; D002318:Cardiovascular Diseases / Q000175:diagnosis / Q000401:mortality*; D003663:Decision Trees; D005260:Female; D006333:Heart Failure / Q000175:diagnosis / Q000401:mortality; D006760:Hospitalization; D006801:Humans; D016015:Logistic Models; D008297:Male; D008875:Middle Aged; D009203:Myocardial Infarction / Q000175:diagnosis / Q000401:mortality; D012306:Risk