Development of a Machine Learning Model for Predicting Weaning Outcomes Based Solely on Continuous Ventilator Parameters during Spontaneous Breathing Trials

Abstract

Discontinuing mechanical ventilation remains challenging. We developed a machine learning model to predict weaning outcomes using only continuous monitoring parameters obtained from ventilators during spontaneous breathing trials (SBTs). Patients who received mechanical ventilation in the medical intensive care unit at a tertiary university hospital from 2019–2021 were included in this study. During the SBTs, three waveforms and 25 numerical data were collected as input variables. The proposed convolutional neural network (CNN)-based weaning prediction model extracts features from input data with diverse lengths. Among 138 enrolled patients, 35 (25.4%) experienced weaning failure. The dataset was randomly divided into training and test sets (8:2 ratio). The area under the receiver operating characteristic curve for weaning success by the prediction model was 0.912 (95% confidence interval [CI], 0.795–1.000), with an area under the precision-recall curve of 0.767 (95% CI, 0.434–0.983). Furthermore, we used gradient-weighted class activation mapping technology to provide visual explanations of the model’s prediction, highlighting influential features. This tool can assist medical staff by providing intuitive information regarding readiness for extubation without requiring any additional data collection other than SBT data. The proposed predictive model can assist clinicians in making ventilator weaning decisions in real time, thereby improving patient outcomes.

Keywords

• ventilator weaning
• machine learning
• extubation
• prediction
• visual explanation