Skip to main page content
U.S. flag

An official website of the United States government

Dot gov

The .gov means it’s official.
Federal government websites often end in .gov or .mil. Before sharing sensitive information, make sure you’re on a federal government site.

Https

The site is secure.
The https:// ensures that you are connecting to the official website and that any information you provide is encrypted and transmitted securely.

Access keys NCBI Homepage MyNCBI Homepage Main Content Main Navigation
. 2023 Oct;18(4):278-287.
doi: 10.18502/jthc.v18i4.14827.

Prediction of the Fatal Acute Complications of Myocardial Infarction via Machine Learning Algorithms

Affiliations

Prediction of the Fatal Acute Complications of Myocardial Infarction via Machine Learning Algorithms

Reza Ghafari et al. J Tehran Heart Cent. 2023 Oct.

Abstract

Background: Myocardial infarction (MI) is a major cause of death, particularly during the first year. The avoidance of potentially fatal outcomes requires expeditious preventative steps. Machine learning (ML) is a subfield of artificial intelligence science that detects the underlying patterns of available big data for modeling them. This study aimed to establish an ML model with numerous features to predict the fatal complications of MI during the first 72 hours of hospital admission.

Methods: We applied an MI complications database that contains the demographic and clinical records of patients during the 3 days of admission based on 2 output classes: dead due to the known complications of MI and alive. We utilized the recursive feature elimination (RFE) method to apply feature selection. Thus, after applying this method, we reduced the number of features to 50. The performance of 4 common ML classifier algorithms, namely logistic regression, support vector machine, random forest, and extreme gradient boosting (XGBoost), was evaluated using 8 classification metrics (sensitivity, specificity, precision, false-positive rate, false-negative rate, accuracy, F1-score, and AUC).

Results: In this study of 1699 patients with confirmed MI, 15.94% experienced fatal complications, and the rest remained alive. The XGBoost model achieved more desirable results based on the accuracy and F1-score metrics and distinguished patients with fatal complications from surviving ones (AUC=78.65%, sensitivity=94.35%, accuracy=91.47%, and F1-score=95.14%). Cardiogenic shock was the most significant feature influencing the prediction of the XGBoost algorithm.

Conclusion: XGBoost algorithms can be a promising model for predicting fatal complications following MI.

Keywords: Artificial intelligence; Machine learning; Mortality; Myocardial infarction; Prognosis.

PubMed Disclaimer

Figures

Figure 1.
Figure 1.
The images present the confusion matrices of the classifiers at the time of admission.
Figure 2.
Figure 2.
The image presents the most notable features of the XGBoost model.

References

    1. Reeder GS. Identification and treatment of complications of myocardial infarction. Mayo Clin Proc. 1995;70:880–884. - PubMed
    1. Virani SS, Alonso A, Aparicio HJ, Benjamin EJ, Bittencourt MS, Callaway CW, Carson AP, Chamberlain AM, Cheng S, Delling FN, Elkind MSV, Evenson KR, Ferguson JF, Gupta DK, Khan SS, Kissela BM, Knutson KL, Lee CD, Lewis TT, Liu J, Loop MS, Lutsey PL, Ma J, Mackey J, Martin SS, Matchar DB, Mussolino ME, Navaneethan SD, Perak AM, Roth GA, Samad Z, Satou GM, Schroeder EB, Shah SH, Shay CM, Stokes A, VanWagner LB, Wang NY, Tsao CW, American Heart Association Council on Epidemiology and Prevention Statistics Committee and Stroke Statistics Subcommittee . Heart Disease and Stroke Statistics-2021 Update: A Report From the American Heart Association. Circulation 2021;143:e254–e743. - PubMed
    1. Kutty RS, Jones N, Moorjani N. Mechanical complications of acute myocardial infarction. Cardiol Clin 2013;31:519–531, vii–viii. - PubMed
    1. Montrief T, Davis WT, Koyfman A, Long B. Mechanical, inflammatory, and embolic complications of myocardial infarction: An emergency medicine review. Am J Emerg Med 2019;37:1175–1183. - PubMed
    1. Cao Y, Fang X, Ottosson J, Näslund E, Stenberg E. A Comparative Study of Machine Learning Algorithms in Predicting Severe Complications after Bariatric Surgery. J Clin Med 2019;8:668. - PMC - PubMed

LinkOut - more resources