Mortality predictors in patients with COVID-19 pneumonia: a machine learning approach using eXtreme Gradient Boosting model

Abstract

Recently, global health has seen an increase in demand for assistance as a result of the COVID-19 pandemic. This has prompted many researchers to conduct different studies looking for variables that are associated with increased clinical risk, and find effective and safe treatments. Many of these studies have been limited by presenting small samples and a large data set. Using machine learning (ML) techniques we can detect parameters that help us to improve clinical diagnosis, since they are a system for the detection, prediction and treatment of complex data. ML techniques can be valuable for the study of COVID-19, especially because they can uncover complex patterns in large data sets. This retrospective study of 150 hospitalized adult COVID-19 patients, of which we established two groups, those who died were called Case group (n = 53) while the survivors were Control group (n = 98). For analysis, a supervised learning algorithm eXtreme Gradient Boosting (XGBoost) has been used due to its good response compared to other methods because it is highly efficient, flexible and portable. In this study, the response to different treatments has been evaluated and has made it possible to accurately predict which patients have higher mortality using artificial intelligence, obtaining better results compared to other ML methods.