Skip to main content
Springer Nature Link
Log in

A hybrid machine learning approach for early mortality prediction of ICU patients

  • Regular Paper
  • Published:
Progress in Artificial Intelligence Aims and scope Submit manuscript

Abstract

Hospitals face many pressures, including limited budgets and resources. The intensive care unit (ICU), has attracted much attention from the medical community. Patients in the ICU are monitored continuously and their physiological data is consistently measured. This provides a valuable opportunity to analyze significant clinical data. Predicting the mortality of patients in the ICU is crucial. By using this prediction, intensivists identify patients who will not benefit from receiving treatment in the ICU and focus more on the care of patients who will benefit from this treatment. To date, various scoring systems and machine learning models have been developed to predict mortality in intensive care units. In this paper, our goal is to provide a model that can predict mortality for up to 24 h after ICU admission. This research has been conducted using Medical Information Mart for Intensive Care III (MIMIC-III) database. Relevant data have been extracted, preprocessed, and prepared for data mining analysis. The proposed framework has two algorithmic stages. In the first step, time-series data within 24 h of admission is fed to a convolutional neural network with optimized hyperparameters. In the second step, by using the output of the previous stage and a filtering strategy for temporal features, a new data set is created and is fed to an Xgboost algorithm, for final classification. The proposed framework outperforms severity of illness scores and machine learning models within 24 h of admission to the ICU and attains a ROC AUC of 0.863 (± 0.004).

This is a preview of subscription content, log in via an institution to check access.

Access this article

Log in via an institution

Subscribe and save

Springer+ Basic
$34.99 /Month
  • Get 10 units per month
  • Download Article/Chapter or eBook
  • 1 Unit = 1 Article or 1 Chapter
  • Cancel anytime
Subscribe now

Buy Now

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Institutional subscriptions

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7
Fig. 8
Fig. 9
Fig. 10
Fig. 11
Fig. 12

Similar content being viewed by others

Explore related subjects

Discover the latest articles, news and stories from top researchers in related subjects.

