Heliyon (Jan 2024)
Machine learning predictors of risk of death within 7 days in patients with non-traumatic subarachnoid hemorrhage in the intensive care unit: A multicenter retrospective study
Abstract
Non-traumatic subarachnoid hemorrhage (SAH) is a critical neurosurgical emergency with a high mortality rate, imposing a significant burden on both society and families. Accurate prediction of the risk of death within 7 days in SAH patients can provide valuable information for clinicians, enabling them to make better-informed medical decisions. In this study, we developed six machine learning models using the MIMIC III database and data collected at our institution. These models include Logistic Regression (LR), AdaBoosting (AB), Multilayer Perceptron (MLP), Bagging (BAG), Gradient Boosting Machines (GBM), and Extreme Gradient Boosting (XGB). The primary objective was to identify predictors of death within 7 days in SAH patients admitted to intensive care units. We employed univariate and multivariate logistic regression as well as Pearson correlation analysis to screen the clinical variables of the patients. The initially screened variables were then incorporated into the machine learning models, and the performance of these models was evaluated. Furthermore, we compared the performance differences among the six models and found that the MLP model exhibited the highest performance with an AUC of 0.913. In this study, we conducted risk factor analysis using Shapley values to identify the factors associated with death within 7 days in patients with SAH. The risk factors we identified include Gcsmotor, bicarbonate, wbc, spo2, heartrate, age, nely, glucose, aniongap, GCS, rbc, sysbp, sodium, and gcseys. To provide clinicians with a useful tool for assessing the risk of death within 7 days in SAH patients, we developed a web calculator based on the MLP machine learning model.