An HMM-ensemble approach to predict severity progression of ICU treatment for hospitalized COVID-19 patients

COVID-19-related pneumonia requires different modalities of Intensive Care Unit (ICU) interventions at different times to facilitate breathing, depending on severity progression. The ability for clinical staff to predict how patients admitted to hospital will require more or less ICU treatment on a daily basis is critical to ICU management. For real datasets that are sparse and incomplete and where the most important state transitions (dismissal, death) are rare, a standard Hidden Markov Model (HMM) approach is insufficient, as it is prone to overfitting. In this paper we propose a more sophisticated ensemble-based approach that involves training multiple HMMs, each specialized in a subset of the state transitions, and then selecting the more plausible predictions either by selecting or combining the models. We have validated the approach on a live dataset of about 1, 000 patients from a partner hospital. Our results show that rare events, as well as the transitions to the most severe treatments outperform state of the art approaches.