banner
Home > Open Journal Systems

A precise coronary artery disease prediction using Boosted C5.0 decision tree model

Surjeet Dalal, Umesh Kumar Lilhore, Sarita Simaiya, Vivek Jaglan, Anand Mohan, Sachin Ahuja, Akshat Agrawal, Martin Margala, Prasun Chakrabarti

Abstract


In coronary artery disease, plaque builds up in the arteries that carry oxygen-rich blood to the heart. Having plaque in the arteries can constrict or impede blood flow, leading to a heart attack. Shortness of breath and soreness in the chest are common symptoms. Lifestyle modifications, medication, and potentially surgery are all options for treatment. In coronary artery disease, plaque builds up in the arteries that carry oxygen-rich blood to the heart. Having plaque in the arteries can constrict or impede blood flow, leading to a heart attack. Shortness of breath and soreness in the chest are common symptoms. Lifestyle modifications, medication, and potentially surgery are all options for treatment. This paper presents a Hybrid Boosted C5.0 model to predict coronary artery disease more precisely. A Hybrid Boosted C5.0 model is formed by combining the C5.0 decision tree and boosting methods. Boosting is a supervised machine learning method that leverages numerous inadequate models to construct a more robust and powerful model. The proposed model and some well-known existing machine learning models, i.e., decision tree, AdaBoost, and random forest, were implemented using an online coronary artery disease dataset of 6611 patients and compared based on various performance measuring parameters. Experimental analysis shows that the proposed model achieved an accuracy of 91.62% at training and 81.33% at the testing phase. The AUC value achieved in the training and testing phase is 0.957 and 0.88, respectively. The Gini value achieved in the training and testing phase is 0.914 and 0.759, respectively, far better than the proposed method

Keywords


machine learning; C5.0 decision tree algorithm; coronary artery disease; prediction; over-fitting; boosting

Full Text:

PDF

References


1. Wang G, Gao Y, Xu F, et al. GW28-e0388 A novel machine-learning model for identification of significant coronary artery disease. Journal of the American College of Cardiology 2017; 70(16): C113. doi: 10.1016/j.jacc.2017.07.400

2. Stuckey T, Singh N, Goswami R, et al. TCT-177 Assessing coronary artery disease by cardiac phase tomography using machine-learned algorithms in obese and elderly subjects. Journal of the American College of Cardiology 2017; 70(18): B75–B76. doi: 10.1016/j.jacc.2017.09.245

3. Griffin WF, Choi AD, Riess JS, et al. AI evaluation of stenosis on coronary CT angiography, comparison with quantitative coronary angiography and fractional flow reserve. JACC: Cardiovasc Imaging 2022; 16(2): 193–205. doi: 10.1016/j.jcmg.2021.10.020

4. Rahman F, Finkelstein N, Alyakin A, et al. Using machine learning for early prediction of cardiogenic shock in patients with acute heart failure. Journal of the Society for Cardiovascular Angiography & Interventions 2022; 1(3): 100308. doi: 10.1016/j.jscai.2022.100308

5. Ross EG, Shah NH, Dalman RL, et al. The use of machine learning for the identification of peripheral artery disease and future mortality risk. Journal of Vascular Surgery 2016; 64(5): 1515–1522.e3. doi: 10.1016/j.jvs.2016.04.026

6. Park JY, Noh YK, Choi BG, et al. TCTAP A-010 A machine learning-based approach to prediction of acute coronary syndrome. Journal of the American College of Cardiology 2015; 65(17): S6. doi: 10.1016/j.jacc.2015.03.057

7. Stuckey T, Singh N, Goswami R, et al. TCT-154 Gender based assessment of coronary artery disease by cardiac phase tomography using machine-learned algorithms. Journal of the American College of Cardiology 2017; 70(18): B66. doi: 10.1016/j.jacc.2017.09.218

8. Betancur JA, Otaki Y, Fish M, et al. Rest scan does not improve automatic machine learning prediction of major adverse coronary events after high speed myocardial perfusion imaging. Journal of the American College of Cardiology 2017; 69(11): 1590. doi: 10.1016/s0735-1097(17)34979-3

9. Ghosh P, Lilhore UK, Simaiya S, et al. Prediction of the risk of heart attack using machine learning techniques. In: Sharma S, Peng SL, Agrawal J, et al. (editors). Data, Engineering and Applications. Springer, Singapore; 2022. Volume 907. pp. 613–621.

