Spatio-Temporal ECG Network for Detecting Cardiac Disorders from Multi-Lead ECGs

Abstract

Automatic detection and classification of cardiac disorders play a critical role in the analysis of clinical electrocardiogram (ECG). Deep learning methods are effective for automated feature extraction and have shown promising results in ECG classification. In this work, we proposed a deep spatio-temporal ECG network (ST-ECGNet) to extract robust spatio-temporal features for detecting multiple cardiac disorders from the multi-lead ECG data. The proposed ST-ECGNet combines a Convolutional Neural Network (CNN) module for extracting local spatial features, an attention module for capturing global spatial features, and a Bi-directional Gated Recurrent Unit (Bi-GRU) module for extracting temporal features from ECG data. Specifically, the attention mechanism enables our deep learning architecture to focus on the most important and useful parts of the input to make more accurate predictions. In PhysioNet/Computing in Cardiology Challenge 2021, our entry was not officially ranked and scored on the test data of the Challenge, because our code was not successfully processed during the official phase and failed to run with errors.