A novel application of deep learning for single-lead ECG classification

Detecting and classifying cardiac arrhythmias is critical to the diagnosis of patients with cardiac abnormalities. In this paper, a novel approach based on deep learning methodology is proposed for the classification of single-lead electrocardiogram (ECG) signals. We demonstrate the application of the Restricted Boltzmann Machine (RBM) and deep belief networks (DBN) for ECG classification following detection of ventricular and supraventricular heartbeats using single-lead ECG. The effectiveness of this proposed algorithm is illustrated using real ECG signals from the widely-used MIT-BIH database. Simulation results demonstrate that with a suitable choice of parameters, RBM and DBN can achieve high average recognition accuracies of ventricular ectopic beats (93.63%) and of supraventricular ectopic beats (95.57%) at a low sampling rate of 114 Hz. Experimental results indicate that classifiers built into this deep learning-based framework achieved state-of-the art performance models at lower sampling rates and simple features when compared to traditional methods. Further, employing features extracted at a sampling rate of 114 Hz when combined with deep learning provided enough discriminatory power for the classification task. This performance is comparable to that of traditional methods and uses a much lower sampling rate and simpler features. Thus, our proposed deep neural network algorithm demonstrates that deep learning-based methods offer accurate ECG classification and could potentially be extended to other physiological signal classifications, such as those in arterial blood pressure (ABP), nerve conduction (EMG), and heart rate variability (HRV) studies.

[1]  Zhang Yue,et al.  Classification of Electrocardiogram Signals with Deep Belief Networks , 2014, CSE 2014.

[2]  Jian‐He Xu,et al.  Department of Bioengineering , 2010 .

[3]  Christian Szegedy,et al.  DeepPose: Human Pose Estimation via Deep Neural Networks , 2013, 2014 IEEE Conference on Computer Vision and Pattern Recognition.

[4]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[5]  Jiwen Lu,et al.  Discriminative Deep Metric Learning for Face Verification in the Wild , 2014, 2014 IEEE Conference on Computer Vision and Pattern Recognition.

[6]  W.J. Tompkins,et al.  ECG beat detection using filter banks , 1999, IEEE Transactions on Biomedical Engineering.

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

[8]  W.J. Tompkins,et al.  A patient-adaptable ECG beat classifier using a mixture of experts approach , 1997, IEEE Transactions on Biomedical Engineering.

[9]  Arantza Illarramendi,et al.  Real-time classification of ECGs on a PDA , 2005, IEEE Transactions on Information Technology in Biomedicine.

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

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

[12]  Michel Verleysen,et al.  Weighted Conditional Random Fields for Supervised Interpatient Heartbeat Classification , 2012, IEEE Transactions on Biomedical Engineering.

[13]  Stanislaw Osowski,et al.  Support vector machine-based expert system for reliable heartbeat recognition , 2004, IEEE Transactions on Biomedical Engineering.

[14]  Geoffrey E. Hinton,et al.  3D Object Recognition with Deep Belief Nets , 2009, NIPS.

[15]  Marc'Aurelio Ranzato,et al.  Building high-level features using large scale unsupervised learning , 2011, 2013 IEEE International Conference on Acoustics, Speech and Signal Processing.

[16]  Mohammed Ismail,et al.  Adaptive technique for P and T wave delineation in electrocardiogram signals , 2014, 2014 36th Annual International Conference of the IEEE Engineering in Medicine and Biology Society.

[17]  Lei Wang,et al.  A restricted Boltzmann machine based two-lead electrocardiography classification , 2015, 2015 IEEE 12th International Conference on Wearable and Implantable Body Sensor Networks (BSN).

[18]  Geoffrey E. Hinton.  Training Products of Experts by Minimizing Contrastive Divergence , 2002, Neural Computation.

[19]  Dong Yu,et al.  Context-Dependent Pre-Trained Deep Neural Networks for Large-Vocabulary Speech Recognition , 2012, IEEE Transactions on Audio, Speech, and Language Processing.

[20]  Liqing Zhang,et al.  ECG Arrhythmias Recognition System Based on Independent Component Analysis Feature Extraction , 2006, TENCON 2006 - 2006 IEEE Region 10 Conference.

[21]  Honglak Lee,et al.  Convolutional deep belief networks for scalable unsupervised learning of hierarchical representations , 2009, ICML '09.

[22]  Jianqing Li,et al.  Patient-Specific Deep Architectural Model for ECG Classification , 2017, Journal of healthcare engineering.

[23]  Yann LeCun,et al.  Pedestrian Detection with Unsupervised Multi-stage Feature Learning , 2012, 2013 IEEE Conference on Computer Vision and Pattern Recognition.

[24]  U. Raghavendra,et al.  Automated identification of shockable and non-shockable life-threatening ventricular arrhythmias using convolutional neural network , 2018, Future Gener. Comput. Syst..

