1. A Comparative Machine Learning Modelling Approach for Patients' Mortality Prediction in Hospital Intensive Care Unit
- Author
-
Mahmoud Aldraimli, Nodira Nazyrova, Abdumalik Djumanov, Ikboljon Sobirov, Thierry J. Chaussalet, Sotirov, S.S., Pencheva, T., Kacprzyk, J., Atanassov, K.T., Sotirova, E., and Staneva, G.
- Subjects
Imbalanced Data ,Tomek links ,ICU Mortality Prediction ,MIMIC III ,Ensembles ,Classification ,SMOTE - Abstract
Mortality prediction in a hospital Intensive Care Unit (ICU) is a challenge that must be addressed with high precision. Machine Learning (ML) is a powerful tool in predictive modelling but subject to the problem of class im-balance. In this study, we tackle class imbalance with combining new features, data re-sampling, ensemble learning and an appropriate selection of evaluation metrics in a clinical setting. We built and evaluated 126 ML mod-els to predict mortality in 48546 ICU admissions extracted from the Medical Information Mart for Intensive Care III (MIMIC-III) repository. In our study design, six mortality prediction datasets are extracted; five of which are legacy dataset sets while the remainder is our new constructed dataset. For our combined data models, when testing on isolated data, our selection of features enhanced the prediction performances beyond those for the traditional legacy sets used in research. The legacy datasets are the Simplified Acute Physiology Score (SAPS II), the Sequential Organ Failure Assessment score (SOFA), the Glasgow Coma Scale (GCS), Elixhauser Comorbidity Index (ECI) and Demographics & Disease Groups (DDG). Our approach has a considerable impact on the classification; it resulted in an improvement in the mortality status prediction. For evaluation, we implement a comparative multi-stage evaluation filter for binary classification to compare all models. The best models are identified. The Area Under Receiver Operator Characteristic curves of the tested models range from 0.57 to 0.94. These encouraging results can guide further development of models to allow for more reliable ICU mortality predictions.
- Published
- 2022
- Full Text
- View/download PDF