Prediction of prolonged mechanical ventilation in the intensive care unit via machine learning: a COVID-19 perspective

Early recognition of risk factors for prolonged mechanical ventilation (PMV) could allow for early clinical interventions, prevention of secondary complications such as nosocomial infections, and effective triage of hospital resources. This study tested the hypothesis that an ensemble machine learning (ML) analysis of clinical data at time of intubation could identify patients at risk of PMV, using a COVID-19 dataset to classify patients into PMV (> 14 days) and non-PMV (≤ 14 days) groups. While several factors are known to cause PMV, including acid-base, weakness, and delirium, lesser-utilized but routinely measured parameters such as platelet count, glucose levels and fevers may also be relevant. Patient data from a single University Hospital were analyzed via the ML workflow to predict patients at risk of PMV and identify key clinical markers. Model performance was evaluated on a chronologically distinct cohort. The ML workflow identified patients at risk of PMV with AUROC_TRAIN=0.960 (F1_TRAIN = 0.935) and AUROC_TEST=0.804 (F1_TEST = 0.800). Top key features for classification included glucose, platelet count, temperature, LVEF, bicarbonate (arterial blood gas), and BMI. Data analysis at intubation time via the proposed workflow offers the potential to accurately predict patients at risk of PMV, with the goal to improve patient management and triage of hospital resources.