LASSO Regression-Based Diagnosis of Acute ST-Segment Elevation Myocardial Infarction (STEMI) on Electrocardiogram (ECG)

Electrocardiogram (ECG) is an important tool for the detection of acute ST-segment elevation myocardial infarction (STEMI). However, machine learning (ML) for the diagnosis of STEMI complicated with arrhythmia and infarct-related arteries is still underdeveloped based on real-world data. Therefore, we aimed to develop an ML model using the Least Absolute Shrinkage and Selection Operator (LASSO) to automatically diagnose acute STEMI based on ECG features. A total of 318 patients with STEMI and 502 control subjects were enrolled from Jan 2017 to Jun 2019. Coronary angiography was performed. A total of 180 automatic ECG features of 12-lead ECG were input into the model. The LASSO regression model was trained and validated by the internal training dataset and tested by the internal and external testing datasets. A comparative test was performed between the LASSO regression model and different levels of doctors. To identify the STEMI and non-STEMI, the LASSO model retained 14 variables with AUCs of 0.94 and 0.93 in the internal and external testing datasets, respectively. The performance of LASSO regression was similar to that of experienced cardiologists (AUC: 0.92) but superior (p < 0.05) to internal medicine residents, medical interns, and emergency physicians. Furthermore, in terms of identifying left anterior descending (LAD) or non-LAD, LASSO regression achieved AUCs of 0.92 and 0.98 in the internal and external testing datasets, respectively. This LASSO regression model can achieve high accuracy in diagnosing STEMI and LAD vessel disease, thus providing an assisting diagnostic tool based on ECG, which may improve the early diagnosis of STEMI.

Automated summary

This study aimed to develop a machine learning model using the LASSO algorithm to automatically diagnose acute STEMI based on ECG features. The model achieved high accuracy in diagnosing STEMI and showed similar performance to experienced cardiologists and better performance than other doctors. For identifying LAD or non-LAD, the model also performed well.