Supervised machine learning for the early prediction of acute respiratory distress syndrome (ARDS)

Abstract

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 when tested for the detection of ARDS at onset and 0.827, 0.810, and 0.790 for the prediction of ARDS at 12-, 24-, and 48-h windows prior to onset, respectively.

Conclusion: Supervised machine learning predictions may help predict patients with ARDS up to 48 h prior to onset.

Keywords: Acute respiratory distress syndrome; Clinical decision support systems; Electronic health records; Intensive care unit; Machine learning; Medical informatics.