Advertisement

Transfer Learning for Electrocardiogram Classification Under Small Dataset

  • Longting Chen
  • Guanghua XuEmail author
  • Sicong Zhang
  • Jiachen Kuang
  • Long Hao
Conference paper
Part of the Lecture Notes in Computer Science book series (LNCS, volume 11794)

Abstract

The First China ECG Intelligent Competition is held by Tsinghua University. It is aimed to intelligently classify electrocardiogram (ECG) signals into two categories in preliminary and nine categories in rematch. The detailed ECG categories are listed in subsequent section. Our team proposes a deep residual network for diagnosing cardiovascular diseases automatically based on ECG, making full use of the network’s hierarchical feature learning and feature representation ability. Considering that the amount of this competition data is small, especially in the stage of preliminary where there are only 600 training samples, while the deep learning-based method is data-hungry. Transfer learning idea is introduced into the training process of proposed deep neural networks. The proposed network is firstly trained on the Physionet/CinC Challenge 2017 dataset that is an open-public ECG data with single lead. Then it is continuously fine-tuned on the competition dataset with 12 leads. The performance of the proposed network is improved a lot. The proposed method achieves \( F_{1} \) score of 0.89 and 0.86 in the hidden test set of preliminary and rematch, respectively. The research code will be released later.

Keywords

Electrocardiogram classification Deep residual network Transfer learning 

Notes

Acknowledgements

This research is supported by grant with number 51775415, 2017YFC1308500, and 2018ZDCXL-GY-06-01. Thank Tsinghua University and other member of the organizing committee for providing ECG dataset and platform.

References

  1. 1.
    Warrick, P., Homsi, M.N.: Cardiac arrhythmia detection from ECG combining convolutional and long short-term memory networks. In: 2017 Computing in Cardiology (CinC), pp. 1–4. IEEE (2017)Google Scholar
  2. 2.
    Khazaee, A., Ebrahimzadeh, A.: Classification of electrocardiogram signals with support vector machines and genetic algorithms using power spectral features. Biomed. Signal Process. Control 5, 252–263 (2010)CrossRefGoogle Scholar
  3. 3.
    Li, T., Zhou, M.: ECG classification using wavelet packet entropy and random forests. Entropy 18(8), 285 (2016)CrossRefGoogle Scholar
  4. 4.
    Jadhav, S.M., Nalbalwar, S.L., Ghatol, A.: Artificial neural network based cardiac arrhythmia classification using ECG signal data. In: International Conference on Electronics and Information Engineering (ICEIE), pp. V1-228–V1-231 (2010)Google Scholar
  5. 5.
    He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 770–778 (2016)Google Scholar
  6. 6.
    Shi, B., Bai, X., Yao, C.: An end-to-end trainable neural network for image-based sequence recognition and its application to scene text recognition. IEEE Trans. Pattern Anal. Mach. Intell. 39(11), 2298–2304 (2017)CrossRefGoogle Scholar
  7. 7.
    Young, T., Hazarika, D., Poria, S., Cambria, E.: Recent trends in deep learning based natural language processing. IEEE Comput. Intell. Mag. 13(3), 55–75 (2018)CrossRefGoogle Scholar
  8. 8.
    Hannun, A.Y., et al.: Cardiologist-level arrhythmia detection and classification in ambulatory electrocardiograms using a deep neural network. Nat. Med. 25(1), 65–69 (2019)CrossRefGoogle Scholar
  9. 9.
    Rajpurkar, P., Hannun, A.Y., Haghpanahi, M., Bourn, C., Ng, A.Y.: Cardiologist-level arrhythmia detection with convolutional neural networks. arXiv preprint arXiv:1707.01836 (2017)
  10. 10.
    Zihlmann, M., Perekrestenko, D., Tschannen, M.: Convolutional recurrent neural networks for electrocardiogram classification. In: 2017 Computing in Cardiology (CinC), pp. 1–4. IEEE (2017)Google Scholar
  11. 11.
    Deng, J., Dong, W., Socher, R., Li, L.-J., Li, K., Fei-Fei, L.: ImageNet: a large-scale hierarchical image database. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 248–255 (2009)Google Scholar
  12. 12.
    Lin, T.-Y., et al.: Microsoft COCO: common objects in context. In: Fleet, D., Pajdla, T., Schiele, B., Tuytelaars, T. (eds.) ECCV 2014. LNCS, vol. 8693, pp. 740–755. Springer, Cham (2014).  https://doi.org/10.1007/978-3-319-10602-1_48CrossRefGoogle Scholar
  13. 13.
    Clifford, G.D., et al.: AF Classification from a short single lead ECG recording: the PhysioNet/computing in cardiology challenge 2017. In: 2017 Computing in Cardiology (CinC), pp. 1–4. IEEE (2017)Google Scholar
  14. 14.
    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
  15. 15.
    Ioffe, S., Szegedy, C.: Batch normalization: accelerating deep network training by reducing internal covariate shift. arXiv preprint arXiv:1502.03167 (2015)
  16. 16.
    He, K., Zhang, X., Ren, S., Sun, J.: Delving deep into rectifiers: surpassing human-level performance on imagenet classification. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 1026–1034. IEEE (2015)Google Scholar
  17. 17.
    Zhang, M., Zhou, Z.: A review on multi-label learning algorithms. IEEE Trans. Knowl. Data Eng. 26(8), 1819–1837 (2014)CrossRefGoogle Scholar
  18. 18.
    The First China ECG Intelligent Competition dataset, http://mdi.ids.tsinghua.edu.cn. Accessed 1 July 2019

Copyright information

© Springer Nature Switzerland AG 2019

Authors and Affiliations

  • Longting Chen
    • 1
  • Guanghua Xu
    • 1
    Email author
  • Sicong Zhang
    • 1
  • Jiachen Kuang
    • 1
  • Long Hao
    • 1
  1. 1.Xi’an Jiaotong UniversityXi’anPeople’s Republic of China

Personalised recommendations