Explainable machine learning to predict long-term mortality in critically ill ventilated patients: a retrospective
Ming-Cheng Chan1,2, Kai-Chih Pai3, Shao-An Su4
1Division of Critical Care and Respiratory Therapy, Department of Internal Medicine, Taichung Veterans General Hospital, Taichung, Taiwan.
Background:
Machine learning (ML) model is increasingly used to predict short-term outcome in critically ill patients, but the study for long-term outcome is sparse. We used explainable ML approach to establish 30-day, 90-day and 1-year mortality prediction model in critically ill ventilated patients.
Methods:
We retrospectively included patients who were admitted to intensive care units during 2015-2018 at a tertiary hospital in central Taiwan and linked with the Taiwanese nationwide death registration data. Three ML models, including extreme gradient boosting (XGBoost), random forest (RF) and logistic regression (LR), were used to establish mortality prediction model. Furthermore, we used feature importance, Shapley Additive exPlanations (SHAP) plot, partial dependence plot (PDP), and local interpretable model-agnostic explanations (LIME) to explain the established model.
Results:
We enrolled 6994 patients and found the accuracy was similar among the three ML models, and the area under the curve value of using XGBoost to predict 30-day, 90-day and 1-year mortality were 0.858, 0.839 and 0.816, respectively. The calibration curve and decision curve analysis further demonstrated accuracy and applicability of models. SHAP summary plot and PDP plot illustrated the discriminative point of APACHE (acute physiology and chronic health exam) II score, haemoglobin and albumin to predict 1-year mortality. The application of LIME and SHAP force plots quantified the probability of 1-year mortality and algorithm of key features at individual patient level.
Conclusions:
We used an explainable ML approach, mainly XGBoost, SHAP and LIME plots to establish an explainable 1-year mortality prediction ML model in critically ill ventilated patients.
Related Concept Videos
01:29Mechanical Ventilation I: Indication and Settings
01:23Mechanical Ventilation II: Invasive Ventilation
Negative-Pressure Ventilators
Negative-pressure ventilators create a vacuum around the chest or body to draw air into the lungs, simulating breathing. This method does not require an...
01:21Acute Respiratory Failure-II
The underlying physiological abnormalities that contribute to hypoxemic respiratory failure include:
01:20Assessment of Ventilation I: Respiratory Rate
A Ventilation assessment is critical for monitoring a patient's health status. Respiration, one of the most accessible vital signs, provides insights into the function of numerous body systems and can indicate serious health issues, such as brainstem injuries from head trauma.
Critical Guidelines for Assessing Ventilation:
01:19Factors Affecting Pulmonary Ventilation
Alveolar Surface Tension
The alveolar fluid lines the luminal surface of the alveoli and exerts a force called surface tension. This force is caused by the polar water molecules in the liquid being more strongly attracted to each...