Advertisement

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.

Keywords

Electrocardiogram (ECG) Arrhythmia CNN 

References

  1. 1.
    The First China ECG Intelligent Competition. http://mdi.ids.tsinghua.edu.cn. Accessed 12 July 2019
  2. 2.
    Rajpurkar, P., Hannun, A.Y., Haghpanahi, M., et al.: Cardiologist-level arrhythmia detection with convolutional neural networks. arXiv preprint arXiv:1707. 01836 (2017)
  3. 3.
    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)
  4. 4.
    Donoho, D.L.: Denoising by soft thresholding. IEEE Trans. Inf. Theory 41, 613–627 (1995)CrossRefGoogle Scholar
  5. 5.
    Alfaouri, M., Daqrouq, K.: ECG signal denoising by wavelet transform thresholding. Am. J. Appl. Sci. 5(3), 276–281 (2008)CrossRefGoogle Scholar
  6. 6.
    He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: 2015 IEEE Conference, pp. 770–778 (2015)Google Scholar
  7. 7.
    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)
  8. 8.
    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)Google Scholar
  9. 9.
    Ioffe, S., Szegedy, C.: Batch normalization: accelerating deep network training by reducing internal covariate shift. In: 2013 IEEE Conference, pp. 448–456 (2013)Google Scholar
  10. 10.
    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_38CrossRefGoogle Scholar
  11. 11.
    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)MathSciNetzbMATHGoogle Scholar
  12. 12.
    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)Google Scholar
  13. 13.
    Zhang, M.-L., Zhou, Z.-H.: A review on multi-label learning algorithms. IEEE Trans. Knowl. Data Eng. 26(8), 1819–1837 (2014)CrossRefGoogle Scholar
  14. 14.
    Tsoumakas, G., Katakis, I.: Multi-label classification: an overview. Int. J. Data Warehous. Min. 3(3), 1–13 (2007)CrossRefGoogle Scholar

Copyright information

© Springer Nature Switzerland AG 2019

Authors and Affiliations

  1. 1.Chengdu Spaceon Electronics Co., Ltd.ChengduChina

Personalised recommendations