Heart Failure Prediction using Machine learning with Metaheuristic feature selection techniques
Keywords:
Machine Learning, Genetic Algorithm, Feature Selection, Ant Colony Optimization, Particle Swarm OptimizationSynopsis
This is a Chapter in:
Book:
Competitive Tools, Techniques, and Methods
Print ISBN 978-1-6692-0008-6
Online ISBN 978-1-6692-0007-9
Series:
Chronicle of Computing
Chapter Abstract:
Heart failure is a major complication of cardiovascular diseases (CVDs), which are the leading cause of mortality globally. However, early treatment and detection of heart failure may increase the chance of survival. Using current clinical data, machine learning (ML) algorithms present an effective approach for predicting heart failure. Utilizing ML algorithms and feature selection using metaheuristic methods, we present a novel framework in this research for the prediction of heart failure. We employ several ML algorithms and perform feature selection using four meta-heuristic techniques: Grey Wolf Optimization (GWO), Particle Swarm Optimization (PSO), Genetic Algorithm (GA), and Ant Colony Optimization (ACO). The performance of each combination is evaluated and compared based on the F-score metric, which we aim to maximize. (PSO) has achieved the highest when choosing the relevant features which increased the overall accuracy from 0.83 to 0.90. The results indicate that our proposed framework can effectively identify relevant features and improve the predictive performance of ML algorithms for heart failure. Furthermore, we provide a comprehensive comparison of the meta-heuristic techniques, highlighting their advantages and limitations for feature selection in heart failure prediction.
Cite this paper as:
Elgendya O., Nassifa A. B., Soudana B., (2024) Heart Failure Prediction using Machine learning with Metaheuristic feature selection techniques. In: Tiako P.F. (ed) Competitive Tools, Techniques, and Methods. Chronicle of Computing. OkIP. AHIT24#15. https://doi.org/10.55432/978-1-6692-0007-9_4
Presented at:
The 2024 OkIP International Conference on Advances in Health Information Technology (AHIT) in Oklahoma City, Oklahoma, USA, and Online, on October 3, 2024.
Contact:
Omar Elgendya
oelgendy@sharjah.ac.ae
References
C. W. Tsao et al., “Heart Disease and Stroke Statistics—2023 Update: A Report From the American Heart Association,” Circulation, vol. 147, no. 8, pp. E93–E621, Feb. 2023, doi:10.1161/CIR.0000000000001123.
J. Xu, X. Luo, G. Wang, H. Gilmore, and A. Madabhushi, “A Deep Convolutional Neural Network for segmenting and classifying epithelial and stromal regions in histopathological images,” Neurocomputing, vol. 191, pp. 214–223, May 2016, doi: 10.1016/j.neucom.2016.01.034.
G. Motalleb, “Artificial neural network analysis in preclinical breast cancer,” Cell J., vol. 15, no. 4, pp. 324–331, Dec. 2014, Accessed: Jan. 14, 2021. [Online]. Available: /pmc/articles/PMC3866536/?report=abstract
M. I. Jordan and T. M. Mitchell, “Machine learning: Trends, perspectives, and prospects,” Science (80-. )., vol. 349, no. 6245, pp. 255–260, Jul. 2015, doi: 10.1126/SCIENCE.AAA8415.
L. P. Kaelbling, M. L. Littman, and A. W. Moore, “Reinforcement Learning: A Survey,” J. Artif. Intell. Res., vol. 4, pp. 237–285, May 1996, doi: 10.1613/JAIR.301.
S. N. Sivanandam and S. N. Deepa, “Genetic Algorithm Optimization Problems,” in Introduction to Genetic Algorithms, Springer Berlin Heidelberg, 2007, pp. 165–209. doi: 10.1007/978-3-540-73190-0_7.
J. Kennedy and R. Eberhart, “Particle swarm optimization,” Proc. ICNN’95 - Int. Conf. Neural Networks, vol. 4, pp. 1942–1948, doi: 10.1109/ICNN.1995.488968.
N. A. Abujabal and A. B. Nassif, “Meta-heuristic algorithms-based feature selection for breast cancer diagnosis: A systematic review,” Int. Conf. Electr. Comput. Commun. Mechatronics Eng. ICECCME 2022, 2022, doi: 10.1109/ICECCME55909.2022.9988285.
M. Dorigo, M. Birattari, and T. Stutzle, “Ant colony optimization,” IEEE Comput. Intell. Mag., vol. 1, no. 4, pp. 28–39, Nov. 2006, doi: 10.1109/MCI.2006.329691.
M. Dorigo and G. Di Caro, “Ant colony optimization: A new meta-heuristic,” in Proceedings of the 1999 Congress on Evolutionary Computation, CEC 1999, IEEE Computer Society, 1999, pp. 1470–1477. doi: 10.1109/CEC.1999.782657.
D. Karaboga and B. Basturk, “Artificial Bee Colony (ABC) optimization algorithm for solving constrained optimization problems,” Lect. Notes Comput. Sci. (including Subser. Lect. Notes Artif. Intell. Lect. Notes Bioinformatics), vol. 4529 LNAI, pp. 789–798, 2007, doi: 10.1007/978-3540-72950-1_77/COVER.
D. Karaboga, “Artificial bee colony algorithm,” Scholarpedia, vol. 5, no. 3, p. 6915, 2010, doi: 10.4249/scholarpedia.6915.
T. M. Le, T. N. Pham, and S. V. T. Dao, “A Novel Wrapper-Based Feature Selection for Heart Failure Prediction Using an Adaptive Particle Swarm Grey Wolf Optimization,” Stud. Fuzziness Soft Comput., vol. 410, pp. 315–336, 2021, doi: 10.1007/978-3-030-70111-6_15/COVER.
B. Soudan, F. F. Dandachi, and A. Bou Nassif, "Attempting cardiac arrest prediction using artificial intelligence on vital signs from electronic health records," Smart Health, vol. 25, pp. 100294, 2022.
S. Irin Sherly and G. Mathivanan, “An efficient honey badger based Faster region CNN for chronc heart Failure prediction,” Biomed. Signal Process. Control, vol. 79, p. 104165, Jan. 2023, doi: 10.1016/J.BSPC.2022.104165.
D. Chicco and L. Oneto, “An Enhanced Random Forests Approach to Predict Heart Failure from Small Imbalanced Gene Expression Data,” IEEE/ACM Trans. Comput. Biol. Bioinforma., vol. 18, no. 6, pp. 2759–2765, 2021, doi: 10.1109/TCBB.2020.3041527.
S. J. Khiabani, A. Batani, and E. Khanmohammadi, “A hybrid decision support system for heart failure diagnosis using neural networks and statistical process control,” Healthc.
Anal., vol. 2, no. September, p. 100110, 2022, doi:
1016/j.health.2022.100110.
M. M. Yaqoob, M. Nazir, M. A. Khan, S. Qureshi, and A. Al-Rasheed, “Hybrid Classifier-Based Federated Learning in Health Service Providers for Cardiovascular Disease Prediction,” Appl. Sci., vol. 13, no. 3, 2023, doi: 10.3390/app13031911.
P. K. Sahoo and P. Jeripothula, “Heart Failure Prediction Using Machine Learning Techniques,” SSRN Electron. J., Dec. 2020, doi: 10.2139/SSRN.3759562.
autoencoders,” Measurement, vol. 205, p. 112166, Dec. 2022, doi: 10.1016/J.MEASUREMENT.2022.112166
A. Noor, L. Ali, H. T. Rauf, U. Tariq, and S. Aslam, “An integrated decision support system for heart failure prediction based on feature
transformation using grid of stacked
