PROGNOSTIC VALUE OF A RANDOM FOREST MACHINE LEARNING MODEL FOR MORTALITY IN PATIENTS WITH ACUTE MYOCARDIAL INFARCTION

Công Duy Trần, Quang Sang Lý, Quốc Thành Mai, Đinh Quốc Anh Nguyễn, Thanh Trúc Thái, Văn Sỹ Hoàng

Main Article Content

Abstract

Objectives: To evaluate the prognostic value of a random forest machine learning model for predicting mortality in patients with acute myocardial infarction (AMI). Subjects and methods: This retrospective cohort study included 540 patients with AMI treated at Cho Ray Hospital from January 2020 to September 2021. Clinical, laboratory, coronary angiographic, and treatment data were used to train a random forest model using Python 3.12. Data were standardized, class imbalance was addressed with the SMOTE method, and principal component analysis was applied to identify key predictors. Model performance was assessed using the area under the receiver operating characteristic curve (AUC) and accuracy. Results: The mean age was 64.0 ± 11.6 years, and 71.3% were male. The 12-month mortality rate was 10.7%. The most important predictors of mortality included Killip class, clinical type of AMI, lesions of the right coronary artery and the left anterior descending artery, smoking, use of angiotensin-converting enzyme inhibitors/angiotensin II receptor blockers, anemia, admission troponin I level, age, and Gensini score. In the original dataset, the model achieved AUC values ranging from 0.653 to 0.730, with accuracy between 0.889 and 0.907. After data standardization and class balancing, AUC improved to 0.709–0.730. Conclusion: The random forest machine learning model demonstrated fair prognostic performance for mortality in patients with AMI.

Article Details

References

1. Cho SM, Austin PC, Ross HJ, Abdel-Qadir H, Chicco D, Tomlinson G, et al. Machine Learning Compared With Conventional Statistical Models for Predicting Myocardial Infarction Readmission and Mortality: A Systematic Review. Can J Cardiol. 2021;37(8):1207-1214. doi: 10.1016/j.cjca.2021.02.020.
2. Lee HC, Park JS, Choe JC, Ahn JH, Lee HW, Oh JH, et al. Prediction of 1-Year Mortality from Acute Myocardial Infarction Using Machine Learning. Am J Cardiol. 2020;133:23-31. doi: 10.1016/j.amjcard.2020.07.048.
3. Yang J, Li Y, Li X, Tao S, Zhang Y, Chen T, et al. A Machine Learning Model for Predicting In-Hospital Mortality in Chinese Patients With ST-Segment Elevation Myocardial Infarction: Findings From the China Myocardial Infarction Registry. J Med Internet Res. 2024;26:e50067. doi: 10.2196/50067.
4. Li YW, Yang XD, Cai YC, Kong XY. Artificial intelligence based prediction of acute myocardial infarction mortality risk. 2nd International Conference on Health Big Data and Intelligent Healthcare (ICHIH). Zhuhai, China: IEEE. 2023: 164-169. doi:10.1109/ICHIH60370.2023. 10396320.
5. Zhang X, Wang X, Xu L, Liu J, Ren P, Wu H. The predictive value of machine learning for mortality risk in patients with acute coronary syndromes: a systematic review and meta-analysis. Eur J Med Res. 2023;28(1):451. doi: 10.1186/s40001-023-01027-4.
6. Hadanny A, Shouval R, Wu J, Shlomo N, Unger R, Zahger D, et al. Predicting 30-day mortality after ST elevation myocardial infarction: Machine learning-based random forest and its external validation using two independent nationwide datasets. J Cardiol. 2021;78(5):439-446. doi: 10.1016/j.jjcc.2021.06.002.
7. Oliveira M, Seringa J, Pinto FJ, Henriques R, Magalhães T. Machine learning prediction of mortality in Acute Myocardial Infarction. BMC Med Inform Decis Mak. 2023;23(1):70. doi: 10.1186/ s12911-023-02168-6.
8. Thygesen K, Alpert JS, Jaffe AS, et al. Fourth universal definition of myocardial infarction. Circulation. 2018;138: e618-e651. doi: 10.1161/ CIR.0000000000000617.