Multi-Label Classification of Multi-lead ECG Based on Deep 1D Convolutional Neural Networks with Residual and Attention Mechanism.
Yamin Liu,Hanshuang Xie,Qineng Cao,Jiayi Yan,Fan Wu,Huaiyu Zhu,Yun Pan
DOI: https://doi.org/10.23919/cinc53138.2021.9662873
2021-01-01
Abstract:Arrhythmia automatic analysis techniques provide convenience for the prevention and diagnosis of cardiac disease. Aiming at classify cardiac abnormalities into 27 classes with either 12-lead, 6-lead, 4-lead, 3-lead or 2-lead multi-label ECG recordings, we develop a deep 1-Dimensional Convolutional Neural Network (1D CNN) with residual block and squeeze-and-excitation (SE) attention mechanism (namely 1D RANet). First, we introduce residual block and SE attention block into 1D CNN to extract deep features adaptively and avoid vanishing gradient or network degradation. Second, to improve the robustness, we use data augmentation techniques such as band-pass and wavelet-based filter, noise addition, time-frequency switch and so on. For the convenience of training, all recordings are cropped or padded to 60 seconds with resampled rates of 300 Hz. Our classifiers received challenge scores of 0.73, 0.53, 0.59, 0.57 and 0.47 for the 12-lead, 6-lead, 4-lead, 3-lead, and 2-lead versions of the public training set, 0.56, 0.54, 0.56, 0.54 and 0.55 for the hidden validation set and 0.52, 0.49, 0.51, 0.50 and 0.49 for the hidden test sets (Team name: Proton). However, our entries did not rank since we missed the preprint submitting deadline of the Computing in Cardiology Conference.