A New Multi-Agent Feature Wrapper Machine Learning Approach for Heart Disease Diagnosis
Abstract
Heart disease (HD) is a serious widespread life-threatening disease. The heart of patients with HD fails to pump sufficient amounts of blood to the entire body. Diagnosing the occurrence of HD early and efficiently may prevent the manifestation of the debilitating effects of this disease and aid in its effective treatment. Classical methods for diagnosing HD are sometimes unreliable and insufficient in analyzing the related symptoms. As an alternative, noninvasive medical procedures based on machine learning (ML) methods provide reliable HD diagnosis and efficient prediction of HD conditions. However, the existing models of automated ML-based HD diagnostic methods cannot satisfy clinical evaluation criteria because of their inability to recognize anomalies in extracted symptoms represented as classification features from patients with HD. In this study, we propose an automated heart disease diagnosis (AHDD) system that integrates a binary convolutional neural network (CNN) with a new multi-agent feature wrapper (MAFW) model. The MAFW model consists of four software agents that operate a genetic algorithm (GA), a support vector machine (SVM), and Naïve Bayes (NB). The agents instruct the GA to perform a global search on HD features and adjust the weights of SVM and BN during initial classification. A final tuning to CNN is then performed to ensure that the best set of features are included in HD identification. The CNN consists of five layers that categorize patients as healthy or with HD according to the analysis of optimized HD features. We evaluate the classification performance of the proposed AHDD system via 12 common ML techniques and conventional CNN models by using a cross-validation technique and by assessing six evaluation criteria. The AHDD system achieves the highest accuracy of 90.1%, whereas the other ML and conventional CNN models attain only 72.3%–83.8% accuracy on average. Therefore, the AHDD system proposed herein has the highest capability to identify patients with HD. This system can be used by medical practitioners to diagnose HD efficiently.
Keywords
Heart disease; machine learning; multi-agent feature wrapper model; heart disease diagnosis; HD cleveland datasets; convolutional neural network
References
[1] A. L. Bui, T. B. Horwich and G. C. Fonarow, “Epidemiology and risk prole of heart failure,” Nature Reviews Cardiology, vol. 8, no. 1, pp. 30, 2011.
[2] P. A. Heidenreich, J. G. Trogdon, O. A. Khavjou, J. Butler, K. Dracup et al., “Forecasting the future of
cardiovascular disease in the united states: A policy statement from the American heart association,”Circulation, vol. 123, no. 8, pp. 933–944, 2011.
[3] M. Durairaj and N. Ramasamy, “A comparison of the perceptive approaches for preprocessing the data set for predicting fertility success rate,” International Journal of Control Theory and Applications, vol.9, no. 27, pp. 255–260, 2016.
[4] J. Mourao-Miranda, A. L. Bokde, C. Born, H. Hampel and M. Stetter, “Classifying brain states and determining the discriminating activation patterns: Support vector machine on functional MRI data, NeuroImage, vol. 28, no. 4, pp. 980–995, 2005.
[5] S. Ghwanmeh, A. Mohammad and A. Al-Ibrahim, “Innovative articial neural networks-based decision support system for heart diseases diagnosis,” Journal of Intelligent Learning Systems and Applications, vol. 5, no. 3, pp. 8, 2013.
[6] I. B. Salah, R. De la Rosa, K. Ouni and R. B. Salah, “Automatic diagnosis of valvular heart diseases by impedance cardiography signal processing,” Biomedical Signal Processing and Control, vol. 57, pp. 101758, 2020.
[7] V. L. Roger, “The heart failure epidemic,” International Journal of Environmental Research and Public Health, vol. 7, no. 4, pp. 1807–1830, 2010.
[8] K. Vanisree and J. Singaraju, “Decision support system for congenital heart disease diagnosis based on signs and symptoms using neural networks,” International Journal of Computer Applications, vol. 19 no. 6, pp. 6–12, 2011.
[9] A. Adeli and M. Neshat, “A fuzzy expert system for heart disease diagnosis,” in Proc. of Int. Multi Conf. of Engineers and Computer Scientists, Hong Kong, vol. 1, pp. 28–30, 2010.
[10] M. A. Mohammed, M. K. A. Ghani, N. A. Arunkumar, O. I. Obaid, S. A. Mostafa et al., “Genetic case-based reasoning for improved mobile phone faults diagnosis,” Computers & Electrical Engineering,vol. 71, pp. 212–222, 2018.
[11] A. Methaila, P. Kansal, H. Arya and P. Kumar, “Early heart disease prediction using data mining techniques,” Computer Science & Information Technology Journal, vol. 8, no. 2, pp. 53–59, 2014.
[12] O. W. Samuel, G. M. Asogbon, A. K. Sangaiah, P. Fang and G. Li, “An integrated decision support system based on ANN and fuzzy_AHP for heart failure risk prediction,” Expert Systems with Applications, vol. 68, pp. 163–172, 2017.
[13] P. K. Anooj, “Clinical decision support system: Risk level prediction of heart disease using weighted fuzzy rules and decision tree rules,” Central European Journal of Computer Science, vol. 1, no. 4, pp. 482–498, 2011.
[14] P. C. Austin, J. V. Tu, J. E. Ho, D. Levy and D. S. Lee, “Using methods from the datamining and machine-learning literature for disease classication and prediction: A case study examining classication of heart failure subtypes,” Journal of Clinical Epidemiology, vol. 66, no. 4,pp. 398–407, 2013.
[15] Y. Xing, J. Wang and Z. Zhao, “Combination data mining methods with new medical data to predicting outcome of coronary heart disease,” in Int. Conf. on Convergence Information Technology, Gyeongju, South Korea, pp. 868–872, 2