MACHINE LEARNING TECHNIQUES FOR EARLY HEART FAILURE PREDICTION
DOI:
https://doi.org/10.24191/mjoc.v6i2.13708Keywords:
Heart Failure Prediction, Logistic Regression, Naive Bayes, Random Forest, Support Vector MachineAbstract
This paper discusses the performance of four popular machine learning techniques for predicting heart failure using a publicly available dataset from kaggle.com, which are Random Forest (RF), Support Vector Machine (SVM), Naive Bayes (NB), and Logistic Regression (LR). They were selected due to their good performance in medical-related applications. Heart failure is a common public health problem, and there is a need to improve the management of heart failure cases to increase the survival rate. The vast amount of medical data related to heart failure and the availability of powerful computing devices allow researchers to conduct more experiments. The performance of the machine learning techniques was measured by accuracy, precision, recall, f1-score, sensitivity, and specificity in predicting heart failure with 13 symptoms or features. Experimental analysis showed that RF produces the highest performance score, which is 0.88 compared to SVM, NB, and LR. Further experiments with RF were also conducted to determine the important features in predicting heart failure, and the results indicated that all 13 symptoms or features are important.
References
Dwivedi, A.K. (2018). Performance evaluation of different machine learning techniques for prediction of heart disease. Neural Computing & Applications, 29(10), 685–693.
Data Flair. (n.d.). Support Vector Machines Tutorial -Learn to implement SVM in Python. Retrieved from https://data-flair.training/blogs/svm-support-vector-machine-tutorial/
Goldstein, B.A, Navar A.M, Pencina, M.J & Ioannidis, J.P. (2017). Opportunities and challenges in developing risk prediction models with electronic health records data: a systematic review. J Am Med Inform Assoc, 24(1), 198-208.
Handelman, G. S., Kok, H. K., Chandra, R. V., Razavi, A. H., Lee, M. J. & Asadi, H. (2018). eDoctor: machine learning and the future of medicine. J Intern Med, 284(6), 603-619.
Hosmer, D.W Jr & Lemeshow, S. (2004). Applied Logistic Regression, Wiley, New York.
Jordan, M. I. & Mitchell, T. M. (2015). Machine Learning: Trends, perspectives and prospects, Science, 349(6245), 255-260.
Khan, S., Mohd, N. N., Shahzad, A., Ulah, A., Mushtaq, M., Mir, J. & Aamir, M. (2017). Comparative Analysis for Heart Disease Prediction. JOIV: International Journal on Informatics Visualization,1(4-2), 227-231.
Koehrsen, W. (2017, December 28). Random Forest Simple Explanation. Retrieved from https://www.spotx.tv/resources/blog/developer-blog/exploring-random-forest-internal-knowledge-and-converting-the-model-to-table-form/
Mohan, V. (2013). Comparative Analysis of Classification Function Techniques for Heart Disease Prediction. International Journal of Innovative Research in Computer and Communication Engineering, 1(3),735-741.
Pouriyeh, S., Vahid, S., Sannino, G., De Pietro, G., Arabnia, H. & Gutierrez, J. (2017). A comprehensive investigation and comparison of Machine Learning Techniques in the domain of heart disease, 2017. IEEE Symposium on Computers and Communications (ISCC), pp. 204-207
Shahrel, M. Z., Mutalib, S., & Abdul-Rahman, S. (2021). PriceCop-Price Monitor and Prediction Using Linear Regression and LSVM-ABC Methods for E-commerce Platform. International Journal of Information Engineering & Electronic Business, 13(1). 1-14.
Shaikhina, T, Lowe, D., Daga, S., Briggs, D., Higgins, R. & Khovanova, N. (2019). Decision Tree and Random Forest Models for Outcome Prediction in Antibody Incompatible Kidney Transplantation. Biomedical Signal Processing and Contro, 52, 456-462.
Shrivastava, A. K., Singh, H. V., Raizada, A. & Singh, S. K. (2015). C-reactive Protein. Inflammation and Coronary Heart Disease. The Egyptian Heart Journal, 67(2). 89-97.
Steele, A. J., Denaxas, S. C., Shah, A. D., Hemingway, H. & Luscombe, N. M. (2018). Machine Learning models in electronic health records can outperform conventional survival models for predicting patient mortality in coronary artery disease, PLoS One, 13(8), e0202344.
Themistocleous, I., Stefanakis, M., & Douda, H. T. (2017). Coronary heart disease part I: pathophysiology and risk factors. Journal of Physical Activity, Nutrition and Rehabilitation, 167–175.
Thota, L., Nimmala, S., &Manasa, K. (2018). Heart Disease Prediction Using Random Forest Algorithm. Global Journal of Engineering Science and Research, 5(8), 248-252.
Tsangaratos, P., & Ilia, I. (2016). Comparison of a Logistic Regression and Naive Bayes Classifier in Landslide Susceptibility Assessments: The Influence of Models Complexity and Training Dataset Size. CATENA, 145.164-179.
Tripoliti, E. E., Papadopoulos, T. G., Karanasiou, G. S., Naka, K. K. & Fotiadis, D. I., (2017). Heart Failure: Diagnosis, Severity Estimation and Prediction of Adverse Events Through Machine Learning Techniques. Journal of Computational and Structural Biotechnology, 15, 26-47.
Zheng, Y., Guo, X., Qin, J. & Xiao, S. (2015). Computer-assisted diagnosis for chronic heart failure by the analysis of their cardiac reserve and heart sound characteristics. Computer Methods Programs Biomed, I22, 372–83.
Published
Issue
Section
License
Copyright (c) 2021 Nur Shahellin Mansur Huang, Zaidah Ibrahim, Norizan Mat Diah
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
