Tri-Hybrid Naive Bayes Classification Model for Early Cardiovascular Disease Detection

  • Zulkiflu Umar Department of Computer Science, Kebbi State University of Science and Technology, Aleiro, Nigeria
  • Danlami Gabi Department of Computer Science, Kebbi State University of Science and Technology, Aleiro, Nigeria
  • Nurudeen M. Ibrahim
Keywords: Heart disease prediction, Data mining, Naïve Bayes, Neural network, Machine learning.

Abstract

Cardiovascular disease is a major global health problem affecting people around the globe. Although modern medicine has contributed in providing significant data to help mitigate cardiovascular disease, however, there is still needs to provide an ideal solution that will help in detecting early cardiovascular diseases. As a result, existing researchers have proposed several hybrid cardiovascular disease detection techniques using both machine learning and deep learning approaches. However, most solution requires improvement especially in term of accuracy detection. In this paper, a tri-hybrid Naïve Bayes classification algorithm for accuracy prediction of early cardiovascular disease detection is developed. In the developed algorithm, the strength of the conventional Naïve Bayes classification algorithm is improved with that of decision tree approach to achieve better accuracy. To further enhance its accuracy, Neural Network (NN) procedure is later incorporated into the improved Naïve Bayes classification algorithm. Implementation of the developed tri-hybrid algorithm is carried out on Waikato Environment for Knowledge Analysis (WEKA) and evaluated using dataset from the kaggle website that contain instances of 1025 and 14 attributes. Experimental results based on confusion matrix for performance evaluation indicates the developed tri-hybrid Naïve Bayes algorithm has achieved high classification accuracy and recall rates of 98.54 and 99% respectively when compared with that of the benchmarked schemes. The performance of the proposed solution deployed can help cardiologist make better prediction in the diagnosis of the heart disease. Although several machine learning algorithms as seen in the literatures are trained on historical data, these algorithms may not be able to accurately predict the risk of heart disease in patients who are exposed to new or emerging risk factors.  Further research is to experiments a wider range of algorithms with a build in model that can detect early cardiovascular diseases as well improve accuracy.

References

Adebimpe, M. M. (2019). Prediction Of Heart Disease Using Bayesian Network Model. MSc Thesis, June 2019. https://repository.aust.edu.ng/xmlui/bitstream/handle/123456789/4903/Muibideen%20Mistura.pdf?sequence=1&isAllowed=y

Annepu, D., and Gowtham, G. (2019). Cardiovascular disease prediction using machine learning techniques. International Research Journal of Engineering and Technology, 6 (4): 3963–3971.

D. Gabi, D., Ismail, A. S., Anazida. Z. and Zalmiyah. Z. (2019). Quality of service task scheduling algorithm for time-cost trade off scheduling problem in cloud computing environment. International Journal of Intelligent Systems Technologies and Applications, 8(1): 469.

Hayatu, H. I., Mohammed, A., Barroon, A., Ali, Y. S., and Mohammed, U. S. (2020). Feature Relevance Analysis and Classification of Kaduna State Road Traffic Accident Data using Machine Learning Techniques. International Journal of Information Processing and Communication, 10(1):86–98.

Jahangiri, S. (2023). An Improved Naïve Bayes Approach to Diagnose Cardiovascular Disease : A Case Study. 0–36.

Kumar, N. K., Sindhu, G. S., Prashanthi, D. K., and Sulthana, A. S. (2020). Analysis and Prediction of Cardio Vascular Disease using Machine Learning Classifiers. Ml, 15–21.

Mutyala, N. K. (2018). Prediction of Heart Diseases Using Data Mining and Machine Learning Algorithms and Tools. International Journal of Scientific Research in Computer Science, Engineering and Information Technology, 3(3):2456-3307.

Nashif, S. R. (2018). Heart disease detection by using machine learning algorithms. World Journal of Engineering and Technology, 1(6): 854–873.

Padmanabhan, M., Yuan, P., Chada, G., and Nguyen, H. V. (2019). Physician-friendly machine learning: a case study with cardiovascular disease risk prediction. Journal of Clinical Medicine. 8(7): 1050.

Prasad, R., Anjali, P., Adil, S. and Deepa, N. (2019). Heart disease prediction using logistic regression algorithm using machine learning. International Journal of Engineering and Advanced Technology, 8 (35): 659–662.

Pujianto, U., Luki, A., Ar, H., and Mohammad, A. M. (2019). Comparison of Naïve Bayes Algorithm and Decision Tree C4 . 5 for Hospital Readmission Diabetes Patients using HbA1c Measurement. 2(2): 58–71.

Raihan, M., Mondal, S., More, A., Boni, P. K., and Sagor, M. F. (2017). Smartphone based heart attack risk prediction system with statistical analysis and data mining approaches. Advances in Science, Technology and Engineering Systems Journal, 2 (3): 1815–1822.

Reddy, P. K., Reddy, T. S., Balakrishnan, S., Basha, S. M., & Poluru, R. K. (2019). Heart disease prediction using machine learning algorithm. International Journal of Innovative Technology and Exploring, 8 (10):2603–2606.

Repaka, A. N. (2019). Design And Implementing Heart Disease Prediction Using Naives Bayesian. 2019 3rd International Conference on Trends in Electronics and Informatics (ICOEI), pp. 292–297.

Rubini, P. E., Subasini, C. A., Vanitha Katharine, A., Kumaresan, V., Gowdhamkumar, S., and Nithya, T. M. (2021). A cardiovascular disease prediction using machine learning algorithms. Annals of the Romanian Society for Cell Biology, 25(2): 904–912.

Shah, Devansh, S., Patel1, S., and Santosh, K. B. (2020). Heart Disease Prediction using Machine Learning Techniques, SN Computer Science (2020) 1:345.

Sridhar, A. and Kapardhi, A. (2018). Predicting heart disease using machine learning algorithm. International Research Journal of Engineering and Technology, 6 (4): 36-38.

Subhadra, K. and Vikas, B. (2019). Neural network based intelligent system for predicting heart disease. International Journal of Innovative Technology and Exploring Engineering (IJITEE), 8(5): 484-487.

Tougui, I., Jilbab, A. and El Mhamdi, J. (2020). Heart disease classification using data mining tools and machine learning techniques. Health and Technology, 10(5):1137–1144.

Usman, I., Aliyu, S., Isma, B. and Yusuf, A. (2019). A Comparative Study of Base Classifiers in Predicting Students’ Performance Based on Interaction with LMS Platform. FUDMA Journal of Sciences (FJS), 3(1):231–239.

Yahaya, L., David O. N. and Joshua G. E. (2020). A Comprehensive Review on Heart Disease Prediction Using Data Mining and Machine Learning Techniques. American Journal of Artificial Intelligence, 4(1): 20-29.

Yusuf, A. B. ., Dima, R. M. and Aina, S. K. (2021). Optimized Breast Cancer Classification using Feature Selection and Outliers Detection. Journal of the Nigerian Society of Physical Sciences, 3(4): 298–307.

Published
2023-06-30