References

  1. Hadorn, D.C.: Assessing the Performance of Mortality Prediction Models, vol. 181. Rand, Santa Monica (1993)

    Google Scholar 

  2. Lee, J., Dubin, J.A., Maslove, D.M.: Mortality prediction in the ICU. In: Secondary Analysis of Electronic Health Records, pp. 315–324 (2016)

  3. Le Gall, J.-R., Lemeshow, S., Saulnier, F.: A new simplified acute physiology score (SAPS II) based on a European/North American multicenter study. JAMA 270(24), 2957–2963 (1993)

    Article  Google Scholar 

  4. Ding, Y., Wang, Y., Zhou, D.: Mortality prediction for ICU patients combining just-in-time learning and extreme learning machine. Neurocomputing 281, 12–19 (2018)

    Article  Google Scholar 

  5. Eslami, S., Tarlow, D., Kohli, P., Winn, J.: Just-in-time learning for fast and flexible inference. In: Advances in Neural Information Processing Systems, vol. 27 (2014)

  6. Delen, D., Walker, G., Kadam, A.: Predicting breast cancer survivability: a comparison of three data mining methods. Artif. Intell. Med. 34(2), 113–127 (2005)

    Article  Google Scholar 

  7. Crawford, E.D., Batuello, J.T., Snow, P., Gamito, E.J., McLeod, D.G., Partin, A.W., Stone, N., Montie, J., Stock, R., Lynch, J., et al.: The use of artificial intelligence technology to predict lymph node spread in men with clinically localized prostate carcinoma. Cancer Interdiscip. Int. J. Am. Cancer Soc. 88(9), 2105–2109 (2000)

    Google Scholar 

  8. Ramon, J., Fierens, D., Güiza, F., Meyfroidt, G., Blockeel, H., Bruynooghe, M., Van Den Berghe, G.: Mining data from intensive care patients. Adv. Eng. Inform. 21(3), 243–256 (2007)

    Article  Google Scholar 

  9. Yakovlev, A., Metsker, O., Kovalchuk, S., Bologova, E.: Prediction of in-hospital mortality and length of stay in acute coronary syndrome patients using machine-learning methods. J. Am. Coll. Cardiol. 71(11S), 242–242 (2018)

    Article  Google Scholar 

  10. Davoodi, R., Moradi, M.H.: Mortality prediction in intensive care units (ICUs) using a deep rule-based fuzzy classifier. J. Biomed. Inform. 79, 48–59 (2018)

    Article  Google Scholar 

  11. Sadeghi, R., Banerjee, T., Romine, W.: Early hospital mortality prediction using vital signals. Smart Health 9, 265–274 (2018)

    Article  Google Scholar 

  12. Noei, M., Parvizimosaed, M., Bigdeli, A.S., Yalpanian, M.: A secure hybrid permissioned blockchain and deep learning platform for CT image classification. In: 2022 International Conference on Machine Vision and Image Processing (MVIP), pp. 1–5. IEEE (2022)

  13. Aczon, M., Ledbetter, D., Ho, L., Gunny, A., Flynn, A., Williams, J., Wetzel, R.: Dynamic mortality risk predictions in pediatric critical care using recurrent neural networks. arXiv preprint arXiv:1701.06675 (2017)

  14. Che, Z., Purushotham, S., Cho, K., Sontag, D., Liu, Y.: Recurrent neural networks for multivariate time series with missing values. Sci. Rep. 8(1), 1–12 (2018)

    Article  Google Scholar 

  15. Purushotham, S., Meng, C., Che, Z., Liu, Y.: Benchmarking deep learning models on large healthcare datasets. J. Biomed. Inform. 83, 112–134 (2018)

    Article  Google Scholar 

  16. Caicedo-Torres, W., Gutierrez, J.: ISeeU: Visually interpretable deep learning for mortality prediction inside the ICU. J. Biomed. Inform. 98, 103269 (2019)

    Article  Google Scholar 

  17. Ghorbani, R., Ghousi, R., Makui, A., Atashi, A.: A new hybrid predictive model to predict the early mortality risk in intensive care units on a highly imbalanced dataset. IEEE Access 8, 141066–141079 (2020)

    Article  Google Scholar 

  18. Sánchez-Hernández, F., Ballesteros-Herráez, J.C., Kraiem, M.S., Sánchez-Barba, M., Moreno-García, M.N.: Predictive modeling of ICU healthcare-associated infections from imbalanced data using ensembles and a clustering-based undersampling approach. Appl. Sci. 9(24), 5287 (2019)

    Article  Google Scholar 

  19. Wosiak, A., Zakrzewska, D.: Integrating correlation-based feature selection and clustering for improved cardiovascular disease diagnosis. Complexity 2018 (2018)

  20. YA Amer, A., Vranken, J., Wouters, F., Mesotten, D., Vandervoort, P., Storms, V., Luca, S., Vanrumste, B., Aerts, J.-M.: Feature engineering for ICU mortality prediction based on hourly to bi-hourly measurements. Appl. Sci. 9(17), 3525 (2019)

  21. Mansouri, A., Noei, M., Abadeh, M.S.: Predicting hospital length of stay of neonates admitted to the NICU using data mining techniques. In: 2020 10th International Conference on Computer and Knowledge Engineering (ICCKE), pp. 629– 635. IEEE (2020)

  22. Johnson, A.E., Pollard, T.J., Shen, L., Lehman, L.-W.H., Feng, M., Ghassemi, M., Moody, B., Szolovits, P., Anthony Celi, L., Mark, R.G.: MIMIC-III, a freely accessible critical care database. Sci. Data 3(1), 1–9 (2016)

    Article  Google Scholar 

  23. Quan, H., Sundararajan, V., Halfon, P., Fong, A., Burnand, B., Luthi, J.-C., Saunders, L.D., Beck, C.A., Feasby, T.E., Ghali, W.A.: Coding algorithms for defining comorbidities in ICD-9-CM and ICD-10 administrative data. Med. Care, 1130–1139 (2005)

  24. Mansouri, A., Ordikhani, M., Abadeh, M.S., Tajdini, M.: Predicting serious outcomes in syncope patients using data mining techniques. In: 2019 9th International Conference on Computer and Knowledge Engineering (ICCKE), pp. 409– 413. IEEE (2019)

  25. Noei, M., Abadeh, M.S.: A genetic asexual reproduction optimization algorithm for imputing missing values. In: 2019 9th International Conference on Computer and Knowledge Engineering (ICCKE), pp. 214– 218. IEEE (2019)

  26. Khoobbakht, M., Noei, M., Parvizimosaed, M.: Hybrid flow-rule placement method of proactive and reactive in SDNs. In: 2021 11th International Conference on Computer Engineering and Knowledge (ICCKE), pp. 121– 127. IEEE (2021)

  27. Bommert, A., Sun, X., Bischl, B., Rahnenführer, J., Lang, M.: Benchmark for filter methods for feature selection in high-dimensional classification data. Comput. Stat. Data Anal. 143, 106839 (2020)

    Article  MathSciNet  MATH  Google Scholar 

  28. Schapire, R.E.: Explaining adaboost. In: Empirical Inference, pp. 37– 52. Springer (2013)

  29. Aljamaan, H.I., Elish, M.O.: An empirical study of bagging and boosting ensembles for identifying faulty classes in object-oriented software. In: 2009 IEEE Symposium on Computational Intelligence and Data Mining, pp. 187–194. IEEE (2009)

  30. Chen, T., Guestrin, C.: Xgboost: a scalable tree boosting system. In: Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 785– 794 (2016)

  31. Goldstein, A., Kapelner, A., Bleich, J., Pitkin, E.: Peeking inside the black box: visualizing statistical learning with plots of individual conditional expectation. J. Comput. Graph. Stat. 24(1), 44–65 (2015)

    Article  MathSciNet  Google Scholar 

  32. Lundberg, S.M., Lee, S.-I.: A unified approach to interpreting model predictions. In: Advances in Neural Information Processing Systems, vol. 30 (2017)

Download references

Author information

Authors and Affiliations

  1. Department of Electrical and Computer Engineering, Tarbiat Modares University, Tehran, Iran

    Ardeshir Mansouri & Mohammad Saniee Abadeh

  2. Department of Computer Science, George Mason University, Fairfax, VA, USA

    Mohammadreza Noei

Authors
  1. Ardeshir Mansouri
    View author publications

    You can also search for this author inPubMed Google Scholar

  2. Mohammadreza Noei
    View author publications

    You can also search for this author inPubMed Google Scholar

  3. Mohammad Saniee Abadeh
    View author publications

    You can also search for this author inPubMed Google Scholar

Corresponding author

Correspondence to Ardeshir Mansouri.

Additional information

Publisher's Note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Springer Nature or its licensor holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Mansouri, A., Noei, M. & Saniee Abadeh, M. A hybrid machine learning approach for early mortality prediction of ICU patients. Prog Artif Intell 11, 333–347 (2022). https://doi.org/10.1007/s13748-022-00288-0

Download citation

  • Received:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s13748-022-00288-0

Keywords