Predicting ICU Mortality in Rheumatic Heart Disease: Comparison of XGBoost and Logistic Regression

Frontiers in Cardiovascular Medicine

article

Yixian Xu¹  Didi Han²  Tao Huang³  Xiaoshen Zhang⁴  Hua Lu⁴  Si Shen⁵  Jun Lyu³  Hao Wang¹ 

[1] Department of Anesthesiology, The First Affiliated Hospital of Jinan University;School of Public Health, Xi'an Jiaotong University Health Science Center;Department of Clinical Research, The First Affiliated Hospital of Jinan University;Department of Cardiovascular Surgery, The First Affiliated Hospital of Jinan University;Department of Radiology, Medical Imaging Center, The First Affiliated Hospital of Jinan University

关键词: MIMIC-IV;    rheumatic heart disease;    XGBoost;    logistic regression;    intensive care unit;    mortality;    prediction;   

DOI  :  10.3389/fcvm.2022.847206

学科分类：地球科学(综合)

来源: Frontiers

【 摘 要 】

Background Rheumatic heart disease (RHD) accounts for a large proportion of Intensive Care Unit (ICU) deaths. Early prediction of RHD can help with timely and appropriate treatment to improve survival outcomes, and the XGBoost machine learning technology can be used to identify predictive factors; however, its use has been limited in the past. We compared the performance of logistic regression and XGBoost in predicting hospital mortality among patients with RHD from the Medical Information Mart for Intensive Care IV (MIMIC-IV) database. Methods The patients with RHD in the MIMIC-IV database were divided into two groups retrospectively according to the availability of data and its clinical significance based on whether they survived or died. Backward stepwise regression was used to analyze the independent factors influencing patients with RHD, and to compare the differences between the two groups. The XGBoost algorithm and logistic regression were used to establish two prediction models, and the areas under the receiver operating characteristic curves (AUCs) and decision-curve analysis (DCA) were used to test and compare the models. Finally, DCA and the clinical impact curve (CIC) were used to validate the model. Results Data on 1,634 patients with RHD were analyzed, comprising 207 who died during hospitalization and 1,427 survived. According to estimated results for the two models using AUCs [0.838 (95% confidence interval = 0.786–0.891) and 0.815 (95% confidence interval = 0.765–0.865)] and DCA, the logistic regression model performed better. DCA and CIC verified that the logistic regression model had convincing predictive value. Conclusions We used logistic regression analysis to establish a more meaningful prediction model for the final outcome of patients with RHD. This model might be clinically useful for patients with RHD and help clinicians to provide detailed treatments and precise management.

【 授权许可】

CC BY