Multi-label Classification of Abnormalities in 12-Lead ECG Using 1D CNN and LSTM.

Cheng-Si Luo,Hongxiu Jiang,Quanchi Li,Nini Rao
DOI: https://doi.org/10.1007/978-3-030-33327-0_7
2019-01-01
Abstract:In this study, we proposed a method based on Convolutional Neural Network (CNN) and Long Short-Term Memory (LSTM) to classify 12-lead ECG into 9 categories (1 normal, 8 abnormal). The only preprocessing techniques we used are the baseline drift removal based on median filtering and signal segmentation. Then an 18-layer deep 1D CNN consisting of residual blocks and skip architectures that is followed by a bi-directional LSTM layer was developed. During the training session, we suggested a new strategy to compute the Dice loss for multi-label classification. The average F1-score we achieved on the hidden testing dataset of the First China ECG Intelligent Competition (FCEIC) is 85.11%. With the same model, we achieved 82.21% (5-fold cross-validation) on the Chinese physiological signal challenge 2018 (CPSC2018) training dataset.
What problem does this paper attempt to address?