A deep learning approach for real-time detection of atrial fibrillation

Abstract Goal: To develop a robust and real-time approach for automatic detection of atrial fibrillation (AF) in long-term electrocardiogram (ECG) recordings using deep learning (DL). Method: An end-to-end model combining the Convolutional- and Recurrent-Neural Networks (CNN and RNN) was proposed to extract high level features from segments of RR intervals (RRIs) in order to classify them as AF or normal sinus rhythm (NSR). Results: The model was trained and validated on three different databases including a total of 89 subjects. It achieved a sensitivity and specificity of 98.98% and 96.95%, respectively, validated through a 5-fold cross-validation. Additionally, the proposed model was found to be computationally efficient and it was capable of analyzing 24 h of ECG recordings in less than one second. The proposed algorithm was also tested on the unseen datasets to examine its robustness in detecting AF for new recordings which resulted in 98.96% and 86.04% for specificity and sensitivity, respectively. Conclusion: Compared to the state-of-the-art models evaluated on standard benchmark ECG datasets, the proposed model produced better performance in detecting AF. Additionally, since the model learns features directly from the data, it avoids the need for clever/cumbersome feature engineering.

[1]  P. Stein,et al.  Heart rate variability: a measure of cardiac autonomic tone. , 1994, American heart journal.

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

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

[4]  Yoshua Bengio,et al.  Random Search for Hyper-Parameter Optimization , 2012, J. Mach. Learn. Res..

[5]  Alireza Mehrnia,et al.  Automatic detection of atrial fibrillation using stationary wavelet transform and support vector machine , 2015, Comput. Biol. Medicine.

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

[7]  José Luis Rojo-Álvarez,et al.  Feature selection using support vector machines and bootstrap methods for ventricular fibrillation detection , 2012, Expert Syst. Appl..

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

[9]  Yüksel Özbay,et al.  A novel approach for classification of ECG arrhythmias: Type-2 fuzzy clustering neural network , 2009, Expert Syst. Appl..

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

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

[12]  Charu C. Aggarwal,et al.  Neural Networks and Deep Learning , 2018, Springer International Publishing.

[13]  Rabab Kreidieh Ward,et al.  Robust greedy deep dictionary learning for ECG arrhythmia classification , 2017, 2017 International Joint Conference on Neural Networks (IJCNN).

[14]  Behboud Mashoufi,et al.  A new personalized ECG signal classification algorithm using Block-based Neural Network and Particle Swarm Optimization , 2016, Biomed. Signal Process. Control..

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

[16]  Silvia G. Priori,et al.  ACC/AHA/ESC 2006 guidelines for the management of patients with atrial fibrillation: a report of the American College of Cardiology/American Heart Association task force on practice guidelines and the European society of cardiology committee for PRAC , 2006 .

[17]  L. Davies,et al.  The cost of care. , 1994, The Health service journal.

[18]  Dit-Yan Yeung,et al.  Convolutional LSTM Network: A Machine Learning Approach for Precipitation Nowcasting , 2015, NIPS.

[19]  J. R. Moorman,et al.  Accurate estimation of entropy in very short physiological time series: the problem of atrial fibrillation detection in implanted ventricular devices. , 2011, American journal of physiology. Heart and circulatory physiology.

[20]  Matthew Richardson,et al.  Blending LSTMs into CNNs , 2015, ICLR 2016.

[21]  Tara N. Sainath,et al.  Convolutional, Long Short-Term Memory, fully connected Deep Neural Networks , 2015, 2015 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP).

[22]  Ki H. Chon,et al.  Atrial Fibrillation Detection Using an iPhone 4S , 2013, IEEE Transactions on Biomedical Engineering.

[23]  Majid Moavenian,et al.  A qualitative comparison of Artificial Neural Networks and Support Vector Machines in ECG arrhythmias classification , 2010, Expert Syst. Appl..

[24]  Geoffrey E. Hinton,et al.  On the importance of initialization and momentum in deep learning , 2013, ICML.

[25]  Luca T. Mainardi,et al.  A Support Vector Machine approach for reliable detection of atrial fibrillation events , 2013, Computing in Cardiology 2013.

[26]  Thomas Lavergne,et al.  Cost of care distribution in atrial fibrillation patients: the COCAF study. , 2004, American heart journal.

[27]  Sheng Lu,et al.  Automatic Real Time Detection of Atrial Fibrillation , 2009, Annals of Biomedical Engineering.

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

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

[30]  Pawel Plawiak,et al.  Novel methodology of cardiac health recognition based on ECG signals and evolutionary-neural system , 2018, Expert Syst. Appl..

[31]  G. Hindricks,et al.  P-wave evidence as a method for improving algorithm to detect atrial fibrillation in insertable cardiac monitors. , 2014, Heart rhythm.

[32]  J. McMurray,et al.  Cost of an emerging epidemic: an economic analysis of atrial fibrillation in the UK , 2004, Heart.

[33]  Guigang Zhang,et al.  Deep Learning , 2016, Int. J. Semantic Comput..

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

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

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

[37]  Y. Nesterov A method for solving the convex programming problem with convergence rate O(1/k^2) , 1983 .

[38]  Jian Sun,et al.  Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classification , 2015, 2015 IEEE International Conference on Computer Vision (ICCV).

[39]  Yoshua Bengio,et al.  Empirical Evaluation of Gated Recurrent Neural Networks on Sequence Modeling , 2014, ArXiv.

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

[41]  M. Zoni-Berisso,et al.  Epidemiology of atrial fibrillation: European perspective , 2014, Clinical epidemiology.

[42]  Xiaodong Wang,et al.  A Novel Features Learning Method for ECG Arrhythmias Using Deep Belief Networks , 2016, 2016 6th International Conference on Digital Home (ICDH).

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

[44]  YuSung-Nien,et al.  Integration of independent component analysis and neural networks for ECG beat classification , 2008 .

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

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

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

[48]  Navdeep Jaitly,et al.  Towards End-To-End Speech Recognition with Recurrent Neural Networks , 2014, ICML.

[49]  Thomas Lavergne,et al.  Cost of care distribution in atrial fibrillation patients: the COCAF study , 2004 .

[50]  Sung-Nien Yu,et al.  Integration of independent component analysis and neural networks for ECG beat classification , 2008, Expert Syst. Appl..