Spatio-Temporal 3-Dimentional Variations Modeling with Self-Attention for Multi-Label ECG Classification
Pan Xia,Hao Zhang,Yicheng Yao,Lirui Xu,Zhongrui Bai,Xianxiang Chen,Zhenfeng Li,Yusi Zhu,Xiaoran Li,Lidong Du,Peng Wang,Zhen Fang
DOI: https://doi.org/10.1109/jsen.2024.3392017
IF: 4.3
2024-01-01
IEEE Sensors Journal
Abstract:Multi-label electrocardiogram (ECG) classification is of great significance for the diagnosis of cardiovascular diseases, and it is also a challenging task because of the various of pathological patterns across different cardiac abnormalities. This paper analyzes the complex spatial-temporal variations information contained in the 12-lead ECG from a new perspective, and proposes an advanced deep neural network framework to accurately identify various multi-label ECG abnormalities. Firstly, the 12-lead ECG was modeled as 3-dimensional (3-D) spatial-temporal variations involving the intra-period, inter-period temporal dimensions, and cross-lead spatial dimension. Then, a 3-D spatial-temporal learning network (3D-STLNet) was carefully designed to extract the complex pathological features from the 3-D spatial-temporal variations. Secondly, the multi-head attention (MHA) mechanism was introduced into the proposed network to extract global temporal dependencies in ECG segments, which further improved the performance of the proposed network. By integrating a residual backbone network, a 3D-STLNet, and the MHA module, our proposed network allows end-to-end learning and prediction in a unified framework. The influence of several classic threshold selection strategies on the performance of multi-label ECG classification algorithm was also analyzed, which provides a reference for subsequent research on optimal threshold selection for multi-label ECG classification. Finally, our proposed network was evaluated on multiple large multi-label ECG public datasets, and the performance of our network outperforms that of current state-of-the-art networks, which demonstrates that our network has great competitiveness in identifying a wider range of multi-label ECG abnormalities.