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.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Almazroi, A.A.: Survival prediction among heart patients using machine learning techniques. Math. Biosci. Eng. 19(1), 134–145 (2022)
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)
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)
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
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)
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)
Chandrashekar, G., Sahin, F.: A survey on feature selection methods. Comput. Electr. Eng. 40(1), 16–28 (2014)
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)
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)
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)
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)
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)
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)
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)
Maharana, K., Mondal, S., Nemade, B.: A review: data pre-processing and data augmentation techniques. Global Transitions Proc. 3(1), 91–99 (2022)
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)
Newaz, A., Ahmed, N., Haq, F.S.: Survival prediction of heart failure patients using machine learning techniques. Inform. Med. Unlocked 26, 100772 (2021)
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)
Ö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)
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
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)
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)
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)
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)
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)
Acknowledgments
We want to express our gratitude to Injibara general hospital and Felege Hiwot referral hospital workers.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2024 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
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)