Skip to main content

Development of an Explainable Heart Failure Patients Survival Status Prediction Model Using Machine Learning Algorithms

  • Conference paper
  • First Online:
Pan-African Conference on Artificial Intelligence (PanAfriConAI 2023)

Abstract

Heart failure is a condition where insufficient blood circulation occurs throughout the body, causing the heart to become weak or stiff and not function properly. This can lead to congestive heart failure. The life expectancy of a congestive heart failure patients can vary depending on factors such as disease severity, age, genetics, and other variables. The Centers for Disease Control and Prevention (CDC) has reported that almost half of the patients with congestive heart failure survive for more than five years. Previous studies have attempted to predict the survival rate of heart failure patients, but most of them have been limited by unbalanced data sets and a small number of examples. The main goal of this research is to develop a machine learning model that can accurately predict the survival status of heart failure patients while providing an insight into its decision-making process. Machine learning algorithms are a part of artificial intelligence systems and can be used to predict output values based on input data. However, Many machine learning models lack an explanation of how they arrive at their decisions. Therefore, explainable machine learning algorithms are now being used to understand and explain the decision-making process of machine learning models. In this research, we not only use the latest machine learning algorithms for prediction, but we also incorporate explainability features to understand the results. Hyper-parameter optimization techniques are employed to find the best parameter values for better predictive accuracy. Our data set consists of 1334 heart failure patients’ data collected from Felege Hiwot referral hospital and Injibara general hospital. The prediction models used include Decision Tree, Linear Regression, K-Nearest Neighbor, Deep Neural Network, and XGBoost. Prior to conducting the actual experiment, the data set is balanced using the Synthetic Minority Oversampling Technique (SMOTE). Experimental findings indicate that XGBoost performs better than other machine learning techniques, achieving an AUC value of 0.93 using Grid Search HPO after SMOTE.

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

Access this chapter

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

Chapter
USD 29.95
Price excludes VAT (USA)
  • Available as PDF
  • Read on any device
  • Instant download
  • Own it forever

Tax calculation will be finalised at checkout

Purchases are for personal use only

Institutional subscriptions

Similar content being viewed by others

