An Efficient Machine Learning Based Model To Predict Heart Disease
Article Sidebar
Main Article Content
Across the globe, cardiovascular diseases remain a leading contributor to death rates and accurate prediction are essential to modern health systems. Unhealthy lifestyles are one of the elements leading to the increasing occurrence of heart disease, stress and aging, and it has become essential to create a system capable of delivering precise and reliable results diagnosis. With the growing accessibility of vast healthcare data, machine learning technology is emerging as an important tool for helping clinical decision-making by identifying hidden patterns and relationships in complex data sets. In this study, we developed a machine learning-based system for predicting heart disease. The proposed system uses a structured set of data obtained from a publicly available UCI source, contain important medical parameters. To guarantee high data quality and raise the level of performance of models, multiple preprocessing techniques were implemented, including data cleaning, feature normalization, and handling of missing values, classification variable encoding and outlier detection. Different approaches were tested to identify the most effective model. The models were evaluated based on performance indicators such as recall, accuracy, and precision and ROC-AUC points. The study focuses on the performance of ensemble learning using Random Forest, while comparative analysis shows that KNN achieved slightly higher accuracy on the given dataset. K-Nearest Neighbors performed the best, achieving an accuracy of around 91.8% and superior classification capabilities indicated by ROC curves and overall evaluation metrics. Our proposed approach can be used as an effective decision-making tool for medical professionals to identify high-risk patients in time. Finally, this approach helps reduce mortality rates and can assist doctors in early detection and better decision-making.
Downloads
References
N. Biswas, M. M. Ali, M. A. Rahaman, M. Islam, M. R. Mia, S. Azam et al., “Machine learning-based model to predict heart disease in early stage employing different feature selection techniques,” BioMed Research International, vol. 2023, Article ID 6864343, pp. 1–15, 2023.
I. D. Mienye and Y. Sun, “Effective feature selection for improved prediction of heart disease,” in Pan-African Artificial Intelligence and Smart Systems Conference, pp. 94–107, Springer, Cham, 2021.
S. Mohan, C. Thirumalai, and G. Srivastava, “Effective heart disease prediction using hybrid machine learning techniques,” IEEE Access, vol. 7, pp. 81542–81554, 2019.
M. S. Amin, Y. K. Chiam, and K. D. Varathan, “Identification of significant features and data mining techniques in predicting heart disease,” Telematics and Informatics, vol. 36, pp. 82–93, 2019.
C. B. C. Latha and S. C. Jeeva, “Improving the accuracy of pre diction of heart disease risk based on ensemble classification techniques,” Informatics in Medicine Unlocked, vol. 16, article 100203, 2019.
V. V. Ramalingam, A. Dandapath, and M. K. Raja, “Heart dis ease prediction using machine learning techniques : a survey,” International Journal of Engineering & Technology, vol. 7, no. 2.8, pp. 684–687, 2018
M. F. Rabbi, M. P. Uddin, M. A. Ali et al., “Performance evaluation of data mining classification techniques for heart dis ease prediction,” American Journal of Engineering Research, vol. 7, no. 2, pp. 278–283, 2018.
S. Pouriyeh, S. Vahid, G. Sannino, G. De Pietro, H. Arabnia, and J. Gutierrez, “A comprehensive investigation and comparison of machine learning techniques in the domain of heart disease,” in 2017 IEEE Symposium on Computers and Communications (ISCC), pp. 204–207, Heraklion, Greece, 2017.
J. Patel, D. Tejal Upadhyay, and S. Patel, “heart disease prediction using machine learning and data mining technique,” heart disease, vol. 7, no. 1, pp. 129–137, 2015.
D. Tomar and Agarwal, “Feature selection based least square twin support vector machine for diagnosis of heart disease,” International Journal of Bio-Science and Bio-Technology, vol. 6, no. 2, pp. 69–82, 2014.
M. Buscema, M. Breda, and W. Lodwick, “Training with Input Selection and Testing (TWIST) Algorithm: A Significant Advance in Pattern Recognition Performance of Machine Learning,” Journal of Intelligent Learning Systems and Applications, vol. 5, no. 1, article 27937, 2013.

This work is licensed under a Creative Commons Attribution 4.0 International License.
All articles published in our journal are licensed under CC-BY 4.0, which permits authors to retain copyright of their work. This license allows for unrestricted use, sharing, and reproduction of the articles, provided that proper credit is given to the original authors and the source.