Automatic Classification of CAD ECG Signals With SDAE and Bidirectional Long Short-Term Network

Coronary artery disease (CAD) has been one of main causes of heart diseases globally. The electrocardiogram (ECG) is a widely used diagnostic tool to monitor patients’ heart activities, and medical personnel need to judge whether there are abnormal heartbeats according to captured results. Therefore, it is significant to identify ECG signals accurately and fast. In this paper, a fast and accurate electrocardiogram (ECG) classification system based on deep learning is proposed. In our model, stacked denoising autoencoders (SDAE), as encoder, automatically learns semantic encoding of heartbeats without any complex feature extraction in unsupervised way. Then bidirectional LSTM (Bi-LSTM) classifier achieves classification of heartbeats with semantic encoding. SDAE implements noise-reduction while Bi-LSTM takes full advantage of temporal information in data. At the same time, this method relieves impacts from unbalanced data by employing cost-sensitive loss function. We validate our model on MIT-BIH Arrhythmias Database, SVDB and NSTDB respectively. Compared with state-of-art methods, the final result verify that this newly proposed method not only has high accuracy but also boosts classifying efficiency.

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

[2]  Pu Wang,et al.  LSTM-Based Auto-Encoder Model for ECG Arrhythmias Classification , 2020, IEEE Transactions on Instrumentation and Measurement.

[3]  Jürgen Schmidhuber,et al.  Framewise phoneme classification with bidirectional LSTM and other neural network architectures , 2005, Neural Networks.

[4]  Marina Ronzhina,et al.  ECG features and methods for automatic classification of ventricular premature and ischemic heartbeats: A comprehensive experimental study , 2017, Scientific Reports.

[5]  U. Rajendra Acharya,et al.  ECG beat classification using PCA, LDA, ICA and Discrete Wavelet Transform , 2013, Biomed. Signal Process. Control..

[6]  Philip de Chazal,et al.  Automatic classification of heartbeats using ECG morphology and heartbeat interval features , 2004, IEEE Transactions on Biomedical Engineering.

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

[8]  Jürgen Schmidhuber,et al.  Long Short-Term Memory , 1997, Neural Computation.

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

[10]  Yu-Liang Hsu,et al.  ECG arrhythmia classification using a probabilistic neural network with a feature reduction method , 2013, Neurocomputing.

[11]  U. Rajendra Acharya,et al.  Application of stacked convolutional and long short-term memory network for accurate identification of CAD ECG signals , 2018, Comput. Biol. Medicine.

[12]  W. Pedrycz,et al.  Hyperbox classifiers for ECG beat analysis , 2007, 2007 Computers in Cardiology.

[13]  Mehmet Korürek,et al.  A new arrhythmia clustering technique based on Ant Colony Optimization , 2008, J. Biomed. Informatics.

[14]  Majid Moavenian,et al.  A comparative study of DWT, CWT and DCT transformations in ECG arrhythmias classification , 2010, Expert Syst. Appl..

[15]  A. Kandaswamy,et al.  ECG arrhythmia classification based on logistic model tree , 2009 .

[16]  Miguel C. Soriano,et al.  Electrocardiogram Classification Using Reservoir Computing With Logistic Regression , 2015, IEEE Journal of Biomedical and Health Informatics.

[17]  Wei Lu,et al.  Feature fusion for imbalanced ECG data analysis , 2018, Biomed. Signal Process. Control..

[18]  Chandan Chakraborty,et al.  Application of principal component analysis to ECG signals for automated diagnosis of cardiac health , 2012, Expert Syst. Appl..

[19]  Ali Ghaffari,et al.  ECG arrhythmia recognition via a neuro-SVM-KNN hybrid classifier with virtual QRS image-based geometrical features , 2012, Expert Syst. Appl..

[20]  Yee Whye Teh,et al.  A Fast Learning Algorithm for Deep Belief Nets , 2006, Neural Computation.

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

[22]  Sebastian Zaunseder,et al.  Optimization of ECG Classification by Means of Feature Selection , 2011, IEEE Transactions on Biomedical Engineering.

[23]  Chengyu Liu,et al.  Combining Low-dimensional Wavelet Features and Support Vector Machine for Arrhythmia Beat Classification , 2017, Scientific Reports.

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

[25]  Paul Honeine,et al.  PCA and KPCA of ECG signals with binary SVM classification , 2011, 2011 IEEE Workshop on Signal Processing Systems (SiPS).

[26]  Elif Derya Übeyli.  Recurrent neural networks employing Lyapunov exponents for analysis of ECG signals , 2010, Expert Syst. Appl..

[27]  Jeffrey M. Hausdorff,et al.  Physionet: Components of a New Research Resource for Complex Physiologic Signals". Circu-lation Vol , 2000 .

[28]  U. Rajendra Acharya,et al.  Automated detection of coronary artery disease using different durations of ECG segments with convolutional neural network , 2017, Knowl. Based Syst..