Evaluation of Random Forest and Ensemble Methods at Predicting Complications Following Cardiac Surgery
- 807 Downloads
Cardiac patients undergoing surgery face increased risk of postoperative complications, due to a combination of factors, including higher risk surgery, their age at time of surgery and the presence of co-morbid conditions. They will therefore require high levels of care and clinical resources throughout their perioperative journey (i.e. before, during and after surgery). Although surgical mortality rates in the UK have remained low, postoperative complications on the other hand are common and can have a significant impact on patients’ quality of life, increase hospital length of stay and healthcare costs. In this study we used and compared several machine learning methods – random forest, AdaBoost, gradient boosting model and stacking – to predict severe postoperative complications after cardiac surgery based on preoperative variables obtained from a surgical database of a large acute care hospital in Scotland. Our results show that AdaBoost has the best overall performance (AUC = 0.731), and also outperforms EuroSCORE and EuroSCORE II in other studies predicting postoperative complications. Random forest (Sensitivity = 0.852, negative predictive value = 0.923), however, and gradient boosting model (Sensitivity = 0.875 and negative predictive value = 0.920) have the best performance at predicting severe postoperative complications based on sensitivity and negative predictive value.
KeywordsPostoperative complications Machine learning Cardiac surgery
- 1.Findlay, G.P., Goodwin, A.P.L., Protopapa, K., Smith, N.C.E., Mason, M.: Knowing the risk: a review of the peri-operative care of surgical patients, London (2011)Google Scholar
- 5.SCTS: Blue Book Online. http://bluebook.scts.org/
- 16.Freund, Y., Schapire, R.E.: Experiments with a new boosting algorithm. In: The Thirteenth International Conference on Machine Learning, pp. 148–156 (1996)Google Scholar
- 20.Zhang, H.: The optimality of naive Bayes. In: FLAIRS2004 Conference (2004)Google Scholar