Supervised Machine Learning for the Early Prediction of Acute Respiratory Distress Syndrome (ARDS)

Purpose: Acute respiratory distress syndrome (ARDS) is a serious respiratory condition with high mortality and associated morbidity. The objective of this study is to develop and evaluate a novel application of gradient boosted tree models trained on patient health record data for the early prediction of ARDS. Materials and Methods: 9919 patient encounters were retrospectively analyzed from the Medical Information Mart for Intensive Care III (MIMIC-III) data base. XGBoost gradient boosted tree models for early ARDS prediction were created using routinely collected clinical variables and numerical representations of radiology reports as inputs. XGBoost models were iteratively trained and validated using 10-fold cross validation. Results: On a hold-out test set, algorithm classifiers attained area under the receiver operating characteristic curve (AUROC) values of 0.905, 0.827, 0.810, and 0.790 when tested for the prediction of ARDS at 0-, 12-, 24-, and 48-hour windows prior to onset, respectively. Conclusion: Supervised machine learning predictions may help predict patients with ARDS up to 48 hours prior to onset.