Abstract
In this work, we have proposed an electrocardiogram (ECG) arrhythmia classification method for short 12-lead ECG records to identify nine types (one normal type and eight abnormal types), using a 1D densely connected CNN which is a relatively novel convolutional neural network (CNN) model and shows outstanding performance in the field of pattern recognition. Firstly, noticing that ECG records are one dimensional time series with different noise levels, several wavelet-based shrinkage filtering methods were adopted to the ECG records for data augmentation. Secondly, each ECG record was divided into segments with a fixed length of 10 s, and the total number of segments for an ECG record is 10. And then, 10 segments were fed into an optimized 1D densely connected CNN for training. And lastly, a threshold vector was trained for the multi-label classification since each record may have more than one abnormal types. The approach has been validated against The First China ECG Intelligent Competition data set, obtaining a final F1 score of 0.873 and 0.863 on the validation set and test set, respectively.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
The First China ECG Intelligent Competition. http://mdi.ids.tsinghua.edu.cn. Accessed 12 July 2019
Rajpurkar, P., Hannun, A.Y., Haghpanahi, M., et al.: Cardiologist-level arrhythmia detection with convolutional neural networks. arXiv preprint arXiv:1707. 01836 (2017)
Jun, T.J., Nguyen, H.M., Kang, D., et al.: ECG arrhythmia classification using a 2-d convolutional neural network. arXiv preprint arXiv:1804.06812 (2018)
Donoho, D.L.: Denoising by soft thresholding. IEEE Trans. Inf. Theory 41, 613–627 (1995)
Alfaouri, M., Daqrouq, K.: ECG signal denoising by wavelet transform thresholding. Am. J. Appl. Sci. 5(3), 276–281 (2008)
He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: 2015 IEEE Conference, pp. 770–778 (2015)
Mostayed, A., Luo, J., Shu, X., et al.: Classification of 12-lead ECG signals with bi-directional LSTM network. arXiv preprint arXiv:1811.02090 (2018)
Huang, G., Liu, Z., van der Maaten, L., Weinberger, K.Q.: Densely connected convolutional networks. In: 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 2261–2269 (2017)
Ioffe, S., Szegedy, C.: Batch normalization: accelerating deep network training by reducing internal covariate shift. In: 2013 IEEE Conference, pp. 448–456 (2013)
He, K., Zhang, X., Ren, S., Sun, J.: Identity mappings in deep residual networks. In: Leibe, B., Matas, J., Sebe, N., Welling, M. (eds.) ECCV 2016. LNCS, vol. 9908, pp. 630–645. Springer, Cham (2016). https://doi.org/10.1007/978-3-319-46493-0_38
Srivastava, N., Hinton, G.E., Krizhevsky, A., et al.: Dropout: a simple way to prevent neural networks from overfitting. J. Mach. Learn. Res. 15(1), 1929–1958 (2014)
Xiong, Z., Stiles, M.K., Zhao, J.: Robust ECG signal classification for detection of atrial fibrillation using a novel neural network. In: 2017 Computing in Cardiology (CinC), pp. 24–27 (2017)
Zhang, M.-L., Zhou, Z.-H.: A review on multi-label learning algorithms. IEEE Trans. Knowl. Data Eng. 26(8), 1819–1837 (2014)
Tsoumakas, G., Katakis, I.: Multi-label classification: an overview. Int. J. Data Warehous. Min. 3(3), 1–13 (2007)
Author information
Authors and Affiliations
Corresponding authors
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2019 Springer Nature Switzerland AG
About this paper
Cite this paper
Wang, C., Yang, S., Tang, X., Li, B. (2019). A 12-Lead ECG Arrhythmia Classification Method Based on 1D Densely Connected CNN. In: Liao, H., et al. Machine Learning and Medical Engineering for Cardiovascular Health and Intravascular Imaging and Computer Assisted Stenting. MLMECH CVII-STENT 2019 2019. Lecture Notes in Computer Science(), vol 11794. Springer, Cham. https://doi.org/10.1007/978-3-030-33327-0_9
Download citation
DOI: https://doi.org/10.1007/978-3-030-33327-0_9
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-33326-3
Online ISBN: 978-3-030-33327-0
eBook Packages: Computer ScienceComputer Science (R0)