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.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
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)
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)
Li, T., Zhou, M.: ECG classification using wavelet packet entropy and random forests. Entropy 18(8), 285 (2016)
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)
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)
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)
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)
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)
Rajpurkar, P., Hannun, A.Y., Haghpanahi, M., Bourn, C., Ng, A.Y.: Cardiologist-level arrhythmia detection with convolutional neural networks. ar**v preprint ar**v:1707.01836 (2017)
Zihlmann, M., Perekrestenko, D., Tschannen, M.: Convolutional recurrent neural networks for electrocardiogram classification. In: 2017 Computing in Cardiology (CinC), pp. 1–4. IEEE (2017)
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)
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_48
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)
He, K., Zhang, X., Ren, S., Sun, J.: Identity map**s 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
Ioffe, S., Szegedy, C.: Batch normalization: accelerating deep network training by reducing internal covariate shift. ar**v preprint ar**v:1502.03167 (2015)
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)
Zhang, M., Zhou, Z.: A review on multi-label learning algorithms. IEEE Trans. Knowl. Data Eng. 26(8), 1819–1837 (2014)
The First China ECG Intelligent Competition dataset, http://mdi.ids.tsinghua.edu.cn. Accessed 1 July 2019
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.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2019 Springer Nature Switzerland AG
About this paper
Cite this paper
Chen, L., Xu, G., Zhang, S., Kuang, J., Hao, L. (2019). Transfer Learning for Electrocardiogram Classification Under Small Dataset. 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_6
Download citation
DOI: https://doi.org/10.1007/978-3-030-33327-0_6
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)