Arrhythmic Heartbeat Classification Using 2D Convolutional Neural Networks

Abstract Background Electrocardiogram (ECG) is a method of recording the electrical activity of the heart and it provides a diagnostic means for heart-related diseases. Arrhythmia is any irregularity of the heartbeat that causes an abnormality in the heart rhythm. Early detection of arrhythmia has great importance to prevent many diseases. Manual analysis of ECG recordings is not practical for quickly identifying arrhythmias that may cause sudden deaths. Hence, many studies have been presented to develop computer-aided-diagnosis (CAD) systems to automatically identify arrhythmias. Methods This paper proposes a novel deep learning approach to identify arrhythmias in ECG signals. The proposed approach identifies arrhythmia classes using Convolutional Neural Network (CNN) trained by two-dimensional (2D) ECG beat images. Firstly, ECG signals, which consist of 5 different arrhythmias, are segmented into heartbeats which are transformed into 2D grayscale images. Afterward, the images are used as input for training a new CNN architecture to classify heartbeats. Results The experimental results show that the classification performance of the proposed approach reaches an overall accuracy of 99.7%, sensitivity of 99.7%, and specificity of 99.22% in the classification of five different ECG arrhythmias. Further, the proposed CNN architecture is compared to other popular CNN architectures such as LeNet and ResNet-50 to evaluate the performance of the study. Conclusions Test results demonstrate that the deep network trained by ECG images provides outstanding classification performance of arrhythmic ECG signals and outperforms similar network architectures. Moreover, the proposed method has lower computational costs compared to existing methods and is more suitable for mobile device-based diagnosis systems as it does not involve any complex preprocessing process. Hence, the proposed approach provides a simple and robust automatic cardiac arrhythmia detection scheme for the classification of ECG arrhythmias.

[1]  Nima Hatami,et al.  Classification of time-series images using deep convolutional neural networks , 2017, International Conference on Machine Vision.

[2]  Chun-Cheng Lin,et al.  Heartbeat Classification Using Normalized RR Intervals and Wavelet Features , 2014, 2014 International Symposium on Computer, Consumer and Control.

[3]  U. Rajendra Acharya,et al.  Classification of myocardial infarction with multi-lead ECG signals and deep CNN , 2019, Pattern Recognit. Lett..

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

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

[6]  Aydin Akan,et al.  EEG-based emotion recognition with deep convolutional neural networks , 2020, Biomedizinische Technik. Biomedical engineering.

[7]  Ram Bilas Pachori,et al.  Joint Time-Frequency Domain-Based CAD Disease Sensing System Using ECG Signals , 2019, IEEE Sensors Journal.

[8]  Julian M. W. Quinn,et al.  Deep convolutional neural networks based ECG beats classification to diagnose cardiovascular conditions , 2021, Biomedical Engineering Letters.

[9]  Saroj Kumar Pandey,et al.  ECG Arrhythmia Classification Using Artificial Neural Networks , 2019 .

[10]  Howida A. Shedeed,et al.  Generalization of Convolutional Neural Networks for ECG Classification Using Generative Adversarial Networks , 2020, IEEE Access.

[11]  Jing-Hao Xue,et al.  Adjusting the imbalance ratio by the dimensionality of imbalanced data , 2020, Pattern Recognit. Lett..

[12]  M. I. Owis,et al.  A novel technique for cardiac arrhythmia classification using spectral correlation and support vector machines , 2015, Expert systems with applications.

[13]  Jinsul Kim,et al.  An Automated ECG Beat Classification System Using Convolutional Neural Networks , 2016, 2016 6th International Conference on IT Convergence and Security (ICITCS).

[14]  William Robson Schwartz,et al.  ECG-based heartbeat classification for arrhythmia detection: A survey , 2016, Comput. Methods Programs Biomed..

[15]  Giuseppe De Pietro,et al.  A deep learning approach for ECG-based heartbeat classification for arrhythmia detection , 2018, Future Gener. Comput. Syst..

[16]  Shalin Savalia,et al.  Cardiac Arrhythmia Classification by Multi-Layer Perceptron and Convolution Neural Networks , 2018, Bioengineering.

