Understanding Arteriosclerotic Heart Disease Patients Using Electronic Health Records: A Machine Learning and Shapley Additive exPlanations Approach
2
조회
0
다운로드
1
인용
0
참고문헌
키워드
초록
Objectives: The number of deaths from cardiovascular disease is projected to reach 23.3 million by 2030. As a contribution to preventing this phenomenon, this paper proposed a machine learning (ML) model to predict patients with arteriosclerotic heart disease (AHD). We also interpreted the prediction model results based on the ML approach and deployed modelagnostic ML methods to identify informative features and their interpretations.Methods: We used a hematology Electronic Health Record (EHR) with information on erythrocytes, hematocrit, hemoglobin, mean corpuscular hemoglobin, mean corpuscular hemoglobin concentration, leukocytes, thrombocytes, age, and sex. To detect and predict AHD, we explored random forest (RF), XGBoost, and AdaBoost models. We examined the prediction model results based on the confusion matrix and accuracy measures. We used the Shapley Additive exPlanations (SHAP) framework to interpret the ML model and quantify the contribution of features to predictions.Results: Our study included data from 6,837 patients, with 4,702 records from patients diagnosed with AHD and 2,135 records from patients without an AHD diagnosis. AdaBoost outperformed RF and XGBoost, achieving an accuracy of 0.78, precision of 0.82, F1-score of 0.85, and recall of 0.88. According to the SHAP summary bar plot method, hemoglobin was the most important attribute for detecting and predicting AHD patients. The SHAP local interpretability bar plot revealed that hemoglobin and mean corpuscular hemoglobin concentration had positive impacts on AHD prediction based on a single observation.Conclusions: ML models based on real clinical data can be used to predict AHD.
Abstract
Objectives: The number of deaths from cardiovascular disease is projected to reach 23.3 million by 2030. As a contribution to preventing this phenomenon, this paper proposed a machine learning (ML) model to predict patients with arteriosclerotic heart disease (AHD). We also interpreted the prediction model results based on the ML approach and deployed modelagnostic ML methods to identify informative features and their interpretations.Methods: We used a hematology Electronic Health Record (EHR) with information on erythrocytes, hematocrit, hemoglobin, mean corpuscular hemoglobin, mean corpuscular hemoglobin concentration, leukocytes, thrombocytes, age, and sex. To detect and predict AHD, we explored random forest (RF), XGBoost, and AdaBoost models. We examined the prediction model results based on the confusion matrix and accuracy measures. We used the Shapley Additive exPlanations (SHAP) framework to interpret the ML model and quantify the contribution of features to predictions.Results: Our study included data from 6,837 patients, with 4,702 records from patients diagnosed with AHD and 2,135 records from patients without an AHD diagnosis. AdaBoost outperformed RF and XGBoost, achieving an accuracy of 0.78, precision of 0.82, F1-score of 0.85, and recall of 0.88. According to the SHAP summary bar plot method, hemoglobin was the most important attribute for detecting and predicting AHD patients. The SHAP local interpretability bar plot revealed that hemoglobin and mean corpuscular hemoglobin concentration had positive impacts on AHD prediction based on a single observation.Conclusions: ML models based on real clinical data can be used to predict AHD.
참고문헌 (30)
- Vinciguerra M, "Atherosclerosis as pathogenetic substrate for Sars-Cov2 cytokine storm", J Clin Med, 9(7), pp.2095~ (2020)[DOI]
- "Cardiovascular disease" (2023)[URL]
- "Noncommunicable disease management guide" (2019)[URL]
- Capotosto L, "Early diagnosis of cardiovascular diseases in workers : role of standard and advanced echocardiography", Biomed Res Int, 2018, pp.7354691~ (2018)[DOI]
- Muhammad Y, "Early and accurate detection and diagnosis of heart disease using intelligent computational model", Sci Rep, 10(1), pp.19747~ (2020)[DOI]
- Guo CY, "The comprehensive machine learning analytics for heart failure", Int J Environ Res Public Health, 18(9), pp.4943~ (2021)[DOI]
- Karthick K, "Implementation of a heart disease risk prediction model using machine learning", Comput Math Methods Med, 2022, pp.6517716~ (2022)[DOI]
- Chen Z, "Prediction of atherosclerosis using machine learning based on operations research", Math Biosci Eng, 19(5), pp.4892~4910 (2022)[DOI]
- Park S, "New model for predicting the presence of coronary artery calcification", J Clin Med, 10(3), pp.457~ (2021)[DOI]
- Fan J, "The prediction of asymptomatic carotid atherosclerosis with electronic health records : a comparative study of six machine learning models", BMC Med Inform Decis Mak, 21(1), pp.115~ (2021)[DOI]
- Ward A, "Machine learning and atherosclerotic cardiovascular disease risk prediction in a multiethnic population", NPJ Digit Med, 3, pp.125~ (2020)[DOI]
- Terrada O, "A novel medical diagnosis support system for predicting patients with atherosclerosis diseases", Inf Med Unlocked, 21, pp.100483~ (2020)[DOI]
- Chen T, "XGBoost: a scalable tree boosting system", Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp.785~794 (2016)
- Frizzell JD, "Prediction of 30-day all-cause readmissions in patients hospitalized for heart failure : comparison of machine learning and other statistical approaches", JAMA Cardiol, 2(2), pp.204~209 (2017)[DOI]
- Budholiya K, "An optimized XGBoost based diagnostic system for effective prediction of heart disease", J King Saud Univ Comput Inf Sci, 34(7), pp.4514~4523 (2022)[DOI]
- Lundberg SM, "A unified approach to interpreting model predictions", Adv Neural Inf Process Syst, 30, pp.4765~4774 (2017)
- Cho E, "Data preprocessing combination to improve the performance of quality classification in the manufacturing process", Electronics, 11(3), pp.477~ (2022)[DOI]
- Johnson A, "A novel personalized random forest algorithm for clinical outcome prediction", Stud Health Technol Inform, 290, pp.248~252 (2022)[DOI]
- Absar N, "The efficacy of machine-learning-supported smart system for heart disease prediction", Healthcare(Basel), 10(6), pp.1137~ (2022)[DOI]
- Almustafa KM, "Prediction of heart disease and classifiers’sensitivity analysis", BMC Bioinformatics, 21(1), pp.278~ (2020)[DOI]
- Su X, "Prediction for cardiovascular diseases based on laboratory data : an analysis of random forest model", J Clin Lab Anal, 34(9), pp.e23421~ (2020)[DOI]
- Cao J, "Study on the risk of coronary heart disease in middle-aged and young people based on machine learning methods : a retrospective cohort study", PeerJ, 10, pp.e14078~ (2022)[DOI]
- Mahesh TR, "AdaBoost ensemble methods using k-fold cross validation for survivability with the early detection of heart disease", Comput Intell Neurosci, 2022, pp.9005278~ (2022)[DOI]
- Alelyani S, "Detection and evaluation of machine learning bias", Appl Sci, 11(14), pp.6271~ (2021)[DOI]
- He S, "Comparative evaluation of 15-minute rapid diagnosis of ischemic heart disease by high-sensitivity quantification of cardiac biomarkers", Exp Ther Med, 20(2), pp.1702~1708 (2020)[DOI]
- Lu S, "Understanding heart failure patients EHR clinical features via SHAP interpretation of tree-based machine learning model predictions", AMIA Annu Symp Proc, 2021, pp.813~822 (2022)
- Futagami K, "Pairwise acquisition prediction with SHAP value interpretation", J Financ Data Sci, 7, pp.22~44 (2021)[DOI]
- Wang K, "Interpretable prediction of 3-year all-cause mortality in patients with heart failure caused by coronary heart disease based on machine learning and SHAP", Comput Biol Med, 137, pp.104813~ (2021)[DOI]
- Lee G, "Association of hemoglobin concentration and its change with cardiovascular and all-cause mortality", J Am Heart Assoc, 7(3), pp.e007723~ (2018)[DOI]
- Goel H, "Anemia in cardiovascular disease : marker of disease severity or disease-modifying therapeutic target?", Curr Atheroscler Rep, 23(10), pp.61~ (2021)[DOI]