An Efficient Machine Learning Based Model To Predict Heart Disease

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.

Heart Disease Prediction, Machine Learning (ML) Techniques, Random Forest Classifier, Python, Supervised Learning

1. 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. [Google Scholar] [Crossref]

2. 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. [Google Scholar] [Crossref]

3. S. Mohan, C. Thirumalai, and G. Srivastava, “Effective heart disease prediction using hybrid machine learning techniques,” IEEE Access, vol. 7, pp. 81542–81554, 2019. [Google Scholar] [Crossref]

4. 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. [Google Scholar] [Crossref]

5. 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. [Google Scholar] [Crossref]

6. 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 [Google Scholar] [Crossref]

7. 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. [Google Scholar] [Crossref]

8. 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. [Google Scholar] [Crossref]

9. 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. [Google Scholar] [Crossref]

10. 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. [Google Scholar] [Crossref]

11. 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. [Google Scholar] [Crossref]

• Investigating the Performance of Colour Oxide on the PS/CNSO Oil Paint Production
• Real-Time Fabric Defect Detection Using a Lightweight Deformable YOLO Network
• Recent Trends in the Stock Market: An Analytical Study of Market Dynamics, Investor Behaviour, and Technological Influence
• Evaluation of Structural Dynamics and Equilibrium State, with Case Study of Large Cantilever Projection for a 10- Storey Reinforced Concrete Building in Lagos, Nigeria.
• Multi-Criteria Evaluation of AI-Based Adaptive Learning Platforms in Global Higher Education: A Fuzzy AHP Perspective