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

Bioengineering (Basel). 2023 Oct 5;10(10):1163. doi: 10.3390/bioengineering10101163.

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: extubation; machine learning; prediction; ventilator weaning; visual explanation.

Grants and funding

This research was supported by the MSIT (Ministry of Science and ICT), Korea, under the ITRC (Information Technology Research Center) support program (IITP-2023-2020-0-01461) supervised by the IITP (Institute for Information & communications Technology Planning & Evaluation) and this work was supported by the Korea Medical Device Development Fund grant funded by the Korea government (the Ministry of Science and ICT, the Ministry of Trade, Industry and Energy, the Ministry of Health & Welfare, the Ministry of Food and Drug Safety) (Project Number: RS-2020-KD000095).