10. Goswami R, Stuckey T, Meine F, et al. Coronary artery disease learning and algorithm development study: Early analysis of ejection fraction evaluation. Journal of the American College of Cardiology 2017; 69(11): 953. doi: 10.1016/s0735-1097(17)34342-5

11. Nakanishi R, Dey D, Commandeur F, et al. Machine learning in predicting coronary heart disease and cardiovascular disease events: Results from the multi-ethnic study of atherosclerosis (Mesa). Journal of the American College of Cardiology2018; 71(11): A1483. doi: 10.1016/s0735-1097(18)32024-2

12. Bom MJ, Levin E, Driessen RS, et al. Predictive value of targeted proteomics for coronary plaque morphology in patients with suspected coronary artery disease. eBioMedicine 2019; 39: 109–117. doi: 10.1016/j.ebiom.2018.12.033

13. Ramirez JL, Magaret CA, Khetani SA, et al. PC102. A novel machine learning-driven clinical and proteomic tool for the diagnosis of peripheral artery disease. Journal of Vascular Surgery 2019; 69(6): e233–e234. doi: 10.1016/j.jvs.2019.04.344

14. Collet JP, Zeitouni M, Procopi N, et al. Long-term evolution of premature coronary artery disease. Journal of the American College of Cardiology 2019; 74(15): 1868–1878. doi: 10.1016/j.jacc.2019.08.1002

15. Howard JP, Cook CM, van de Hoef TP, et al. Artificial Intelligence for aortic pressure waveform analysis during coronary angiography: Machine learning for patient safety. JACC: Cardiovascular Interventions 2019; 12(20): 2093–2101. doi: 10.1016/j.jcin.2019.06.036

16. Kim JT, Cho S, Lee SY, et al. The use of machine learning algorithms for the identification of stable obstructive coronary artery disease. Journal of the American College of Cardiology 2020; 75(11): 254. doi: 10.1016/S0735-1097(20)30881-0

17. Kawasaki T, Kidoh M, Kido T, et al. Evaluation of significant coronary artery disease based on CT fractional flow reserve and plaque characteristics using random forest analysis in machine learning. Academic Radiology 2020; 27(12): 1700–1708. doi: 10.1016/j.acra.2019.12.013

18. Vazquez B, Fuentes-Pineda G, Garcia F, et al. Risk markers by sex for in-hospital mortality in patients with acute coronary syndrome: A machine learning approach. Informatics in Medicine Unlocked 2021; 27: 100791. doi: 10.1016/j.imu.2021.100791

19. Sandhu JK, Lilhore UK, Poongodi M, et al. Predicting the risk of heart failure based on clinical data. Human-centric Computing and Information Sciences 2022; 12: 57. doi: 10.22967/HCIS.2022.12.057

20. Schwalm JD, Di S, Sheth T, et al. A machine learning-based clinical decision support algorithm for reducing unnecessary coronary angiograms. Cardiovascular Digital Health Journal 2022; 3(1): 21–30. doi: 10.1016/j.cvdhj.2021.12.001

21. Swathy M, Saruladha K. A comparative study of classification and prediction of Cardio-Vascular Diseases (CVD) using machine learning and deep learning techniques. ICT Express 2022; 8(1): 109–116. doi: 10.1016/j.icte.2021.08.021

22. Ahmad A, Corban MT, Moriarty JP, et al. Coronary reactivity assessment is associated with lower health care—Associated costs in patients presenting with angina and nonobstructive coronary artery disease. Circulation: Cardiovascular Interventions 2023; 16(7): e012387. doi: 10.1161/CIRCINTERVENTIONS.122.012387

23. Chang V, Bhavani VR, Xu AQ, Hossain MA. An artificial intelligence model for heart disease detection using machine learning algorithms. Healthcare Analytics 2022; 2: 100016. doi: 10.1016/j.health.2022.100016

24. Hossain MM, Swarna RA, Mostafiz R. Analysis of the performance of feature optimization techniques for the diagnosis of machine learning-based chronic kidney disease. Machine Learning with Applications 2022; 9: 100330. doi: 10.1016/j.mlwa.2022.100330

25. Liu Y, Ren H, Fanous H, et al. A machine learning model in predicting hemodynamically significant coronary artery disease: A prospective cohort study. Cardiovascular Digital Health Journal 2022; 3(3): 112–117. doi: 10.1016/j.cvdhj.2022.02.002

