Predicting Heart Disease Using Supervised Machine Learning Techniques: A Comparative Analysis

  • Madan Lal
  • Ali Akbar
Keywords: Cardiovascular Diseases Prediction, Diagnosis, Machine Learning, Deep Learning

Abstract

Early diagnosis of diseases can improve patient outcomes and increase the chances of successful treatment.
One of the biggest causes of death worldwide is cardiovascular disease. Deep learning models have recently been
shown to be quite accurate at doing this task, and machine learning techniques are increasingly being used to predict
cardiac illness. The supervised learning algorithms KNN, Random Forest, Logistic Regression, SVM, and deep
learning model artificial neural networks are all compared in this research, for the prediction of heart disease. We used
a publicly available dataset of Cleveland Heart Disease Dataset on heart disease to train and test the models as well as
compare their performance in terms of various accuracy metrics. Random Forest got highest accuracy with 92.17%
and Logistic Regression with 88.4%, KNN with 90.0% and SVM with 90.08%, while deep learning model
outperformed with 98.92% accuracy. Our results show that across all models, Random Forest has the highest accuracy,
while deep learning models beat other supervised learning techniques in terms of overall accuracy. Additionally, we
developed a web based model and integrated the model with web based for further analysis and research purposes. We
learn that the best model to use relies on the specifics of the task and the available data and that mixing different
models might lead to even better performance gains. Our study clarifies the advantages and disadvantages of different
machine learning methods for predicting heart disease, and it may aid in the development of more accurate and reliable
prediction systems for use in clinical settings.

References

[1] Cardiovascular diseases. Source: WHO(https://www.who.int/health-topics/cardiovascular-diseases#tab=tab_1)
[2] Meshref, H. Cardiovascular disease diagnosis: A machine learning interpretation approach. International Journal of
Advanced Computer Science and Applications, 2019 10(12).[Google Scholar]
[3] Mr.E.Loganathan, I.T.Saranraj, G.Vijayakumar, & S.Sowndharya. CARDIAC DISEASES PREDICTION USING
SVM WITH XG BOOST ALGORITHM. International Journal Of Advance Research And Innovative Ideas In
Education, (2023).9(2), 333-340.
[4] Rishabh Magar et al. 'HEART DISEASE PREDICTION USING MACHINE LEARNING", International Journal of
Emerging Technologies and Innovative Research (www.jetir.org), ISSN:2349-5162, Vol.7, Issue 6, page no.2081-
2085, June-2020.
[5] Jindal, H., Agrawal, S., Khera, R., Jain, R., & Nagrath, P. Heart disease prediction using machine learning algorithms.
In IOP conference series: materials science and engineering (Vol. 1022, No. 1, p. 012072). (2021).IOP Publishing.
[Google Scholar]
[6] Sharma, C., Shambhu, S., Das, P., & Jain, S. Features Contributing Towards Heart Disease Prediction Using Machine
Learning. In Workshop on Advances in Computational Intelligence at ISIC (2021).(pp. 84-92). [Google Scholar]
[7] Bharti, R., Khamparia, A., Shabaz, M., Dhiman, G., Pande, S., & Singh, P. Prediction of heart disease using a
combination of machine learning and deep learning. Computational intelligence and neuroscience, 2021. [Google
Scholar]
[8] Bertsimas, D., Mingardi, L., & Stellato, B. Machine learning for real-time heart disease prediction. IEEE Journal of
Biomedical and Health Informatics, 25(9),(2021). 3627-3637.[Google Scholar]
[9] Adler, E. D., Voors, A. A., Klein, L., Macheret, F., Braun, O. O., Urey, M. A., ... & Yagil, A. Improving risk prediction
in heart failure using machine learning. European journal of heart failure, 22(1), (2020). 139-147. [Google Scholar]
[10] Salhi, D. E., Tari, A., & Kechadi, M. T. (2021). Using machine learning for heart disease prediction. In Advances in
Computing Systems and Applications: Proceedings of the 4th Conference on Computing Systems and Applications
(pp. 70-81). Springer International Publishing. [Google Scholar]
Published
2023-11-22
How to Cite
Lal, M., & Akbar, A. (2023). Predicting Heart Disease Using Supervised Machine Learning Techniques: A Comparative Analysis. Journal of Software Engineering, 2(1), 74-84. Retrieved from http://sjhse.smiu.edu.pk/sjhse/index.php/SJHSE/article/view/58