Arrhythmia Classification with Attention-Based Res-BiLSTM-Net
In the modern clinical diagnosis, the 12-lead electrocardiogram (ECG) signal has proved effective in cardiac arrhythmias classification. However, the manual diagnosis for cardiac arrhythmias is tedious and error-prone through ECG signals. In this work, we propose an end-to-end deep neural network called attention-based Res-BiLSTM-Net for automatic diagnosis of cardiac arrhythmias. Our model is capable of classifying ECG signals with different lengths. The proposed network consists of two parts: the attention-based Resnet and the attention-based BiLSTM. At first, ECG signals are divided into several signal segments with the same length. Then multi-scale features are extracted by our attention-based Resnet through signal segments. Next, these multi-scale features from a same ECG signal are integrated in chronological order. In the end, our attention-based BiLSTM classifies cardiac arrhythmias according to combined features. Our method achieved a good result with an average F1score of 0.8757 on a multi-label arrhythmias classification problem in the First China ECG Intelligent Competition.
KeywordsCardiac arrhythmias classification Attention Resnet BiLSTM
This work was supported by the National Natural Science Foundation of China (Grant Nos. 61672231), Shanghai Natural Science Foundation (Grant No. 18ZR1411400), and Fundamental Research Funds for the Central Universities.
- 3.Mazomenos, E.B., Chen, T., Acharyya, A., et al.: A time-domain morphology and gradient based algorithm for ECG feature extraction. In: 2012 IEEE International Conference on Industrial Technology, pp. 117–122. IEEE (2012)Google Scholar
- 6.Murugesan, B., Ravichandran, V., Ram, K., et al.: ECGNet: deep network for arrhythmia classification. In: 2018 IEEE International Symposium on Medical Measurements and Applications (MeMeA), pp. 1–6. IEEE (2018)Google Scholar
- 7.The First China ECG Intelligent Competition. http://mdi.ids.tsinghua.edu.cn/#/
- 8.He, K., Zhang, X., Ren, S., et al.: Deep residual learning for image recognition. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 770–778 (2016)Google Scholar
- 9.Xu, B., Wang, N., Chen, T., et al.: Empirical evaluation of rectified activations in convolutional network. arXiv preprint arXiv:1505.00853 (2015)
- 10.Yao, X., Li, X., Ye, Q., et al.: A robust deep learning approach for automatic seizure detection. arXiv preprint arXiv:1812.06562 (2018)
- 12.Gers, F.A., Schmidhuber, J., Cummins, F.: Learning to forget: continual prediction with LSTM (1999)Google Scholar