Abstract

Can Machine Learning Methods Predict Extubation Outcome in Premature Infants as well as Clinicians?

Martina Mueller, Jonas S Almeida, Romesh Stanislaus and Carol L Wagner

Rationale: Though treatment of the prematurely born infant breathing with assistance of a mechanical ventilator has much advanced in the past decades, predicting extubation outcome at a given point in time remains challenging. Numerous studies have been conducted to identify predictors for extubation outcome; however, the rate of infants failing extubation attempts has not declined. Objective: To develop a decision-support tool for the prediction of extubation outcome in premature infants using a set of machine learning algorithms. Methods: A dataset assembled from 486 premature infants on mechanical ventilation was used to develop predictive models using machine learning algorithms such as Artificial Neural Networks (ANN), Support Vector Machine (SVM), Naïve Bayesian Classifier (NBC), Boosted Decision Trees (BDT), and Multivariable Logistic Regression (MLR). Performance of all models was evaluated using Area Under the Curve (AUC). Results For some of the models (ANN, MLR and NBC) results were satisfactory (AUC: 0.63-0.76); however, two algorithms (SVM and BDT) showed poor performance with AUCs of ~0.5. Conclusion: Clinician’s predictions still outperform machine learning due to the complexity of the data and contextual information that may not be captured in clinical data used as input for the development of the machine learning algorithms. Inclusion of preprocessing steps in future studies may improve the performance of prediction models.