[25]  L. Glass.  Cardiac Oscillations and Arrhythmia Analysis , 2006 .

[26]  Mosabber Uddin Ahmed,et al.  A study of recursive least squares (RLS) adaptive filter algorithm in noise removal from ECG signals , 2015, 2015 International Conference on Informatics, Electronics & Vision (ICIEV).

[27]  Xiaogang Wang,et al.  Hybrid Deep Learning for Face Verification , 2013, ICCV.

[28]  C. Tracy,et al.  ACC/AHA clinical competence statement on electrocardiography and ambulatory electrocardiography. A report of the ACC/AHA/ACP-ASIM Task Force on Clinical Competence (ACC/AHA Committee to Develop a Clinical Competence Statement on Electrocardiography and Ambulatory Electrocardiography). , 2001, Journal of the American College of Cardiology.

[29]  Jürgen Schmidhuber,et al.  Deep learning in neural networks: An overview , 2014, Neural Networks.

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

[31]  Juan Pablo Martínez,et al.  Heartbeat Classification Using Feature Selection Driven by Database Generalization Criteria , 2011, IEEE Transactions on Biomedical Engineering.

[32]  Dong Yu,et al.  Conversational Speech Transcription Using Context-Dependent Deep Neural Networks , 2012, ICML.

[33]  G. Carrault,et al.  Comparing wavelet transforms for recognizing cardiac patterns , 1995 .

[34]  Xiaogang Wang,et al.  Deep Learning Face Representation by Joint Identification-Verification , 2014, NIPS.

[35]  Tony Basil,et al.  A Comparison of Statistical Machine Learning Methods in Heartbeat Detection and Classification , 2012, BDA.

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

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

[38]  Stefan Carlsson,et al.  CNN Features Off-the-Shelf: An Astounding Baseline for Recognition , 2014, 2014 IEEE Conference on Computer Vision and Pattern Recognition Workshops.

[39]  Tara N. Sainath,et al.  Deep Neural Networks for Acoustic Modeling in Speech Recognition: The Shared Views of Four Research Groups , 2012, IEEE Signal Processing Magazine.

[40]  Ken Perlin,et al.  Real-Time Continuous Pose Recovery of Human Hands Using Convolutional Networks , 2014, ACM Trans. Graph..

[41]  G Bortolan,et al.  Premature ventricular contraction classification by the Kth nearest-neighbours rule , 2005, Physiological measurement.

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

[43]  Kenneth E. Barner,et al.  Leveraging a discriminative dictionary learning algorithm for single-lead ECG classification , 2015, 2015 41st Annual Northeast Biomedical Engineering Conference (NEBEC).

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

[45]  Yann LeCun,et al.  Learning a similarity metric discriminatively, with application to face verification , 2005, 2005 IEEE Computer Society Conference on Computer Vision and Pattern Recognition (CVPR'05).

[46]  Stanislaw Osowski,et al.  ECG beat recognition using fuzzy hybrid neural network , 2001, IEEE Trans. Biomed. Eng..

[47]  Rob Fergus,et al.  Visualizing and Understanding Convolutional Networks , 2013, ECCV.

[48]  Pascal Vincent,et al.  Stacked Denoising Autoencoders: Learning Useful Representations in a Deep Network with a Local Denoising Criterion , 2010, J. Mach. Learn. Res..

[49]  Ming Yang,et al.  DeepFace: Closing the Gap to Human-Level Performance in Face Verification , 2014, 2014 IEEE Conference on Computer Vision and Pattern Recognition.

[50]  Brian Kingsbury,et al.  New types of deep neural network learning for speech recognition and related applications: an overview , 2013, 2013 IEEE International Conference on Acoustics, Speech and Signal Processing.

[51]  Li Deng,et al.  A deep convolutional neural network using heterogeneous pooling for trading acoustic invariance with phonetic confusion , 2013, 2013 IEEE International Conference on Acoustics, Speech and Signal Processing.

[52]  Jenna Wiens,et al.  Active Learning Applied to Patient-Adaptive Heartbeat Classification , 2010, NIPS.

[53]  Carsten Peterson,et al.  Clustering ECG complexes using Hermite functions and self-organizing maps , 2000, IEEE Trans. Biomed. Eng..

[54]  Philip de Chazal,et al.  Detection of supraventricular and ventricular ectopic beats using a single lead ECG , 2013, EMBC.

[55]  Dan Wang,et al.  Modeling Physiological Data with Deep Belief Networks. , 2013, International journal of information and education technology.

[56]  Cordelia Schmid,et al.  DeepFlow: Large Displacement Optical Flow with Deep Matching , 2013, 2013 IEEE International Conference on Computer Vision.

[57]  Geoffrey E. Hinton.  A Practical Guide to Training Restricted Boltzmann Machines , 2012, Neural Networks: Tricks of the Trade.