Cardiac arrhythmia detection from ECG combining convolutional and long short-term memory networks

Objectives: Atrial fibrillation (AF) is a common heart rhythm disorder associated with deadly and debilitating consequences including heart failure, stroke, poor mental health, reduced quality of life and death. Having an automatic system that diagnoses various types of cardiac arrhythmias would assist cardiologists to initiate appropriate preventive measures and to improve the analysis of cardiac disease. To this end, this paper introduces a new approach to detect and classify automatically cardiac arrhythmias in electrocardiograms (ECG) recordings. Methods: The proposed approach used a combination of Convolution Neural Networks (CNNs) and a sequence of Long Short-Term Memory (LSTM) units, with pooling, dropout and normalization techniques to improve their accuracy. The network predicted a classification at every 18th input sample and we selected the final prediction for classification. Results were cross-validated on the Physionet Challenge 2017 training dataset, which contains 8,528 single lead ECG recordings lasting from 9s to just over 60s. Results: Using the proposed structure and no explicit feature selection, 10-fold stratified cross-validation gave an overall F-measure of 0.83.10±0.015 on the held-out test data (mean ± standard deviation over all folds) and 0.80 on the hidden dataset of the Challenge entry server.

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

[2]  Fai Wong,et al.  Ensemble learning on heartbeat type classification , 2011, Proceedings 2011 International Conference on System Science and Engineering.

[3]  Lovekesh Vig,et al.  Anomaly detection in ECG time signals via deep long short-term memory networks , 2015, 2015 IEEE International Conference on Data Science and Advanced Analytics (DSAA).

[4]  J. S. Sahambi,et al.  Classification of ECG arrhythmias using multi-resolution analysis and neural networks , 2003, TENCON 2003. Conference on Convergent Technologies for Asia-Pacific Region.

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

[6]  Qiao Li,et al.  AF classification from a short single lead ECG recording: The PhysioNet/computing in cardiology challenge 2017 , 2017, 2017 Computing in Cardiology (CinC).

[7]  Patrick Davey ECG at a Glance , 2008 .

[8]  Nikhil Buduma,et al.  Fundamentals of deep learning , 2017 .

[9]  Liqing Zhang,et al.  ECG Feature Extraction and Classification Using Wavelet Transform and Support Vector Machines , 2005, 2005 International Conference on Neural Networks and Brain.