Development and Performance Evaluation of a Heart Disease Prediction Model Using Convolutional Neural Network

Main Article Content

Adebimpe Esan
Juwon Akingbade
Adetunji Omoniji
Adedayo Sobowale
Tomilayo Adebiyi

Abstract

Heart disease is a leading cause of mortality globally and its prevalence is increasing year after year. Recent statistics from the World Health Organization show that about 17.9 million individuals are embattled with heart diseases annually and people under the age of 70 account for one-third of these deaths. Hence, there is need to intensify research on early heart disease prediction and artificial intelligence-based heart disease prediction systems. Previous heart disease prediction systems using machine learning techniques are unable to manage large amount of data, resulting in poor prediction accuracy. Hence, this research employs Convolutional Neural Networks, a deep learning approach for prediction of heart diseases. The dataset for training and testing the model was obtained from a government owned hospital in Nigeria and Kaggle. The resulting system was evaluated using precision, recall, f1-score and accuracy metrics. The results obtained are: 0.94, 0.95, 0.95 and 0.95 for precision, recall, f1-score and accuracy respectively. This show that the CNN-based model responded very well to the prediction of heart diseases for both negative and positive classes. The results obtained were also compared to some selected machine-learning models like Random Forest, Naïve Bayes, KNN and Logistic Regression and results show that the developed model achieved a significant improvement over the methods considered. Therefore, convolutional neural network is more suitable for heart disease prediction than some state-of-the-art machine-learning models. The contribution to knowledge of this research is the use of Afrocentric dataset for heart disease prediction. Future research should consider increasing the data size for model training to achieve improved accuracy.

Downloads

Download data is not yet available.

Article Details

How to Cite
[1]
A. Esan, J. Akingbade, A. Omoniji, A. Sobowale, and T. Adebiyi, “Development and Performance Evaluation of a Heart Disease Prediction Model Using Convolutional Neural Network”, AJERD, vol. 7, no. 1, pp. 35–40, Mar. 2024.
Section
Articles

References

Reddy, V. K., Raju, K.P., Kumar, M. J, Sujatha, C.H., & Prakash, P. R. (2016). Prediction of Heart Diseases Using Decision Tree Approach. International Journal of Advanced Research in Computer Science and Software Engineering, 6(3), 530-532.

Yamala, S. (2020). Prediction of Heart Diseases Using Support Vector Machine. International Journal for Research in Applied Science and Engineering Technology (IJRASET), 8 (2), 126-135. DOI: https://doi.org/10.22214/ijraset.2020.2021

Uma, N.D. (2018) Prediction System for Heart Disease using Naïve Bayes and particle swarm Optimization. Biomedical Research, 29 (12), 2646-2649. DOI: https://doi.org/10.4066/biomedicalresearch.29-18-620

Hakak, S., Alazab, M., Khan, S., Gadekallu, T. R., Maddikunta, P. K. R., & Khan, W. Z. (2021). An ensemble machine learning approach through effective feature extraction to classify fake news. Future Generation Computer Systems, 117, 47-58. DOI: https://doi.org/10.1016/j.future.2020.11.022

Zhu, W., Xie, L., Han, J., & Guo, X. (2020). The Application of Deep Learning in Cancer Prognosis Prediction. Cancers, 12(3), 603. DOI: https://doi.org/10.3390/cancers12030603

Youness, K. & Mohamed, B. (2018). Heart Disease Prediction and Classification Using Machine Learning Algorithms Optimized by Particle Swarm Optimization and Ant Colony Optimization. International Journal of Intelligent Engineering and Systems, 12(1), 242-252. DOI: https://doi.org/10.22266/ijies2019.0228.24

Sumit, S. & Mahesh, P. (2020). Heart Diseases Prediction using Deep Learning Neural Network Model. International Journal of Innovative Technology and Exploring Engineering (IJITEE), 9 (3), 2244-2248. DOI: https://doi.org/10.35940/ijitee.C9009.019320

Rohit, B., Aditya, K., Mohammad, S., Gaurav, D., Sagar, P., & Parneet, S. (2021). Prediction of Heart Disease Using a Combination of Machine Learning and Deep Learning. Comput Intell Neurosci, 2021, 1-11. doi: 10.1155/2021/8387680. DOI: https://doi.org/10.1155/2021/8387680

Mienye, I. D., Sun, Y., & Wang, Z. (2020). An improved ensemble learning approach for the prediction of heart disease risk. Informatics in Medicine Unlocked, 20(1), 1-5. DOI: https://doi.org/10.1016/j.imu.2020.100402

Polaraju K. & Durga Prasad, D. (2017). Prediction of Heart Disease using Multiple Linear Regression Model. International Journal of Engineering Development and Research, 5(4), 1419-1425.

Pandita, A., Vashisht, S., Tyagi, A. & Yadav, S. (2021). Prediction of Heart Diseases using Machine Learning Algorithms. International Journal for Research in Applied Science and Engineering Technology, 9(5), 2422-2429. DOI: https://doi.org/10.22214/ijraset.2021.3412

Akella, A, & Akella, S. (2021). Machine Learning Algorithms for Predicting Coronary Artery Diseases: Efforts toward an open source Solution. Future Science OA, 7(6), 698-706. DOI: https://doi.org/10.2144/fsoa-2020-0206

Miao, F., Cai, Y. P., Zhang, Y. X., Fan, X. M., & Li, Y. (2018). Predictive modeling of hospital mortality for patients with heart failure by using an improved random survival forest. IEEE Access, 6(1), 7244-7253. DOI: https://doi.org/10.1109/ACCESS.2018.2789898

Ravindhar, N. V., Anand, H., Shanmugasundaram, R., & Winster, G. (2019). Intelligent Diagnosis of Cardiac Disease Prediction using Machine Learning. International Journal of Innovative Technology and Exploring Engineering (IJITEE), 8(11), 1417-1421. DOI: https://doi.org/10.35940/ijitee.J9765.0981119

Gavhane, A., Kokkula, G., Pandya, I., & Devadkar, K. (2018). Prediction of heart disease using machine learning. Second International conference on Electronics, Communication and Aerospace Technology (ICECA), 1275-1278, DOI:10.1109/ICECA.2018.8474922 DOI: https://doi.org/10.1109/ICECA.2018.8474922

Yadav, D. C., & Pal, S. (2020). Prediction of heart disease using feature selection and random forest ensemble method. International Journal of Pharmaceutical Research, 12(4), 56-66. DOI: https://doi.org/10.31838/ijpr/2020.12.04.013

Ghosh, P., Azam, S., Jonkman, M., Karim, A., Shamrat, M., Ignatious, E., Shultana, S., & Beeravolu, A. (2021). Efficient prediction of cardiovascular disease using machine learning algorithms with Relief and LASSO feature selection techniques. IEEE Access, 9(1), 19321-19330. DOI: https://doi.org/10.1109/ACCESS.2021.3053759

Shah, D., Patel, S.B., & Bharti, S.K. (2020). Heart Disease Prediction using Machine Learning Techniques. SN Computer Science, 1(6). 1-10 DOI: https://doi.org/10.1007/s42979-020-00365-y