26. Li Q, Campan A, Ren A, Eid WE. Automating and improving cardiovascular disease prediction using machine learning and EMR data features from a regional healthcare system. International Journal of Medical Informatics 2022; 163: 104786. doi: 10.1016/j.ijmedinf.2022.104786

27. Huang Z, Xiao J, Wang X, et al. Clinical evaluation of the automatic coronary artery disease reporting and data system (CAD-RADS) in coronary computed tomography angiography using convolutional neural networks. Academic Radiology 2023; 30(4): 698–706. doi: 10.1016/j.acra.2022.05.015

28. Gharleghi R, Adikari D, Ellenberger K, et al. Automated segmentation of normal and diseased coronary arteries—The ASOCA challenge. Computerized Medical Imaging and Graphics 2022; 97: 102049. doi: 10.1016/j.compmedimag.2022.102049

29. Uma KV, Pudumalar S, Sharon blessie E. A combined classification algorithm based on C5.0 and NB to predict chronic obstructive pulmonary disease. In: Proceedings of the 2018 IEEE International Conference on Computational Intelligence and Computing Research (ICCIC); 13–15 December 2018; Madurai, India. pp. 1–4.

30. Mehta S, Shukla D. Optimization of C5.0 classifier using Bayesian theory. In: Proceedings of the 2015 International Conference on Computer, Communication and Control (IC4); 10–12 September 2015; Indore, India. pp. 1–6.

31. Coronary artery disease analysis & prediction. Available online: https://www.kaggle.com/code/homelysmile/coronary-artery-disease-analysis-prediction/data?select=DataClean-fullage.csv (accessed on 15 September 2022).

32. Wang M, Gao K, Wang L, Miu X. A novel hyperspectral classification method based on C5.0 decision tree of multiple combined classifiers. In: Proceedings of the 2012 Fourth International Conference on Computational and Information Sciences; 17–19 August 2012; Chongqing, China. pp. 373–376.

33. Jincheng Y, Ping J, Guangyu C, et al. Application of C5.0 algorithm in failure prediction of smart meters. In: Proceedings of the 2016 13th International Computer Conference on Wavelet Active Media Technology and Information Processing (ICCWAMTIP); 16–18 December 2016; Chengdu, China. pp. 328–333.

34. Pashaei E, Ozen M, Aydin N. Improving medical diagnosis reliability using Boosted C5.0 decision tree empowered by Particle Swarm Optimization. In: Proceedings of the 2015 37th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC); 25–29 August 2015; Milan, Italy. pp. 7230–7233.

35. Dalal S, Onyema EM, Kumar P, et al. A hybrid machine learning model for timely prediction of breast cancer. International Journal of Modeling, Simulation, and Scientific Computing 2023. doi: 10.1142/S1793962323410234

36. Edeh MO, Dalal S, Dhaou IB, et al. Artificial intelligence-based ensemble learning model for prediction of hepatitis C disease. Frontiers in Public Health 2022; 10: 892371. doi: 10.3389/fpubh.2022.892371

37. Onyema EM, Shukla PK, Dalal S, et al. Enhancement of patient facial recognition through deep learning algorithm: ConvNet. Journal of Healthcare Engineering 2021; 2021: 5196000. doi: 10.1155/2021/5196000

38. Ramesh TR, Lilhore UK, Poongodi M, et al. Predictive analysis of heart diseases with machine learning approaches. Malaysian Journal of Computer Science 2022; 2022: 132–148. doi: 10.22452/mjcs.sp2022no1.10

39. Chauhan AS, Lilhore UK, Gupta AK, et al. Comparative analysis of supervised machine and deep learning algorithms for kyphosis disease detection. Applied Sciences 2023; 13(8): 5012. doi: 10.3390/app13085012

40. Asif D, Bibi M, Arif MS, Mukheimer A. Enhancing heart disease prediction through ensemble learning techniques with hyperparameter optimization. Algorithms 2023; 16(6): 308. doi: 10.3390/a16060308




DOI: https://doi.org/10.32629/jai.v6i3.628

Refbacks

  • There are currently no refbacks.


Copyright (c) 2023 Surjeet Dalal, Umesh Kumar Lilhore, Sarita Simaiya, Vivek Jaglan, Anand Mohan, Sachin Ahuja, Akshat Agrawal, Martin Margala, Prasun Chakrabarti

License URL: https://creativecommons.org/licenses/by-nc/4.0