Predicting Mortality Rate based on Comprehensive Features of Intensive Care Unit Patients

Abstract

Predictive analytics is gaining momentum in health-care since the adoption of electronic health record (EHR) system in hospitals. In particular, machine learning models are built using the critical care EHR data and the information provided during the ICU admissions to predict the mortality of patients admitted in ICU. As per the MIMIC-IV dataset, the survival rate of patients admitted in ICU is found to be 89.76%. This paper proposes a hybrid prediction technique that uses Random Forest and XGBoost for predicting the mortality rate. The proposed techniques performed well in predicting mortality rate despite the class imbalance problem of the dataset. The experiments conducted on MIMIC-IV dataset yields prediction accuracy of 89.72%.