Stacking Model for Heart Stroke Prediction using Machine Learning Techniques

Subasish Mohapatra; Indrani Mishra; Subhadarshini Mohanty

doi:10.4108/eetpht.9.4057

Authors

Subasish Mohapatra Odisha University of Technology and Research
Indrani Mishra Odisha University of Technology and Research
Subhadarshini Mohanty Odisha University of Technology and Research

DOI:

https://doi.org/10.4108/eetpht.9.4057

Keywords:

Feature Selection, Heart Stroke, Machine Learning, Random Forest, RF, Navies Bayes, NB, Logistic Regression, LR, Decision Tree, DT, Adaboost, K- Nearest- Neighbor, Gradient Boosting, GB

Abstract

The paper presents an adaptive model that utilized the machine learning algorithms to predict the heart diseases. As heart disease is one of the leading causes of death and understanding its mechanism, effective prevention, diagnosis, and treatment is very crucial. With the help of data analytics, machine learning, artificial intelligence, it is possible to provide optimal solution to the heart diseases. But still getting optimal accuracy is a challenging issue. Identifying the data pattern, correlation and algorithms affects the accuracy very much. In this work, a stacking model has been proposed to find the best models out of it and validate the model for better prediction accuracy. The model is stacked with seven algorithms different machine learning algorithms such as Radom Forest, Naïve Bayes, Linear Regression, Decision Tree, Ad boost, K Nearest Neighbour, and Gradient Boosting. The experiment was carried out with a training and testing ration of 80:20 in ration. Evaluations are carried out in different measures such as Precision, Recall, F Score, and Accuracy to demonstrate the efficiency of the algorithms. Form the experimentation it is observed that the gradient boosting outperforms the other competitive approaches as this algorithm combines weak predictive models to form a stronger ensemble model that can make highly accurate predictions with an accuracy of 94.67 percentages.

Downloads

Download data is not yet available.

References

World Health Organization. Cardiovascular diseases (CVDs);Retrieved from: https://www.who.int/news-room/fact-sheets/detail/cardiovascular-diseases-(cvds) (2019).

Géron, A. Hands-on machine learning with Scikit-Learn, Keras, and TensorFlow: Concepts, tools, and techniques to build intelligent systems. O'Reilly Media; (2019).

Chollet, F. Deep learning with Python. Manning Publications; (2017).

Géron, A. Building machine learning systems with TensorFlow. Packt Publishing; (2017).

Reddy, S., et al. "Heart disease prediction using a random forest algorithm." Journal of Medical Systems. (2020);44(9): 1-10.

Elazab, A., et al. "Heart disease risk level identification based on clustering techniques." Journal of Ambient Intelligence and Humanized Computing. (2020); 11(9):4407-4422..

Liu, W., et al. "Deep learning-based coronary artery disease prediction." Journal of Medical Systems.(2021);45(3): 1-11.

Subramanya, R., Sierla, S., & Vyatkin, V., From DevOps to MLOps: Overview and application to electricity market forecasting. Applied Sciences,(2022) 12(19), 9851. DOI: https://doi.org/10.3390/app12199851

Jiang, Z., et al. "Applying MLOps to Heart Disease Prediction: A Case Study." Proceedings of the International Conference on Machine Learning (ICML), (2020); 235-243.

Chen, M., et al. "Random Forest for Heart Disease Prediction." Journal of Healthcare Engineering. (2020); 11(3): 123-137.

Jiang, B., Cao, H., Hu, J., & Li, Z. Application of machine learning operations in the development of a heart disease prediction model. Frontiers in Cardiovascular Medicine.(2020);7: 73.

Chen, M., Zhang, Z., Li, L., Huang, L., & Zou, J. A random forest model for predicting heart disease risk. BMC medical informatics and decision making. (2020); 20(1): 1-8. DOI: https://doi.org/10.1186/s12911-020-01358-w

Kaur, H., Chawla, R., & Jain, A. Prediction of heart disease using decision tree. International Journal of Advanced Research in Computer Science. (2020); 11(2): 32-38.

Nandhini, R., Muralidharan, V., & Sivakumar, R. Prediction of heart disease using logistic regression and decision tree. International Journal of Engineering & Technology.(2018);7(4.35): 51-54.

M. Marimuthu, P. Latha and K. M. Mehata, "K-Nearest Neighbor Algorithm for Prediction of Heart Disease using Clinical Data,"International Conference on Computer Communication and Informatics, Coimbatore. 2012; pp. 1-4.

Purushothama, B. R., Raghavendra, R., & Venugopal, K. R. Heart disease prediction using Naïve Bayes. International Journal of Engineering and Technology.(2017); 9(4):3224-3229.