SE-ECGNet: A Multi-scale Deep Residual Network with Squeeze-and-Excitation Module for ECG Signal Classification

The classification of electrocardiogram (ECG) signals, which takes much time and suffers from a high rate of misjudgment, is recognized as an extremely challenging task for cardiologists. The major difficulty of the ECG signals classification is caused by the long-term sequence dependencies. Most existing approaches for ECG signal classification use Recurrent Neural Network models, e.g., LSTM and GRU, which are unable to extract accurate features for such long sequences. Other approaches utilize 1-Dimensional Convolutional Neural Network (CNN), such as ResNet or its variant, and they can not make good use of the multi-lead information from ECG signals.Based on the above observations, we develop a multi-scale deep residual network for the ECG signal classification task. We are the first to propose to treat the multi-lead signal as a 2-dimensional matrix and combines multi-scale 2-D convolution blocks with 1-D convolution blocks for feature extraction. Our proposed model achieves 99.2% F1-score in the MIT-BIH dataset and 89.4% F1-score in Alibaba dataset and outperforms the state-of-the-art performance by 2% and 3%, respectively, view related code and data at https://github.com/Amadeuszhao/SE-ECGNet

[1]  Tingting Lv,et al.  Bidirectional Recurrent Neural Network And Convolutional Neural Network (BiRCNN) For ECG Beat Classification , 2018, 2018 40th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC).

[2]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[3]  Rodrigo Castañeda-Miranda,et al.  DSP-based arrhythmia classification using wavelet transform and probabilistic neural network , 2017, Biomed. Signal Process. Control..

[4]  Pankoo Kim,et al.  A Deep Bidirectional GRU Network Model for Biometric Electrocardiogram Classification Based on Recurrent Neural Networks , 2019, IEEE Access.

[5]  Fatemeh Afghah,et al.  Inter- and Intra- Patient ECG Heartbeat Classification for Arrhythmia Detection: A Sequence to Sequence Deep Learning Approach , 2018, ICASSP 2019 - 2019 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP).

[6]  Enhua Wu,et al.  Squeeze-and-Excitation Networks , 2017, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[7]  Gurkan Ozturk,et al.  Arrhythmia Classification via k-Means Based Polyhedral Conic Functions Algorithm , 2016, 2016 International Conference on Computational Science and Computational Intelligence (CSCI).

[8]  Meng Wu,et al.  ENCASE: An ENsemble ClASsifiEr for ECG classification using expert features and deep neural networks , 2017, 2017 Computing in Cardiology (CinC).

[9]  Juan Pablo Martínez,et al.  An Automatic Patient-Adapted ECG Heartbeat Classifier Allowing Expert Assistance , 2012, IEEE Transactions on Biomedical Engineering.

[10]  Qiang Zhang,et al.  Classification of ECG signals based on 1D convolution neural network , 2017, 2017 IEEE 19th International Conference on e-Health Networking, Applications and Services (Healthcom).

[11]  Cláudia Brito,et al.  Electrocardiogram Beat-Classification Based on a ResNet Network , 2019, MedInfo.

[12]  Kyungtae Kang,et al.  PcHD: Personalized classification of heartbeat types using a decision tree , 2014, Comput. Biol. Medicine.

[13]  Xiaolong Zhai,et al.  Automated ECG Classification Using Dual Heartbeat Coupling Based on Convolutional Neural Network , 2018, IEEE Access.

[14]  Keerthi Ram,et al.  ECGNet: Deep Network for Arrhythmia Classification , 2018, 2018 IEEE International Symposium on Medical Measurements and Applications (MeMeA).

[15]  Kaiming He,et al.  Focal Loss for Dense Object Detection , 2017, 2017 IEEE International Conference on Computer Vision (ICCV).

[16]  Özal Yildirim,et al.  A novel wavelet sequence based on deep bidirectional LSTM network model for ECG signal classification , 2018, Comput. Biol. Medicine.

[17]  Man-Wai Mak,et al.  Towards End-to-End ECG Classification With Raw Signal Extraction and Deep Neural Networks , 2019, IEEE Journal of Biomedical and Health Informatics.

[18]  Rik Vullings,et al.  An Adaptive Kalman Filter for ECG Signal Enhancement , 2011, IEEE Transactions on Biomedical Engineering.

[19]  Bashar A. Rajoub.  An efficient coding algorithm for the compression of ECG signals using the wavelet transform , 2002, IEEE Transactions on Biomedical Engineering.

[20]  Pablo Laguna,et al.  A wavelet-based ECG delineator: evaluation on standard databases , 2004, IEEE Transactions on Biomedical Engineering.

[21]  Andrew Y. Ng,et al.  Cardiologist-Level Arrhythmia Detection with Convolutional Neural Networks , 2017, ArXiv.

[22]  Hadi Sadoghi Yazdi,et al.  ECG Arrhythmia Classification with Support Vector Machines and Genetic Algorithm , 2009, 2009 Third UKSim European Symposium on Computer Modeling and Simulation.

[23]  Jian Sun,et al.  Identity Mappings in Deep Residual Networks , 2016, ECCV.

[24]  Donghui Zhang,et al.  Wavelet Approach for ECG Baseline Wander Correction and Noise Reduction , 2005, 2005 IEEE Engineering in Medicine and Biology 27th Annual Conference.

[25]  Kailash Chandra Ray,et al.  Efficient methodology for electrocardiogram beat classification , 2016, IET Signal Process..

[26]  Dumitru Erhan,et al.  Going deeper with convolutions , 2014, 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[27]  M. Turakhia,et al.  Diagnostic utility of a novel leadless arrhythmia monitoring device. , 2013, The American journal of cardiology.

[28]  Alan R. Jones,et al.  Fast Fourier Transform , 1970, SIGP.

[29]  Ross B. Girshick,et al.  Focal Loss for Dense Object Detection , 2017, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[30]  Gert Cauwenberghs,et al.  Incremental and Decremental Support Vector Machine Learning , 2000, NIPS.

[31]  J. Platt.  Sequential Minimal Optimization : A Fast Algorithm for Training Support Vector Machines , 1998 .

[32]  C. Li,et al.  Detection of ECG characteristic points using wavelet transforms. , 1995, IEEE transactions on bio-medical engineering.

[33]  Jiann-Shiun Yuan,et al.  ECG Arrhythmia Classification Using Transfer Learning from 2- Dimensional Deep CNN Features , 2018, 2018 IEEE Biomedical Circuits and Systems Conference (BioCAS).

[34]  S. Sriram,et al.  Signal denoising techniques for partial discharge measurements , 2005, IEEE Transactions on Dielectrics and Electrical Insulation.

[35]  G.B. Moody,et al.  The impact of the MIT-BIH Arrhythmia Database , 2001, IEEE Engineering in Medicine and Biology Magazine.

[36]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.