AI Chat Paper
Note: Please note that the following content is generated by AMiner AI. SciOpen does not take any responsibility related to this content.
{{lang === 'zh_CN' ? '文章概述' : 'Summary'}}
{{lang === 'en_US' ? '中' : 'Eng'}}
Chat more with AI
PDF (4.8 MB)
Collect
Submit Manuscript AI Chat Paper
Show Outline
Outline
Show full outline
Hide outline
Outline
Show full outline
Hide outline
Research Article | Open Access | Online First

Heart Disease Detection: A Comprehensive Analysis of Machine Learning, Ensemble Learning, and Deep Learning Algorithms

Haseeb Khan1( )Ahmad Bilal2Muhmmad Aqeel Aslam3Hira Mustafa2
School of Mechanical Engineering, Kyungpook National University, Republic of Korea
Department of Electrical and Computer Engineering, Habib University, Pakistan
Department of Electrical Engineering, GIFT University Gujranwala, Pakistan
Show Author Information

Graphical Abstract

Abstract

Heart disease is a major health concern impacting a significant number of individuals. Early detection is crucial for effective treatment and management. Conventional techniques for the detection of heart diseases are time-consuming, inconvenient, expensive, and unsuitable for diagnosis. To take advantage of innovative approaches such as artificial intelligence (AI), this paper presents the detection of heart diseases at an early stage and enhances the accuracy of AI models for predictions, because accurate identification is crucial in the early stages, if not detected promptly, it may lead to death. Two methods are employed for the early detection of heart disease. In the first approach, traditional machine learning, ensemble learning, and artificial neural networks are utilized. In the second approach, a hybridization approach is applied to machine and ensemble learning algorithms to boost model performance. The heart_statlog_cleveland_hungary_final dataset is utilized and split using k-fold cross-validation with a value of k set at 10. The performance metrics such as accuracy, sensitivity, specificity, precision, and F1 score are calculated. The results indicate that the hybrid technique, namely Bagging combined with random forest (RF), emerges as the top performer, boasting the highest average accuracy of 94.34%, average specificity of 93.7%, average sensitivity of 93.5%, average precision 94%, and an average F1 score of 94.2%. In conclusion, the hybrid approach of Bagging with RF would be better for detecting heart disease at an early stage.

References

[1]
WebMD Editorial Contributors. Heart Disease: Types, Causes, and Symptoms. 2023. Available at https://www.webmd.com/heart-disease/heart-disease-types-causes-symptoms
[2]
World Health Organization. Cardiovascular diseases. 2011. Available at https://www.who.int/health-topics/cardiovascular-diseases#tab=tab_1
[3]
British Heart Foundation. Heart and Circulatory Disease Statistics 2022. Available at https://www.bhf.org.uk/-/media/files/for-professionals/research/heart-statistics/bhf-statistics-compendium-2022.pdf
[4]
Mayoclinic. Heart Disease. 2022. Available at https://www.mayoclinic.org/diseases-conditions/heart-disease
[5]
J.X. Chen, G.C. Xi, Y.W. Xing, et al. Predicting syndrome by NEI specifications: A comparison of five data mining algorithms in coronary heart disease. In: Proceedings of the International Conference on Life System Modeling and Simulation, 2007: 129–135.
[6]
S. Pouriyeh, S. Vahid, G. Sannino, et al. A comprehensive investigation and comparison of Machine Learning Techniques in the domain of heart disease. In: Proceedings of the 2017 IEEE Symposium on Computers and Communications (ISCC), 2017: 204–207.
[7]
A. Singh, R. Kumar. Heart disease prediction using machine learning algorithms. In: Proceedings of the 2020 International Conference on Electrical and Electronics Engineering (ICE3), 2020: 452–457.
[8]
S. Kusuma, K.R. Jothi. Cardiovascular disease prediction and comparative analysis of varied classifier techniques. In: Proceedings of 2021 2nd Global Conference for Advancement in Technology (GCAT), 2021: 1–7.
[9]

A. Methaila, P. Kansal, H. Arya, et al. Early heart disease prediction using data mining techniques. Journal of Computer Science &Information Technology, 2014, 24: 53−59. https://doi.org/10.5121/csit.2014.4807

[10]

R. Aggrawal, S. Pal. Multi-machine learning binary classification, feature selection and comparison technique for predicting death events related to heart disease. International Journal of Pharmaceutical Research, 2020, 13(1): 428−439. https://doi.org/10.31838/ijpr/2021.13.01.080

[11]
D. Ayon, S. Jyoti, S. Neeta. Analysis of supervised machine learning algorithms for heart disease prediction with reduced number of attributes using principal component analysis. International Journal of Computer Applications, 2016, 140(2): 27–31.
[12]
M.C. Tu, D. Shin, D. Shin. Effective diagnosis of heart disease through bagging approach. In: Proceedings of 2009 2nd International Conference on Biomedical Engineering and Informatics, 2009: 1–4.
[13]

