- Research
- Open access
- Published:
Interpretable machine learning model for early prediction of 28-day mortality in ICU patients with sepsis-induced coagulopathy: development and validation
European Journal of Medical Research volume 29, Article number: 14 (2024)
Abstract
Objective
Sepsis-induced coagulopathy (SIC) is extremely common in individuals with sepsis, significantly associated with poor outcomes. This study attempted to develop an interpretable and generalizable machine learning (ML) model for early predicting the risk of 28-day death in patients with SIC.
Methods
In this retrospective cohort study, we extracted SIC patients from the Medical Information Mart for Intensive Care III (MIMIC-III), MIMIC-IV, and eICU-CRD database according to Toshiaki Iba's scale. And the overlapping in the MIMIC-IV was excluded for this study. Afterward, only the MIMIC-III cohort was randomly divided into the training set, and the internal validation set according to the ratio of 7:3, while the MIMIC-IV and eICU-CRD databases were considered the external validation sets. The predictive factors for 28-day mortality of SIC patients were determined using recursive feature elimination combined with tenfold cross-validation (RFECV). Then, we constructed models using ML algorithms. Multiple metrics were used for evaluation of performance of the models, including the area under the receiver operating characteristic curve (AUROC), area under the precision recall curve (AUPRC), accuracy, sensitivity, specificity, negative predictive value, positive predictive value, recall, and F1 score. Finally, Shapley Additive Explanations (SHAP), Local Interpretable Model-Agnostic Explanations (LIME) were employed to provide a reasonable interpretation for the prediction results.
Results
A total of 3280, 2798, and 1668 SIC patients were screened from MIMIC-III, MIMIC-IV, and eICU-CRD databases, respectively. Seventeen features were selected to construct ML prediction models. XGBoost had the best performance in predicting the 28-day mortality of SIC patients, with AUC of 0.828, 0.913 and 0.923, the AUPRC of 0.807, 0.796 and 0.921, the accuracy of 0.785, 0.885 and 0.891, the F1 scores were 0.63, 0.69 and 0.70 in MIMIC-III (internal validation set), MIMIC-IV, and eICU-CRD databases. The importance ranking and SHAP analyses showed that initial SOFA score, red blood cell distribution width (RDW), and age were the top three critical features in the XGBoost model.
Conclusions
We developed an optimal and explainable ML model to predict the risk of 28-day death of SIC patients 28-day death risk. Compared with conventional scoring systems, the XGBoost model performed better. The model established will have the potential to improve the level of clinical practice for SIC patients.
Graphical Abstract
Introduction
Sepsis is life-threatening organ dysfunction caused by a dysregulated host response to infection [1]. The global incidence is approximately 50 million person-years, which poses severe challenges to the public health systems of the countries [2]. Although research on the pathogenesis and treatment of sepsis has been carried out for decades, no specific treatment has been found so far. Currently, the hospital mortality of adults with sepsis is about 189/100,000 person-years, while the intensive care unit (ICU) mortality is as high as over 42% [3]. It is well known that coagulation abnormalities usually occur in sepsis patients. According to the International Society On thrombosis and Haemostasis Guideline, its incidence is maintained at roughly 50–70% [4]. The primary pathogenesis of coagulopathy in patients with sepsis is exceptionally complex, including massive activation of platelets and other inflammation cells (such as neutrophils and lymphocytes) and vascular endothelial damage. These mechanisms are manifested in the body's dysregulated response to inflammation, the platelet count decrease, the coagulation reaction enhancement and the anticoagulation mechanism injury, and large immune-micro thrombus formation, which in turn affects the perfusion of organs [5, 6].
Although the definition of sepsis-induced coagulopathy (SIC) is still controversial, it is generally considered that this stage is from the initial compensated period to the decompensated disseminated intravascular coagulation period. The gold standard for SIC is still unclear, and the Toshiaki Iba scale is mainly used for diagnosis, which is comprehensively judged from three aspects: the degree of thrombocytopenia, the international normalized ratio (INR) level, and the SOFA score [7]. A single-center retrospective observational trial with large samples has reported a strong association between SIC and poor outcomes in hospitalized patients [8], and any delayed or omitted interventions may be detrimental to such patients [9]. Thus, it is of utmost importance to identify the high-risk group of SIC patients and implement timely intervention therapy, which is essential to reduce mortality. To date, no single recognized evaluation criteria have been found to predict the prognosis of patients with SIC accurately.
Therefore, this present study aimed to establish a machine learning (ML) model for early prediction of 28-day death in SIC patients based on the Medical Information Mart for Intensive Care III (MIMIC-III) database and further validated in MIMIC-IV and eICU-CRD databases. In addition, we adopted the Shapley additive explanation (SHAP) and Local Interpretable Model-Agnostic Explanations (LIME) methods to provide a reasonable interpretation for the prediction results and assisted the clinical practice of intensivists and relevant researchers.
Methods
Data sources
Data used in this present study were obtained from three large, open-access databases called MIMIC-III (v 1.4), MIMIC-IV (v 1.0) and eICU-CRD. The MIMIC-III (v 1.4) contained comprehensive records of 46,520 patients admitted to the Beth Israel Deaconess Medical Center in Boston, Massachusetts, between June 2001 and October 2012 [10]; while the MIMIC-IV (v 1.0) comprised almost 300,000 patients at the same center who were admitted between 2008 and 2019 [11]. Furthermore, the eICU-CRD database is a multicenter database of over 200,000 ICU admissions in the United States. Considering the partially overlapping patients in MIMIC-III and MIMIC-IV datasets, we extracted the patients from 2012 to 2019 at the MIMIC-IV set using MIMIC-III Clinical Database CareVue subset (2001–2008) and the admission time [12]. The relevant clinical data included demographic characteristics, vital signs, laboratory results, imaging examinations, microbial culture results, medication and procedures records, survival information, and a data dictionary. To achieve authorization, users must complete the collaborative institution training initiative program course by the US National Institutes of Health. Zhou and Lu have finished the online examination and obtained a certification number (Record ID: 53186220, 38455175). Since the MIMIC and eICU-CRD are both publicly available anonymized databases, approval from the ethical committee was exempted.
Study population
Septic patients diagnosed with SIC on the first day of ICU admission were eligible for inclusion in the study. Only the first stay was included for analysis if patients were admitted to ICU more than once. The definition of sepsis was based on the Third International Consensus Definitions for Sepsis and Septic Shock (Sepsis-3), that is, patients with confirmed or suspected infection and a total SOFA score ≧ 2 [1]. Suspected infection refers to antibiotics administered within three days of the date of culture collection. According to Toshiaki Iba's rating scale, SIC was identified based on the PT-INR level, platelet count, and SOFA score [7]. The details about the SIC diagnostic criteria can be found in Additional file 3: Table S3. The exclusion criteria were (1) minors (< 18 years old); (2) pregnant women; (3) patients with congenital coagulopathy; (4) patients with neoplasm were also excluded, taking into account the effect of tumors and related chemotherapy agents on the coagulation function; (5) ICU stays less than 48 h.
Data extraction and feature engineering
PostgreSQL programming (v 4.21) and STATA software (v 15.1) were used to extract data and concatenate each list based on the specific hadm_id or stay_id code. The following information was extracted, including age, gender, weight, ICU types, comorbidities, SOFA (which excluded the platelet), LODS, SAPS II, SIC score, vital signs, laboratory parameters, infection site, mechanical ventilation use, norepinephrine use, and survival record. The average of each vital sign within the first 24 h after ICU entry was calculated and used for the analysis, including heart rate (HR), mean blood pressure (MBP), respiratory rate (RR), and temperature. The laboratory parameter value associated with the greatest severity of illness during the first 24 h after ICU admission was extracted (except for mean blood glucose concentration), including aniongap_max, bicarbonate_min, chloride_max, hematocrit_min, hemoglobin_min, lactate_max, platelet count_min, potassium_max, partial prothrombin time_max (PTT), INR, prothrombin time_max (PT), sodium_min, blood urea nitrogen_max (BUN), white blood cells_max (WBC), PO2-min, PCO2-max, PH-min, mean corpuscular hemoglobin concentration_min (MCHC), red blood cell distribution width_max (RDW), mean corpuscular volume_min (MCV), alanine aminotransferase_max (ALT), aspartate aminotransferase_max (AST), bilirubin, creatinine_max etc.. Comorbidities were identified by the International Classification of Diseases, Ninth Revision (ICD-9), combining with Tenth Revision (ICD-10) diagnosis codes when discharge, including hypertension, chronic obstructive pulmonary disease, diabetes, myocardial infarction, chronic heart failure, and liver disease. The outcome of this present study is the 28-day mortality following ICU admission. The patient was considered a survivor if there was no record of death_time within 28-day after ICU admission.
The feature engineering was completed in three steps. Firstly, missing value identifying and processing. In this study, we used the package “VIM” to recognize the distribution of missing values. Besides, features with more than 30% missing values were removed, such as ALT, AST, and bilirubin. Additional file 4: Figure S1 shows the percentage of missing values in each database. For the remaining features, missing values were imputed using the package “randomForest” of R. Secondly, outliers identifying and processing. Within normally distributed data, outliers were identified based on the 3σ principle. Furthermore, nonparametric data were tested for outliers using the interquartile range method. All outliers were eventually winsorized using the winsor2 command in STATA software. Thirdly, feature selection for model construction. Feature selection was performed by a tenfold Recursive Feature Elimination Cross-Validation (RFECV) based on a random forest regressor in the training set. RFE, as a greedy algorithm, ranked and selected features according to their importance by iterative training [13].
Statistical analysis
Normal distribution was assessed with Agostino tests. The continuous variables were presented as mean (standard deviation) or median (interquartile ranges, IQR) according to the type of data distribution and compared by unpaired Student's test or Mann–Whitney U-test. Categorical variables were compared using the χ2 or Fisher exact test.
The MIMIC-III database was randomly assigned with 70% for training and 30% for internal validation, while the MIMIC-IV and EICU database was used for external validation. Four machine learning methods (logistic regression-LR, XGBoost, support vector machine-SVM, and naive bayesian-NB) and three severity scoring systems (SOFA, SAPS II and SIC score) were, respectively, used to develop models for the ICU 28-day death prediction in SIC patients. We applied a tenfold cross-validated grid-search approach to the predefined models to achieve optimal parameters. The main parameters of XGBoost in this study were set as follows: n_estimators = 30, learning_rate = 0.23, max_depth = 3, gamma = 0. Areas under the receiver operating characteristic curve (AUROC), area under the precision-recall curve (AUPRC), accuracy, sensitivity, specificity, positive predictive value (PPV), negative predictive value (NPV), and F1 score were all calculated to evaluate the prediction performance of each model. All comparisons of AUROCs were performed as two-sided DeLong Tests. While, the F1 score is the harmonic mean of precision and recall, that is defined as follows: F1 = 2 × Precision x Recall/(Precision + Recall) [14]. SHAP and LIME algorithms were commonly applied in explaining the output of the machine learning model [15, 16]. The former applied a game-theoretic approach to evaluate feature contributions toward any model prediction and identify the features most prominently influenced by the provision of SHAP values [17]. In this study, both were used to explain the final prediction model with contributing risk factors resulting in ICU 28-day death in patients with SIC. In addition, the partial dependence plot of each feature contained in the final model was drawn using the "dependence plot" function to assess the connection between each feature and the risk of ICU 28-day death. All statistical analyses were performed using R software (v 3.6.2) and Python software (v 3.8.5). The framework of the prediction models is shown in Fig. 1.
Results
Baseline characteristic
After applying the inclusion and exclusion criteria, 3280 SIC patients were identified from the MIMIC-III database, while 2798 and 1668 SIC patients were from the MIMIC-IV and e-CIU database respectively. Subsequently, patients included in the MIMIC-III database were randomly assigned to training (N = 2296) and internal validation cohort (N = 984) with the ratio of 7:3. As shown in Table 1, the ICU 28 day mortality was 33.9% (779/2296) and 34.0% (335/984) in the training and internal validation cohort. SIC patients who died on the day 28 after ICU entry had an older age (68.34 and 69.28 in the training and internal validation cohort), higher proportion of liver disease (30% and 33%), and higher severity score (SOFA: 9 and 9; LODS: 8 and 8; SAPS II: 53 and 54) compared with survivors. Regarding the SIC score, the percentage of elevated SIC sore in non-survivors was significantly higher than in survivors in the training and internal validation cohort (Table 1). Meanwhile, non-survivors had faster mean HR (91.96 and 92.90 min−1 in training and internal validation cohort) and RR (20.6 and 20.17 min−1), lower MBP (72.42 and 71.63 mmHg) and temperature (36.71 and 36.72 ℃) within 24 h after ICU admission. Significant abnormalities in blood coagulation indexes, such as prolonged PTT (45 and 43.1 s in training and internal validation cohort) and PT time (18.3 and 18.9 s), higher INR (1.9 and 1.9), RDW (16.5 and 16.7%), and MCV (92 and 92.79 fL), were noted in the non-survivors group. The characteristics of the included SIC patients from the MIMIC-IV and eICU-CRD database were presented in Additional file 1: Table S1 and Additional file 2: Table S2.
Prediction model building and evaluation
Before prediction model construction, 20 features were preliminarily screened out using RFECV, including age, SOFA score, LODS score, HR_mean, systolic pressure_mean, MBP, RR_mean, temperature_min, lactate_max, platelet count_min, PTT_max, PT_max, INR_max, BUN_max, WBC_max count, PaO2_min, PH_min, MCHC_min, RDW_max, and MCV_min. However, systolic pressure, INR, PH, and LODS were deleted from the original features list, and gender was included based on expert consultations and clinical judgment results. Thus, 17 features were eventually included for further model building. Additional file 6: Figure S3 presents how the accuracy varies with the number of features in RFECV processing. Considering the potential bias results from the discrepancy of missing values, we further analyzed the differences between missing values for each of the 17 features between survivors and non-survivors in three databases. Moreover, Additional file 5: Figure S2 indicates that there was no significantly different in the distribution of the missing values between survivors and non-survivors in each database.
We utilized four machine learning models, XGBoost, LR, SVM, and NB, with these 17 features mentioned above to predict the risk of ICU 28-day death in SIC patients. The results demonstrated that the XGBoost presented the largest AUROC compare with other models in internal validation cohort and external validation cohort [internal validation cohort: 0.828 95% confidence interval (CI) 0.795, 0.861; MIMIC-IV: 0.913 95% CI 0.905, 0.932; eICU-CRD: 0.923 95% CI 0.913, 0.941] (Fig. 2A–C and Table 2), and these differences were significant when compared by DeLong test (P < 0.001). However, the results of AUROC may be insensitive due to the imbalance distribution of data; hence, we analyzed the AUPRC value of each model. The results presented that the XGBoost also performed best in three cohorts [internal validation cohort: 0.807 95% confidence interval (CI) 0.743, 0.864; MIMIC-IV: 0.796 95% CI 0.703, 0.884; eICU-CRD: 0.921 95% CI 0.874, 0.955] (Fig. 2D, E and Table 2). Besides, XGBoost outperformed the other algorithms, SOFA, SAPS II and SIC score on the aspect of accuracy (internal validation cohort: 0.785; MIMIC-IV cohort: 0.885; EICU cohort: 0.891) and F1score (internal validation cohort:0.63; MIMIC-IV cohort: 0.69; EICU cohort: 0.70). In addition, we drew the calibration plots using the bootstrap method and performed the decision curve analysis (DCA) of each model in three databases. As shown in Additional file 6: Figure S3D, E, F, the bias-corrected line slightly deviated from the ideal line, indicating a good agreement between the prediction and observation. And the DCA results demonstrated that the XGBoost model provided a greater net benefit when the threshold probability was within 0 and 1 in both databases (Additional file 6: Figure S3A, B, C). Therefore, we selected the XGBoost for all further analyses.
Explanation of risk factors
The importance score of 17 features used in the XGBoost model has been calculated to identify the critical features (Fig. 3A). The position on the Y-axis implied the importance ranking, and the X-axis reflected the association between each value of features and the corresponding SHAP value. For instance, the SHAP values for advanced age are generally greater than zero, indicating that with increasing age, the risk of death also increased in SIC patients. In addition, Fig. 3B displays the ranking of the features based on the average absolute SHAP value. The permutation importance results indicated that the top five risk features were SOFA score, RDW-max, age, MCV-min, and mean HR.
The partial dependence plot results showed the effect of a single feature on the output of the XGBoost model. As the SHAP value exceeds zero, it indicated a promoting effect on the outcome (Fig. 4). This study found a positive but no linear association between RDW-max, age, MCV-min, mean HR, mean RR, PT-max and death risk. Moreover, the risk elevated rapidly when BUN-max was above 24 mg/dL, lactate-max was above seven mmol/L, the mean temperature was below 36 ℃, PO2 was below 80 mmHg, MBP was below 70 mmHg, the minimum count of platelet was below 60 × 109/L, and MCHC-min was below 310 g/L in the first 24 h after ICU admission. In gender, men were generally at higher risk for ICU 28-day death than women.
Furthermore, this study assessed the potential interactions between RDW-max and initial SOFA or age. As shown in Additional file 8: Figure S5A, the risk of 28-day death of SIC patients increased when their initial SOFA score was elevated. Patients with a higher level of RDW-max had a lower risk of 28-day death when SOFA ≦ 7, yet, for patients with a SOFA ≧ 8, a higher level of RDW-max appeared to provide more risk of death. In addition, the impact of RDW-max also appeared to vary with age (Additional file 8: Figure S5B). Even though an increased initial RDW-max value induced a higher risk of 28-day death for SIC patients with age ≦ 60, a trend in the opposite direction was seen when age was greater than 60.
Interpretation of individual prediction
This present study explained the XGBoost prediction results of individual SIC patients using the SHAP and LIME, respectively. Additional file 9: Figure S6 provides two typical examples to illustrate the interpretability of SHAP. The features marked blue decreased the risk of death, while red features promoted death. Patient No.1, who belonged to the "true negative" group, was correctly predicted as a survivor (Additional file 9: Figure S6A). Patient No.2, who belonged to the "true positive" group, was correctly predicted as a non-survivor (Additional file 9: Figure S6B). The survivor was predicted to be alive due to higher mean HR (65.35 min−1), RDW (14.1%), BUN (16 mg/dL), platelet count (122 × 109/L), MCV (89 fL), PT(14.8 s), and mean RR (19.45 min−1). The non-survivor was predicted to die due to elevated initial SOFA score (15), mean HR (117.7 min−1), RDW (17%), PT (22.7 s), mean RR (22.12 min−1), and decreased platelet count (34 × 109/L). Besides, we conducted explanations for the two cases mentioned above based on the LIME (Additional file 10: Figure S7). The blue box indicated that the features are risk factors for ICU 28-day death, while the orange box indicated that the features are protective factors. And the LIME was similar to SHAP results.
Discussion
In this study, we have developed and validated machine learning models using 17 selected features, including age, gender, maximum SOFA score, mean HR, MBP, mean RR, temperature, lactate-max, minimum platelet count, PTT-max, PT-max, BUN-max, maximum WBC count, PO2-min, MCHC-min, RDW-max, and MCV-min to predict the ICU 28-day death risk. The above features could be easily collected within 24 h after ICU admission. The performance of each model was evaluated by AUROC, accuracy, sensitivity, specificity, PPV, NPV, and F1 score. XGBoost achieved the best prediction results, while the RF model performed worst. In addition, the SHAP function was used to interpret the prediction results of XGBoost to help intensivists better understand the process of this model decision and provide the basis for early interventions in SIC patients with a high risk of death.
Nowadays, ML has played a crucial role in the early warning and prognosis prediction of diseases [15, 16, 18]. These algorithms can analyze complex and non-linear data and even make a real-time prediction based on time series, which cannot be completed by traditional regression analysis. However, with the continuous development of algorithms, models become increasingly complex, increasing the difficulty of interpretation. This phenomenon is often referred to as the "black box" which is not conducive to the promotion of ML in the medical and health field [19]. To illustrate how these included features affect the 28-day mortality of SIC patients, we employ the SHAP value to analyze each feature. The SHAP is different from traditional feature importance. The latter only reports the importance permutation of features but cannot identify how each feature affects the model prediction results. In comparison, the most significant advantage of SHAP is that it can reflect permutation of importance and illustrate the positive and negative effects of included features. Figure 3 showesthat the five most important factors include the patient's initial SOFA score, RDW value, age, MCV, and mean heart rate, affecting SIC patients' ICU survival for 28 days. Meanwhile, when the values of each feature are different, the impacts on the prognosis are different. In addition, the relationship between some special continuous variables and the risk of unfavorable outcomes may not always be linear. Thus, exploring these features' risk threshold or trigger point in clinical practice has become even more critical. Unfortunately, it was considerably tricky for traditional linear models, such as logistic regression or Cox regression, to accomplish this goal.
In this developed XGBoost model, we constructed each feature's partial dependence plot to analyze further the correlation between each variable and the 28-day death risk. The stability of the urea generation rate is lower than that of serum creatinine and is susceptible to factors other than the kidney. BUN significantly increases when the glomerular filtration rate is reduced by 50% [20]. Meanwhile, Gaudry et al. demonstrated that a BUN level higher than 112 mg/dL is one of the major criteria for initiating restrictive renal replacement therapy [21]. In contrast, we found that the initial BUN level has little impact on the 28-day mortality of SIC patients, and it only exhibits harmful effects when the level is greater than 24 mg/dL. SOFA score and platelet count were included in Toshiaki Iba's scale. A single-center retrospective study by Lyons PG et al. referred to the Toshiaki Iba scale classified SIC patients into three levels, and their results showed that the severity of SIC was positively correlated with the patient's hospital mortality [8]. However, in this present study, Additional file 8: Figure S5 and Additional file 7: Figure S4K, respectively, presented that the significant contribution to mortality was not observed until the initial SOFA score was higher than eight or the platelet count was lower than 60 K/uL. After that, the risk of 28-day death rapidly elevated as the SOFA score increased and platelet count decreased. Thus, it appeared that the initial SIC score is not an ideal indicator for predicting the 28-day death risk in SIC patients. Serum lactate was a common biomarker to achieve risk stratification in sepsis patients. Mikkelsen et al. categorized initial venous lactate of sepsis patients as mild (≦2 mmol/L), middle (2–3.9 mmol/L), or severe degree (≧ 4 mmol/L), and proved that middle and severe lactate degree were all significantly associated with the 28-day mortality of sepsis patients whenever the presence (aOR5.14, 95% CI 1.74–15.18, p = 0.003) or absence ( aOR 3.33, 95% CI 1.47–7.56, p = 0.004) of septic shock using multivariable logistic regression [22]. Nevertheless, Fig. 4L shows that the SHAP value of the majority of samples remains at approximately zero when lactate level is below seven mmol/L and increases rapidly when lactate level is over 7 mmol/L, except for a few deviation samples. The discrepancy maybe since the status hyperlactatemia was also affected by lactate clearance. A series of studies have confirmed strong correlations between lactate clearance and prognosis in septic shock patients, even though increased lactate concentration may indirectly suggest tissue hypoxia [23, 24]. However, ongoing hyperlactatemia or a significant increase in lactate levels may reflect the decreased clearance rather than an increased production in lactate metabolism [25]. This is typically seen in sepsis patients combined with liver dysfunction.
RDW level reflects the size heterogeneity of the erythrocytes and indicates the body's response to oxidative stress and inflammation [26]. In recent years, a growing number of researches have shown the potential value of RDW in predicting the prognosis of sepsis [27,28,29,30]. A meta-analysis that included 11 studies showed that elevated RDW was positively associated with mortality of sepsis patients (HR 1.14, 95% CI 1.09–1.20, p < 0.001). Besides, the related subgroup and sensitivity analysis results based on quality, infection sites, and complications also supported this view [27]. However, few studies had assessed the connection between RDW level and adverse prognosis of sepsis patients in different severity and age. As shown in Additional file 10: Figure S7, we noted that the elderly patients (age greater than 60) with a higher RDW seemed to have a lower 28-day death risk. This phenomenon was opposite to Wang et al.'s experimental observation in which a total of 117 sepsis patients were included. They found that the in-hospital mortality increased 1.18 fold for each 1% increase in RDW [31]. This difference can result from baseline discordance. In addition, the physiologic rising of RDW may occur in some unique elderly patients [32]. Overall, the effect of increased RDW on elderly patients with sepsis is still controversial, which is worthy of follow-up studies.
Strengths and limitations
Compared with our previous study, this research project has several notable strengths. Firstly, the XGBoost has a good nonlinear fitting ability and improves prediction accuracy. Secondly, the SHAP and LIME solve the "black box" problem well for ML models. Thirdly, based on the SHAP values, we ranked the risk factors and illustrated their positive and negative effects on the 28-day mortality of SIC patients. Finally, and most importantly, we explored these features' risk threshold or trigger point based on a partial dependence plot.
However, there were some potential limitations to this study. First, we may have missed the inclusion of the sickest patients because those who died within the first 48 h of ICU admission were excluded. This implied that there are likely to be significant differences in baseline variables between patients who were included and those who were not included. Second, the specificities of the XGBoost model were 0.904 and 0.974, respectively, in the internal validation and external validation set; in contrast, the sensitivities were only 0.646 and 0.523. It suggested the presence of a sizeable false-negative rate in the prediction of 28-day mortality in SIC patients; thus, further clinical experience and medical judgment should be recommended for those where the model yield negative results. Third, despite extensive data, we could not obtain key coagulation indexes, such as D2 polymers, fibrinogen, and thrombin-anti-thrombin III complexes. This study is only the first step to building a death risk prediction model for SIC patients. In future studies, clinical ML models need to account for different domains (e.g., immunology, pathogenesis, and clinical phenotype) to identify SIC patients' progressive trajectories and develop a more accurate and reliable prediction model.
Conclusion
In summary, our study developed an ML model based on MIMIC-III and MIMIC-IV databases to predict the risk of 28-day death of SIC patients early. The XGBoost performed better than LR, NB, SVM, SOFA, and SAPS II scores. SHAP and LIME are reliable methods to intuitively identify the related risk factors that affected the model making final predictions. The results can assist clinicians in screening SIC patients at high risk of 28-day death, contributing to the optimization of medical resources.
Availability of data and materials
The datasets are available in the PhysioNet (https://physionet.org/content/mimiciv/1.0/; https://physionet.org/content/eicu-crd/2.0/; https://physionet.org/content/mimic3-carevue/1.4/; https://physionet.org/content/mimiciii/1.4/).
Abbreviations
- ML:
-
Machine learning
- SIC:
-
Sepsis induced coagulopathy
- MIMIC:
-
Medical information mart for intensive care
- RFE:
-
Recursive feature elimination
- AUROC:
-
Area under the receiver operating characteristic curve
- SHAP:
-
Shapley additive explanations
- LIME:
-
Local interpretable model-agnostic explanations
- SOFA:
-
Sequential organ failure assessment
- ICD:
-
International classification of diseases
- SAPS II:
-
Simplified acute physiology score II
References
Rhodes A, Evans LE, Alhazzani W, et al. Surviving sepsis campaign: international guidelines for management of sepsis and septic shock: 2016. Crit Care Med. 2017;45(3):486–552. https://doi.org/10.1097/CCM.0000000000002255.
Rudd KE, Johnson SC, Agesa KM, et al. Global, regional, and national sepsis incidence and mortality, 1990–2017: Editorials Copyright © 2021 by the Society of Critical Care Medicine and Wolters Kluwer Health, Inc. All Rights Reserved.Critical Care Medicine www.ccmjournal.org 863 analysis for the global burden of disease study. Lancet. 2020;395:200–11. https://doi.org/10.1016/S0140-6736(19)32989-7.
Fleischmann-Struzek C, Mellhammar L, Rose N, et al. Incidence and mortality of hospital- and ICU-treated sepsis: results from an updated and expanded systematic review and meta-analysis. Intensive Care Med. 2020;46:1552–62. https://doi.org/10.1007/s00134-020-06151-x.
Levi M, de Jonge E, van der Poll T. Sepsis and disseminated intravascular coagulation. J Thromb Thrombolysis. 2003;16(1–2):43–7. https://doi.org/10.1023/B:THRO.0000014592.27892.11.
Iba T, Levy JH. Inflammation and thrombosis: Roles of neutrophils, platelets and endothelial cells and their interactions in thrombus formation during sepsis. J Thromb Haemost. 2018;16:231–41. https://doi.org/10.1111/jth.13911.
Song L, Han Z. Research progress on the mechanism and treatment of sepsis related coagulation dysfunction. Chin J Crit Care Med (Electronic Edition). 2017;10:125–9. https://doi.org/10.3877/cma.j.issn.1674-6880.2017.02.013.
Iba T, Nisio MD, Levy JH, et al. New criteria for sepsis-induced coagulopathy (SIC) following the revised sepsis definition: a retrospective analysis of a nationwide survey. BMJ Open. 2017;7:e017046. https://doi.org/10.1136/bmjopen-2017-017046.
Lyons PG, Micek ST, Hampton N, et al. Sepsis-associated coagulopathy severity predicts hospital mortality. Crit Care Med. 2018;46(5):736–42. https://doi.org/10.1097/CCM.0000000000002997.
Iba T, Gando S, Thachil J. Anticoagulant therapy for sepsis-associated disseminated intravascular coagulation: the view from Japan. J Thromb Haemost. 2014;12:1010–9. https://doi.org/10.1111/jth.12596.
Johnson A, Pollard T, Mark R. MIMIC-III clinical database (version 1.4). 2016. PhysioNet. https://doi.org/10.13026/C2XW26.
Johnson A, Bulgarelli L, Pollard T, Horng S, Celi LA, Mark R. MIMIC-IV (version 1.0). 2021. PhysioNet. https://doi.org/10.13026/s6n6-xd98.
Johnson A, Pollard T, Mark R. MIMIC-III clinical database CareVue subset (version 1.4). 2022. PhysioNet. https://doi.org/10.13026/8a4q-w170.
Kernbach JM, Staartjes VE. Foundations of machine learning-based clinical prediction modeling: part II-generalization and overfitting. Acta Neurochir Suppl. 2022;134:15–21. https://doi.org/10.1007/978-3-030-85292-4_3.
Davis J, Goadrich M. The relationship between Precision-Recall and ROC curves. In Proceedings of the 23rd international conference on Machine learning (ICML '06), Association for Computing Machinery, 2006; 233–240. DOI: https://doi.org/10.1145/1143844.1143874
Liu C, Liu X, Mao Z, et al. Interpretable machine learning model for early prediction of mortality in ICU patients with rhabdomyolysis. Med Sci Sports Exerc. 2021. https://doi.org/10.1249/MSS.0000000000002674.
Petch J, Di S, Nelson W. Opening the black box: the promise and limitations of explainable machine learning in cardiology. Can J Cardiol. 2021. https://doi.org/10.1016/j.cjca.2021.09.004.
Štrumbelj E, Kononenko I. Explaining prediction models and individual predictions with feature contributions. Knowl Inform Syst. 2014;41:647–65. https://doi.org/10.1007/s10115-013-0679-x.
Jiang Z, Bo L, Xu Z, et al. An explainable machine learning algorithm for risk factor analysis of in-hospital mortality in sepsis survivors with ICU readmission. Comput Methods Programs Biomed. 2021;204:106040. https://doi.org/10.1016/j.cmpb.2021.106040.
Vellido A. The importance of interpretability and visualization in machine learning for applications in medicine and health care. Neural Comput Appl. 2020;32:18069–83. https://doi.org/10.1007/s00521-019-04051-w.
Uchino S, Bellomo R, Goldsmith D. The meaning of the blood urea nitrogen/creatinine ratio in acute kidney injury. Clin Kidney J. 2012;5(2):187–91. https://doi.org/10.1093/ckj/sfs013.
Gaudry S, Hajage D, Schortgen F, et al. Initiation strategies for renal-replacement therapy in the intensive care unit. N Engl J Med. 2016;375(2):122–33. https://doi.org/10.1056/NEJMoa1603017.
Mikkelsen ME, Miltiades AN, Gaieski DF, et al. Serum lactate is associated with mortality in severe sepsis independent of organ failure and shock. Crit Care Med. 2009;37(5):1670–7. https://doi.org/10.1097/CCM.0b013e31819fcf68.
Lokhandwala S, Andersen LW, Nair S, Patel P, Cocchi MN, Donnino MW. Absolute lactate value vs relative reduction as a predictor of mortality in severe sepsis and septic shock. J Crit Care. 2017;37:179–84. https://doi.org/10.1016/j.jcrc.2016.09.023.
Ryoo SM, Lee J, Lee YS, et al. Lactate level versus lactate clearance for predicting mortality in patients with septic shock defined by sepsis-3. Crit Care Med. 2018;46(6):e489–95. https://doi.org/10.1097/CCM.0000000000003030.
Hernandez G, Bellomo R, Bakker J. The ten pitfalls of lactate clearance in sepsis. Intensive Care Med. 2019;45(1):82–5. https://doi.org/10.1007/s00134-018-5213-x.
Lippi G, Targher G, Montagnana M, et al. Relation between red blood cell distribution width and inflammatory biomarkers in a large cohort of unselected outpatients. Arch Pathol Lab Med. 2009;133(4):628–32. https://doi.org/10.5858/133.4.628.
Zhang L, Yu CH, Guo KP, Huang CZ, Mo LY. Prognostic role of red blood cell distribution width in patients with sepsis: a systematic review and meta-analysis. BMC Immunol. 2020;21(1):40. https://doi.org/10.1186/s12865-020-00369-6.
Jiangquan Fu, Lan Q, Wang D, et al. Predictive value of red cell distribution width on the prognosis of patients with abdominal sepsis. Chin Crit Care Med. 2018;30(3):230–3. https://doi.org/10.3760/cma.j.issn.2095-4352.2018.03.008.
Ling J, Liao T, Wu Y, et al. Predictive value of red blood cell distribution width in septic shock patients with thrombocytopenia: A retrospective study using machine learning. J Clin Lab Anal. 2021;35(12):e24053. https://doi.org/10.1002/jcla.24053.
Wang TH, Hsu YC. Red cell distribution width as a prognostic factor and its comparison with lactate in patients with sepsis. Diagnostics (Basel). 2021;11(8):1474. https://doi.org/10.3390/diagnostics11081474.
Wang AY, Ma HP, Kao WF, Tsai SH, Chang CK. Red blood cell distribution width is associated with mortality in elderly patients with sepsis. Am J Emerg Med. 2018;36(6):949–53. https://doi.org/10.1016/j.ajem.2017.10.056.
Ahmad H, Khan M, Laugle M, et al. Red cell distribution width is positively correlated with atherosclerotic cardiovascular disease 10-year risk score, age, and CRP in spondyloarthritis with axial or peripheral disease. Int J Rheumatol. 2018;2018:2476239. https://doi.org/10.1155/2018/2476239.
Acknowledgements
We thank all participants in the Second Affiliated Hospital of Anhui Medical University and AnHui University.
Funding
This study was supported by a research grant from the National Natural Science Foundation of China (No. 82072134) and the National Natural Science Foundation Youth Science Foundation (No. 81601661).
Author information
Authors and Affiliations
Contributions
ZS, LY and YM designed the study; LZ and MW extracted the data; LY, WZ and XC conducted data quality management and statistical analysis and drafted the manuscript; ZJ and XW participated in the literature search; YM, ZH and HT critically revised the manuscript. All authors contributed to the article and approved the submitted version.
Corresponding authors
Ethics declarations
Ethics approval and consent to participate
MIMIC-III and MIMIC-IV database are publicly available anonymized database, approval for the ethical committee are not necessary.
Consent for publication
Not applicable.
Competing interests
The authors declare that they have no competing interests.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Supplementary Information
Additional file 1: Table S1
. The comparison of baseline demographics and clinical characteristics between surviving patients and those that died in the MIMIC-IV database.
Additional file 2: Table S2
. The comparison of baseline demographics and clinical characteristics between surviving patients and those that died in the eICU-CRD database.
Additional file 3: Table S3
. The details about the SIC diagnostic criteria.
Additional file 4: Figure S1.
The percentage of missing values in MIMIC-III (A), MIMIC-IV (B), and eICU-CRD (C) database.
Additional file 5: Figure S2
. The detailed comparison of the percentage between missing values for each of the 17 factors between survivors and non-survivors in MIMIC-III (A), MIMIC-IV (B), and eICU-CRD (C) database.
Additional file 6: Figure S3
. Decision curve analysis of the XGBoost, LR, SVM, NB, SOFA, SAPS II and SIC in the internal validation set (MIMIC-III, A), MIMIC-IV (B) and eICU-CRD (C) database; calibration curves of each model in the internal validation set (MIMIC-III, D), MIMIC-IV (E) and eICU-CRD (F) database.
Additional file 7: Figure S4
. Feature selection accuracy curve using recursive feature elimination cross-validation. The accuracy get the highest accuracy when the number of variables was 20 (represented as a solid point).
Additional file 8: Figure S5
. The potential interactions between RDW with initial SOFA (A) and age (B). The Y-axis on the left represents the SHAP value of SOFA or age, while Y-axis on the right shows the different values of RDW. Despite SOFA or age being identical, the SHAP value corresponding to different RDW levels may be discrepancies. SOFA = sequential organ failure assessment; RDW = red blood cell distribution width.
Additional file 9: Figure S6
. The interpretation of model prediction results with two actual samples using the SHAP. Patient No.1, who belonged to the "true negative" group, was correctly predicted as a survivor by XGBoost. Patient No.2, who belonged to the "true positive" group, was correctly predicted as a non-survivor. This plot shows significant features contributing to pushing the model output. The blue features decrease the risk of death, while red features promote death.
Additional file 10: Figure S7.
The interpretation of model prediction results with two actual samples using the LIME. The blue box indicated that the features are risk factors for 28-day death, while the orange box suggests the features are protective factors.
Rights and permissions
Open Access This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons licence, and indicate if changes were made. The images or other third party material in this article are included in the article's Creative Commons licence, unless indicated otherwise in a credit line to the material. If material is not included in the article's Creative Commons licence and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this licence, visit http://creativecommons.org/licenses/by/4.0/. The Creative Commons Public Domain Dedication waiver (http://creativecommons.org/publicdomain/zero/1.0/) applies to the data made available in this article, unless otherwise stated in a credit line to the data.
About this article
Cite this article
Zhou, S., Lu, Z., Liu, Y. et al. Interpretable machine learning model for early prediction of 28-day mortality in ICU patients with sepsis-induced coagulopathy: development and validation. Eur J Med Res 29, 14 (2024). https://doi.org/10.1186/s40001-023-01593-7
Received:
Accepted:
Published:
DOI: https://doi.org/10.1186/s40001-023-01593-7