SpEC: A system for patient specific ECG beat classification using deep residual network

Abstract Cardiac disease is a severe medical issue that causes 16 million deaths in the world every year. Electrocardiogram (ECG) is a non-invasive technique used to detect various cardiac disorders. One of the major causes of cardiac arrest is an arrhythmia. It is defined as any deviation of the heartbeat from the normal sinus rhythm (NSR). Furthermore, ECG beat classification is essential to detect life-threatening cardiac arrhythmias. The major limitations of the traditional ECG beat classification systems are the requirement of an extensive training dataset to train the model and inconsistent performance for the detection of ventricular and supraventricular ectopic (V and S) beats. To overcome these limitations, a system denoted as SpEC is proposed in this work based on Stockwell transform (ST) and two-dimensional residual network (2D-ResNet) for improvement of ECG beat classification technique with a limited amount of training data. ST, which is used to represent the ECG signal into a time-frequency domain, provides frequency invariant amplitude response and dynamic resolution. The resultant ST images are applied as input to the proposed 2D-ResNet to classify five different types of ECG beats in a patient-specific way as recommended by the Association for the Advancement of Medical Instrumentation (AAMI). The proposed SpEC system achieved an overall accuracy (Acc) of 99.73%, sensitivity (Sen) = 98.84%, Specificity (Spe) = 99.50%, Positive predictivity (Ppr) = 98.20% on MIT-BIH arrhythmia database, and shows an overall Acc of 89.87% on real-time acquired ECG dataset with classification time of single ECG beat image = 0.2365 (s) in detecting of five arrhythmia classes. The proposed method shows better performance on both the database compared to the earlier reported state-of-art techniques.

[1]  Bin Yao,et al.  ECG Arrhythmia Classification Using STFT-Based Spectrogram and Convolutional Neural Network , 2019, IEEE Access.

[2]  Daeyoung Kim,et al.  ECG arrhythmia classification using a 2-D convolutional neural network , 2018, ArXiv.

[3]  Bernadette Dorizzi,et al.  ECG signal analysis through hidden Markov models , 2006, IEEE Transactions on Biomedical Engineering.

[4]  George Manis,et al.  Heartbeat Time Series Classification With Support Vector Machines , 2009, IEEE Transactions on Information Technology in Biomedicine.

[5]  Qiang Yang,et al.  A Survey on Transfer Learning , 2010, IEEE Transactions on Knowledge and Data Engineering.

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

[7]  Xiaohe Chen,et al.  Arrhythmia Recognition and Classification Using ECG Morphology and Segment Feature Analysis , 2019, IEEE/ACM Transactions on Computational Biology and Bioinformatics.

[8]  Moncef Gabbouj,et al.  A Generic and Robust System for Automated Patient-Specific Classification of ECG Signals , 2009, IEEE Transactions on Biomedical Engineering.

[9]  Huifang Huang,et al.  A new hierarchical method for inter-patient heartbeat classification using random projections and RR intervals , 2014, BioMedical Engineering OnLine.

[10]  Samit Ari,et al.  Patient-specific ECG beat classification technique. , 2014, Healthcare technology letters.

[11]  Yüksel Özbay,et al.  Integration of type-2 fuzzy clustering and wavelet transform in a neural network based ECG classifier , 2011, Expert Syst. Appl..

[12]  Yaoqin Xie,et al.  A Novel Wearable Electrocardiogram Classification System Using Convolutional Neural Networks and Active Learning , 2019, IEEE Access.

[13]  Sanjay L. Nalbalwar,et al.  ECG arrhythmia classification using modular neural network model , 2010, 2010 IEEE EMBS Conference on Biomedical Engineering and Sciences (IECBES).

[14]  Farid Melgani,et al.  Evolutionary Computation Approach to ECG Signal Classification , 2008, BIOSIGNALS.

[15]  Samit Ari,et al.  A system for automatic cardiac arrhythmia recognition using electrocardiogram signal , 2019 .

[16]  Hlaing Minn,et al.  A Patient-Adaptive Profiling Scheme for ECG Beat Classification , 2010, IEEE Transactions on Information Technology in Biomedicine.

[17]  Lalu Mansinha,et al.  Localization of the complex spectrum: the S transform , 1996, IEEE Trans. Signal Process..

[18]  Gaetano D Gargiulo,et al.  Towards Real-Time Heartbeat Classification: Evaluation of Nonlinear Morphological Features and Voting Method , 2019, Sensors.

[19]  Pawel Plawiak,et al.  Novel genetic ensembles of classifiers applied to myocardium dysfunction recognition based on ECG signals , 2017, Swarm Evol. Comput..

[20]  Pawe Pawiak,et al.  Novel methodology of cardiac health recognition based on ECG signals and evolutionary-neural system , 2018 .

[21]  Amjed S. Al-Fahoum,et al.  A quantitative analysis approach for cardiac arrhythmia classification using higher order spectral techniques , 2005, IEEE Transactions on Biomedical Engineering.

[22]  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).

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

[24]  U. Rajendra Acharya,et al.  Automated diagnosis of arrhythmia using combination of CNN and LSTM techniques with variable length heart beats , 2018, Comput. Biol. Medicine.

[25]  Madhuchhanda Mitra,et al.  Application of Cross Wavelet Transform for ECG Pattern Analysis and Classification , 2014, IEEE Transactions on Instrumentation and Measurement.

[26]  Hongxing Liu,et al.  ECG Heartbeat Classification Using Convolutional Neural Networks , 2020, IEEE Access.

[27]  José Carlos Príncipe,et al.  Time-Based Compression and Classification of Heartbeats , 2012, IEEE Transactions on Biomedical Engineering.

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

[29]  Willis J. Tompkins,et al.  A Real-Time QRS Detection Algorithm , 1985, IEEE Transactions on Biomedical Engineering.

[30]  Philip de Chazal,et al.  A Patient-Adapting Heartbeat Classifier Using ECG Morphology and Heartbeat Interval Features , 2006, IEEE Transactions on Biomedical Engineering.

[31]  Gregory T. A. Kovacs,et al.  Robust Neural-Network-Based Classification of Premature Ventricular Contractions Using Wavelet Transform and Timing Interval Features , 2006, IEEE Transactions on Biomedical Engineering.

[32]  U. Rajendra Acharya,et al.  Novel deep genetic ensemble of classifiers for arrhythmia detection using ECG signals , 2019, Neural Computing and Applications.

[33]  G.G. Cano,et al.  An approach to cardiac arrhythmia analysis using hidden Markov models , 1990, IEEE Transactions on Biomedical Engineering.

[34]  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.

[35]  Moncef Gabbouj,et al.  Real-Time Patient-Specific ECG Classification by 1-D Convolutional Neural Networks , 2016, IEEE Transactions on Biomedical Engineering.

[36]  Ming-Feng Yeh,et al.  ECG beat classification using GreyART network , 2007 .

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

[38]  Manu Thomas,et al.  Automatic ECG arrhythmia classification using dual tree complex wavelet based features , 2015 .

[39]  B. V. K. Vijaya Kumar,et al.  Heartbeat Classification Using Morphological and Dynamic Features of ECG Signals , 2012, IEEE Transactions on Biomedical Engineering.