Artificial Intelligence-Based Ensemble Model for Rapid Prediction of Heart Disease

被引：0
作者：
Harika N. [1 ,4 ]
Swamy S.R. [2 ]
Nilima [3 ,4 ]
机构：
[1] I Syneos Health, Hyderabad
[2] Director, Product Management MaxisIT, ClinAsia, Hyderabad
[3] Department of Biostatistics, All India Institute of Medical Sciences, Delhi
[4] Department of Statistics, Manipal Academy of Higher Education, Karnataka, Manipal
来源：
SN Computer Science | 2021年 / 2卷 / 6期
关键词：
Ensemble; Heart disease; Naïve Bayes; Neural networks; Prediction; Rapid; Support vector machine;
D O I：
10.1007/s42979-021-00829-9
中图分类号：
学科分类号：
摘要：
Heart disease is the leading cause of mortality among men and women. Accurate and rapid diagnosis of heart disease will assist in saving many lives. To develop a novel ensemble framework based on heterogeneous classifiers namely support vector machine (SVM), Naïve Bayes (NB), and artificial neural networks (ANN) for rapid prediction of heart disease. The present study also verifies the most accurate algorithm among all three. Data are collected from the UCI machine learning repository. After pre-processing, the data were divided into training and test data in a ratio of 80:20. Using the training data, the three contributing algorithms were trained by providing heart disease status. The algorithms were tested with the unseen data instances and hence evaluated for accuracy. The ensemble technique uses the results from individual classifiers and yields a result based on majority voting method. The ensemble model was observed to predict heart disease with an accuracy of 87.05% followed by ANN (84.74%), NB (81.35%) and SVM (79.66%). Among the individual classifiers, ANN had the least miss-classification rate and performed best in terms of all other model diagnostics. The use of the proposed ensemble classifier is recommended to predict the heart condition to have better accuracy and least miss-classification. © 2021, The Author(s), under exclusive licence to Springer Nature Singapore Pte Ltd.
引用
收藏
相关论文
共 27 条
  • [1] Mitchell R., Michalski J., Carbonell T., An artificial intelligence approach, (2013)
  • [2] Kirubha V., Priya S.M., Survey on data mining algorithms in disease prediction, Int J Comput Trends Tech, 38, 3, pp. 24-128, (2016)
  • [3] Olaniyi E.O., Oyedotun O.K., Adnan K., Heart diseases diagnosis using neural networks arbitration, Int J Intell Syst Appl, 7, 12, (2015)
  • [4] Patel J., TejalUpadhyay D., Patel S., Heart disease prediction using machine learning and data mining technique, Heart Dis, 7, 1, pp. 129-137, (2015)
  • [5] Gawali M., Shirwalkar N., Kalshetti A., Heart disease prediction system using data mining techniques, Int J Pure Appl Math, 120, 6, pp. 499-506, (2018)
  • [6] Dankwa S., Zheng W., Special issue on using machine learning algorithms in the prediction of kyphosis disease: a comparative study, Appl Sci, 9, 16, (2019)
  • [7] Arunachalam S., Cardiovascular disease prediction model using machine learning algorithms, Int J Res Appl Sci, 8, pp. 1006-1019, (2020)
  • [8] Jerez-Aragones J.M., Gomez-Ruiz J.A., Ramos-Jimenez G., Munoz-Perez J., Alba-Conejo E., A combined neural network and decision trees model for prognosis of breast cancer relapse, Artif Intell Med, 27, 1, pp. 45-63, (2003)
  • [9] Abdar M., Yen N.Y., Hung J.C.-S., Improving the diagnosis of liver disease using multilayer perceptron neural network and boosted decision trees, J Med Biol Eng, 38, 6, pp. 953-965, (2018)
  • [10] Bhatla N., Jyoti K., An analysis of heart disease prediction using different data mining techniques, Int J Eng, 1, 8, pp. 1-4, (2012)
123