References

  1. Almazroi, A.A.: Survival prediction among heart patients using machine learning techniques. Math. Biosci. Eng. 19(1), 134–145 (2022)

    Article  Google Scholar 

  2. Angaw, D.A., Ali, R., Tadele, A., Shumet, S.: The prevalence of cardiovascular disease in Ethiopia: a systematic review and meta-analysis of institutional and community-based studies. BMC Cardiovasc. Disord. 21(1), 1–9 (2021)

    Article  Google Scholar 

  3. Aslan, M.F., Sabanci, K., Durdu, A.: A CNN-based novel solution for determining the survival status of heart failure patients with clinical record data: numeric to image. Biomed. Signal Process. Control 68, 102716 (2021)

    Article  Google Scholar 

  4. Bergstra, J., Bardenet, R., Bengio, Y., Kégl, B.: Algorithms for hyper-parameter optimization. In: Shawe-Taylor, J., Zemel, R., Bartlett, P., Pereira, F., Weinberger, K. (eds.) Advances in Neural Information Processing Systems, vol. 24. Curran Associates, Inc. (2011). https://proceedings.neurips.cc/paper_files/paper/2011/file/86e8f7ab32cfd12577bc2619bc635690-Paper.pdf

  5. Brochu, E., Cora, V.M., De Freitas, N.: A tutorial on Bayesian optimization of expensive cost functions, with application to active user modeling and hierarchical reinforcement learning. arXiv preprint arXiv:1012.2599 (2010)

  6. Buda, M., Maki, A., Mazurowski, M.A.: A systematic study of the class imbalance problem in convolutional neural networks. Neural Netw. 106, 249–259 (2018)

    Article  Google Scholar 

  7. Chandrashekar, G., Sahin, F.: A survey on feature selection methods. Comput. Electr. Eng. 40(1), 16–28 (2014)

    Article  Google Scholar 

  8. Chicco, D., Jurman, G.: Machine learning can predict survival of patients with heart failure from serum creatinine and ejection fraction alone. BMC Med. Inform. Decis. Mak. 20(1), 1–16 (2020)

    Article  Google Scholar 

  9. Foziljonova, N., Wasito, I.: Prediction of survival rate of heart failure patients using machine learning techniques. J. Theor. Appl. Inf. Technol. 100(9), 2703–2714 (2022)

    Google Scholar 

  10. Guo, X., Yin, Y., Dong, C., Yang, G., Zhou, G.: On the class imbalance problem. In: 2008 Fourth International Conference on Natural Computation, vol. 4, pp. 192–201. IEEE (2008)

    Google Scholar 

  11. Herrera, D., et al.: Association between periodontal diseases and cardiovascular diseases, diabetes and respiratory diseases: consensus report of the Joint Workshop by the European Federation of Periodontology (EFP) and the European Arm of the World Organization of Family Doctors (WONCA Europe). J. Clin. Periodontol. 50(6), 819–841 (2023)

    Article  Google Scholar 

  12. Hussen, N.M., Workie, D.L., Biresaw, H.B.: Survival time to complications of congestive heart failure patients at Felege Hiwot comprehensive specialized referral hospital, Bahir Dar, Ethiopia. PLoS One 17(10), e0276440 (2022)

    Article  Google Scholar 

  13. Ishaq, A., et al.: Improving the prediction of heart failure patients’ survival using smote and effective data mining techniques. IEEE Access 9, 39707–39716 (2021)

    Article  Google Scholar 

  14. Kebede, B., Getachew, M., Molla, Y., Bahiru, B., Dessie, B.: Management, survival, and predictors of mortality among hospitalized heart failure patients at Debre Markos comprehensive specialized hospital, Northwest Ethiopia: prospective cohort study. SAGE Open Med. 9, 20503121211057336 (2021)

    Article  Google Scholar 

  15. Maharana, K., Mondal, S., Nemade, B.: A review: data pre-processing and data augmentation techniques. Global Transitions Proc. 3(1), 91–99 (2022)

    Article  Google Scholar 

  16. Moreno-Sanchez, P.A.: Development of an explainable prediction model of heart failure survival by using ensemble trees. In: 2020 IEEE International Conference on Big Data (Big Data), pp. 4902–4910. IEEE (2020)

    Google Scholar 

  17. Newaz, A., Ahmed, N., Haq, F.S.: Survival prediction of heart failure patients using machine learning techniques. Inform. Med. Unlocked 26, 100772 (2021)

    Article  Google Scholar 

  18. Obaid, H.S., Dheyab, S.A., Sabry, S.S.: The impact of data pre-processing techniques and dimensionality reduction on the accuracy of machine learning. In: 2019 9th Annual Information Technology, Electromechanical Engineering and Microelectronics Conference (IEMECON), pp. 279–283. IEEE (2019)

    Google Scholar 

  19. Özbay Karakuş, M., Er, O.: A comparative study on prediction of survival event of heart failure patients using machine learning algorithms. Neural Comput. Appl. 34(16), 13895–13908 (2022)

    Article  Google Scholar 

  20. Probst, P., Boulesteix, A., Bischl, B.: Tunability: importance of hyperparameters of machine learning algorithms. J. Mach. Learn. Res. 20, 53:1–53:32 (2018). https://api.semanticscholar.org/CorpusID:88515435

  21. Schaer, R., Müller, H., Depeursinge, A.: Optimized distributed hyperparameter search and simulation for lung texture classification in CT using Hadoop. J. Imaging 2(2), 19 (2016)

    Article  Google Scholar 

  22. Stiglic, G., Kocbek, P., Fijacko, N., Zitnik, M., Verbert, K., Cilar, L.: Interpretability of machine learning-based prediction models in healthcare. Wiley Interdisc. Rev. Data Min. Knowl. Discov. 10(5), e1379 (2020)

    Article  Google Scholar 

  23. Tefera, T.A., Muleta, G., Tadesse, K.: Bayesian survival analysis of heart failure patients: a case study in Jimma University Medical Center, Jimma, Ethiopia (2021)

    Google Scholar 

  24. Wistuba, M., Schilling, N., Schmidt-Thieme, L.: Hyperparameter optimization machines. In: 2016 IEEE International Conference on Data Science and Advanced Analytics (DSAA), pp. 41–50. IEEE (2016)

    Google Scholar 

  25. Yibre, A.M., Koçer, B.: Semen quality predictive model using feed forwarded neural network trained by learning-based artificial algae algorithm. Eng. Sci. Technol. Int. J. 24(2), 310–318 (2021)

    Google Scholar 

Download references

Acknowledgments

We want to express our gratitude to Injibara general hospital and Felege Hiwot referral hospital workers.

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Abdulkerim M. Yibre .

Editor information

Editors and Affiliations

Rights and permissions

Reprints and permissions

Copyright information

© 2024 The Author(s), under exclusive license to Springer Nature Switzerland AG

About this paper

Check for updates. Verify currency and authenticity via CrossMark

Cite this paper

Tsehay Demis, B.G., Yibre, A.M. (2024). Development of an Explainable Heart Failure Patients Survival Status Prediction Model Using Machine Learning Algorithms. In: Debelee, T.G., Ibenthal, A., Schwenker, F., Megersa Ayano, Y. (eds) Pan-African Conference on Artificial Intelligence. PanAfriConAI 2023. Communications in Computer and Information Science, vol 2068. Springer, Cham. https://doi.org/10.1007/978-3-031-57624-9_7

Download citation

  • DOI: https://doi.org/10.1007/978-3-031-57624-9_7

  • Published:

  • Publisher Name: Springer, Cham

  • Print ISBN: 978-3-031-57623-2

  • Online ISBN: 978-3-031-57624-9

  • eBook Packages: Computer ScienceComputer Science (R0)

Publish with us

Policies and ethics