Myocardial Infarction Detection and Localization with Electrocardiogram Based on Convolutional Neural Network

Electrocardiogram (ECG) is widely used in Myocardial infarction (MI) diagnosis. The automatic diagnosis of MI based on the 12-lead ECG needs to consider not only the waveform change features in multi-resolution time series, but also the spatial correlation information between the leads. To this end, this work proposed multiscale spatiotemporal feature extraction method based on Convolutional neural network (CNN) for MI automatic diagnosis. First, the 12-lead ECG is first transformed into an ECG image through wavelet decomposition and 3-dimensional space reconstruction. The MI-CNN model is then constructed to identify MI using 41368 ECG images. Finally, we develop the LL-CNN model, which is utilized only after the ECG signal is identified as an MI event by the MI-CNN model, to localize MI by employing transfer learning to overcome the limited data problem. The proposed method has achieved an accuracy of 99.51% on MI detection, and a macro-F1 of 99.14% on MI localization. Moreover, the features visualization shows that U-wave has significant diagnostic value for MI. The proposed method significantly improves the performance of MI detection and localization compared with other methods. It is promising to be used for MI monitoring and diagnosis.