V. Shorewala. Early detection of coronary heart disease using ensemble techniques. Informatics in Medicine Unlocked, 2021, 26: 100655. https://doi.org/10.1016/j.imu.2021.100655

[14]
Y.K. Singh, N. Sinha, S.K. Singh. Heart disease prediction system using random forest. In: Proceedings of the International Conference on Advances in Computing and Data Sciences, 2017: 613–623.
[15]
J. Emakhu, S. Shrestha, S. Arslanturk. Prediction system for heart disease based on ensemble classifiers. In: Proceedings of the 5th NA International Conference on Industrial Engineering and Operations Management, 2020: 10–14.
[16]

R. Katarya, S.K. Meena. Machine learning techniques for heart disease prediction: A comparative study and analysis. Health and Technology, 2021, 11: 87−97. https://doi.org/10.1007/s12553-020-00505-7

[17]
N.S. Rajliwall, R. Davey, G. Chetty. Cardiovascular risk prediction based on XGBoost. In: Proceedings of the 2018 5th Asia-Pacific World Congress on Computer Science and Engineering (APWC on CSE), 2018: 246–252.
[18]

X.Y. Zhang, M. Wang, W. Wei, et al. An accurate diagnosis of coronary heart disease by Catboost, with easily accessible data. Journal of Physics:Conference Series, 2021, 1955(1): 012027. https://doi.org/10.1088/1742-6596/1955/1/012027

[19]
B.V. Chowdary, J. Kedarnath, R. Vyshnavi, et al. An effective and efficient heart disease prediction model using distributed high performance light GBM. In: Proceedings of the 2021 Fifth International Conference on I-SMAC (IoT in Social, Mobile, Analytics and Cloud) (I-SMAC), 2021: 662–667.
[20]
J.S. Sonawane, D.R. Patil. Prediction of heart disease using multilayer perceptron neural network. In: Proceedings of the International Conference on Information Communication and Embedded Systems (ICICES2014), 2014: 1–6.
[21]

T.T. Hasan, M.H. Jasim, I.A. Hashim. Heart disease diagnosis system based on multi-layer perceptron neural network and support vector machine. International Journal of Current Engineering and Technology, 2017, 7(5): 1842−1853.

[22]

M. Akhil Jabbar, B.L. Deekshatulu, P. Chandra. Classification of heart disease using artificial neural network and feature subset selection. Global Journal of Computer Science and Technology Neural &Artificial Intelligence, 2013, 13(3): 5−14.

[23]
Heart Disease Cleveland UCI. 2021. Available at https://www.kaggle.com/cherngs/heart-disease-cleveland-uci
[24]

J.P. Li, A.U. Haq, S.U. Din, et al. Heart disease identification method using machine learning classification in E-healthcare. IEEE Access, 2020, 8: 107562−107582. https://doi.org/10.1109/access.2020.3001149

[25]
M. Shouman, T. Turner, R. Stocker. Using decision tree for diagnosing heart disease patients. In: Proceedings of the Ninth Australasian Data Mining Conference, 2011, 121: 23–30.
[26]

Puvar, Pruthvirajsinh and Patel Neel et al. Heart disease detection using ensemble learning approach. IOP Conference Series:Materials Science and Engineering, 2021, 1022: 012046. https://doi.org/10.1088/1757-899X/1022/1/012046

[27]
F. Bulut. Heart attack risk detection using Bagging classifier. In: Proceedings of the 2016 24th Signal Processing and Communication Application Conference (SIU), 2016: 2013–2016.
[28]
M.O.G. Nayeem, M.N. Wan, M.K. Hasan. Prediction of disease level using multilayer perceptron of artificial neural network for patient monitoring. International Journal of Soft Computing and Engineering (IJSCE), 2015, 5(4): 17–23.
Nano Biomedicine and Engineering
Cite this article:
Khan H, Bilal A, Aslam MA, et al. Heart Disease Detection: A Comprehensive Analysis of Machine Learning, Ensemble Learning, and Deep Learning Algorithms. Nano Biomedicine and Engineering, 2024, https://doi.org/10.26599/NBE.2024.9290087

330

Views

38

Downloads

0

Crossref

0

Scopus

Altmetrics

Received: 01 January 2024
Revised: 16 March 2024
Accepted: 14 April 2024
Published: 17 July 2024
© The Author(s) 2024.

This is an open-access article distributed under  the  terms  of  the  Creative  Commons  Attribution  4.0 International  License (CC BY) (http://creativecommons.org/licenses/by/4.0/), which  permits  unrestricted  use,  distribution,  and reproduction in any medium, provided the original author and source are credited.

Return