[17]  U. Rajendra Acharya,et al.  Arrhythmia detection using deep convolutional neural network with long duration ECG signals , 2018, Comput. Biol. Medicine.

[18]  Ahmet Mert,et al.  ECG feature extraction based on the bandwidth properties of variational mode decomposition , 2016, Physiological measurement.

[19]  Huazhong Yang,et al.  Patient-specific ECG classification based on recurrent neural networks and clustering technique , 2017, 2017 13th IASTED International Conference on Biomedical Engineering (BioMed).

[20]  U. Rajendra Acharya,et al.  Automated detection of arrhythmias using different intervals of tachycardia ECG segments with convolutional neural network , 2017, Inf. Sci..

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

[22]  Zhengchun Hua,et al.  Automated arrhythmia classification based on a combination network of CNN and LSTM , 2020, Biomed. Signal Process. Control..

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

[24]  Ruxin Wang,et al.  Multi-class Arrhythmia detection from 12-lead varied-length ECG using Attention-based Time-Incremental Convolutional Neural Network , 2020, Inf. Fusion.

[25]  Raghu Nanjundegowda,et al.  Arrhythmia Detection Based on Hybrid Features of T-Wave in Electrocardiogram , 2018, Advances in Systems Analysis, Software Engineering, and High Performance Computing.

[26]  U. Rajendra Acharya,et al.  A deep convolutional neural network model to classify heartbeats , 2017, Comput. Biol. Medicine.

[27]  Jaya Prakash Sahoo,et al.  Analysis of ECG signal for Detection of Cardiac Arrhythmias , 2011 .

[28]  Naif Alajlan,et al.  Deep learning approach for active classification of electrocardiogram signals , 2016, Inf. Sci..

[29]  Santanu Sahoo,et al.  Multiresolution wavelet transform based feature extraction and ECG classification to detect cardiac abnormalities , 2017 .

[30]  Sung Bum Pan,et al.  A Study on User Recognition Using 2D ECG Image Based on Ensemble Networks for Intelligent Vehicles , 2019, Wirel. Commun. Mob. Comput..

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

[32]  Aydin Akan,et al.  Cardiac Arrhythmia Detection from 2D ECG Images by Using Deep Learning Technique , 2019, 2019 Medical Technologies Congress (TIPTEKNO).

[33]  Dong-Joon Choi,et al.  Implementation and Field Trials of OFDM-Based Digital Video Broadcasting System in Commercial Broadcasting Network for Multichannel UHD Service , 2019, Wirel. Commun. Mob. Comput..

[34]  M. K. Moridani,et al.  An Efficient Automated Algorithm for Distinguishing Normal and Abnormal ECG Signal , 2019 .

[35]  Aydin Akan,et al.  Arrhythmia Detection on ECG Signals by Using Empirical Mode Decomposition , 2018, 2018 Medical Technologies National Congress (TIPTEKNO).

[36]  U. Rajendra Acharya,et al.  Deep learning for healthcare applications based on physiological signals: A review , 2018, Comput. Methods Programs Biomed..

[37]  Azeddine Mjahad,et al.  Ventricular Fibrillation and Tachycardia detection from surface ECG using time-frequency representation images as input dataset for machine learning , 2017, Comput. Methods Programs Biomed..

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

[39]  Aboul Ella Hassanien,et al.  ECG signals classification: a review , 2017, Int. J. Intell. Eng. Informatics.

[40]  Min Zhou,et al.  ECG Classification Using Wavelet Packet Entropy and Random Forests , 2016, Entropy.

[41]  Li Wan,et al.  Heartbeat classification using deep residual convolutional neural network from 2-lead electrocardiogram. , 2019, Journal of electrocardiology.

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

[43]  Geoffrey E. Hinton,et al.  Deep Learning , 2015, Nature.

[44]  Ozlem Karabiber Cura,et al.  Abnormal ECG Beat Detection Based on Convolutional Neural Networks , 2020, 2020 Medical Technologies Congress (TIPTEKNO).