Background and aims
The timing of extubation is a difficult decision for the medical team on the PICU. With negative impact on patient outcome when extubating too late or too early. The aim of this study was to create machine learning models for extubation failure prediction
...
Background and aims
The timing of extubation is a difficult decision for the medical team on the PICU. With negative impact on patient outcome when extubating too late or too early. The aim of this study was to create machine learning models for extubation failure prediction after surgery in patients with congenital heart disease. The goal was to assess the influence of time variant features on the performance.
Methods
Data from post cardiac surgery patients admitted to the PICU of the University Medical Centre Utrecht, The Netherlands, between 2009 and 2018 was collected. Ventilator and monitor parameters were extracted in 12-hour segments. Different representations of time-variant features were calculated (per hour/ per 12-hour segment), these representations were tested against machine learning trained on only time-invariant features (age, weight diagnosis). Machine learning algorithms tested were: long short-term memory network (LSTM), logistic regression and random forest model. Models were evaluated by comparing the areas under the receiver operator curves
Results
With only time invariant features a performance of 75% [95%CI 81%-90%] using logistic regression. Adding the time-variant features to a LSTM model a performance was reached 77% [95%CI 80%-90%]. Important features from the logistic regression models were age, weight, heart rate and respiratory rate.
Conclusions
Based on the overall results we concluded that the chosen representations of time variant features did not significantly improve the performance of the models. To improve performance and implementation of machine learning models in the future, transparent and externally validated models need to be developed.