Signal-guided multitask learning for myocardial infarction classification using images of electrocardiogram.

Abstract

Introduction: The diagnosis of myocardial infarction (MI) needs to be swift and accurate, but definitively diagnosing it based on the first test encountered in clinical practice, the electrocardiogram (ECG), is not an easy task. The purpose of the study is to develop a deep learning (DL) algorithm using multitask learning method to differentiate patients experiencing MI from those without coronary artery disease using image-based ECG data.

Methods: A DL model was developed based on 11,227 ECG images. We developed a new ECG interpretation algorithm through signal-guided multitask learning, building on a previously published single-task algorithm. The utility of this model was evaluated by testing 51 physicians in interpreting ECGs with and without the assistance of the DL model.

Results: The proposed model demonstrated superior performance, achieving 90.56% accuracy, 83.82% sensitivity, 93.02% specificity, 81.44% precision, and an F1 score of 82.61% in discriminating MI ECG. Overall, the median accuracy of ECG interpretation improved from 62% to 68% with the DL algorithm. Trainees and specialists in internal medicine experienced significant accuracy increases (60% to 66% for trainees, 72% to 80% for specialists). In the MI group, NSTEMI accuracy was notably lower than STEMI (33% vs. 80%, p < 0.001), but the DL algorithm improved interpretative capabilities in both NSTEMI and STEMI.

Conclusions: Signal-guided multitask DL algorithm demonstrated superior performance compared with previous single-task algorithm. The DL algorithm supports the physicians' decision discriminating MI ECGs from non-MI ECGs. The improvement was consistent in subgroups of STEMI and NSTEMI.