Multi-Scale and Attention based ResNet for Heartbeat Classification

This paper presents a novel deep learning framework for the electrocardiogram (ECG) heartbeat classification. Although there have been some studies with excellent overall accuracy, these studies have not been very accurate in the diagnosis of arrhythmia classes especially such as supraventricular ectopic beat (SVEB) and ventricular ectopic beat (VEB). In our work, we propose a Multi-Scale and Attention based Res Net for heartbeat classification in intra-patient and inter-patient paradigms respectively. Firstly, we extract shallow features from a convolutional layer. Secondly, the shallow features are sent into three branches with different convolution kernels in order to combine receptive fields of different sizes. Finally, fully connected layers are used to classify the heartbeat. Besides, we design a new attention mechanism based on the characteristics of heartbeat data. At last, extensive experiments on benchmark dataset demonstrate the effectiveness of our proposed model.

[1]  Kup-Sze Choi,et al.  Heartbeat classification using disease-specific feature selection , 2014, Comput. Biol. Medicine.

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

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

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

[5]  Yuxi Li,et al.  Deep Reinforcement Learning: An Overview , 2017, ArXiv.

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

[7]  Mohammad Ali Tinati,et al.  Cardiac arrhythmia classification using statistical and mixture modeling features of ECG signals , 2016, Pattern Recognit. Lett..

[8]  Ye Li,et al.  Multiscaled Fusion of Deep Convolutional Neural Networks for Screening Atrial Fibrillation From Single Lead Short ECG Recordings , 2018, IEEE Journal of Biomedical and Health Informatics.

[9]  Nasser Mozayani,et al.  Learning to predict where to look in interactive environments using deep recurrent q-learning , 2016, ArXiv.

[10]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[11]  Gustavo E. A. P. A. Batista,et al.  A study of the behavior of several methods for balancing machine learning training data , 2004, SKDD.

[12]  In-So Kweon,et al.  CBAM: Convolutional Block Attention Module , 2018, ECCV.

[13]  Sandeep Raj,et al.  Sparse representation of ECG signals for automated recognition of cardiac arrhythmias , 2018, Expert Syst. Appl..

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

[15]  Rogério Schmidt Feris,et al.  A Unified Multi-scale Deep Convolutional Neural Network for Fast Object Detection , 2016, ECCV.

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

[17]  LuoXiaoqing,et al.  Heartbeat classification using disease-specific feature selection , 2014 .

[18]  B. V. K. Vijaya Kumar,et al.  Combining general multi-class and specific two-class classifiers for improved customized ECG heartbeat classification , 2012, Proceedings of the 21st International Conference on Pattern Recognition (ICPR2012).

[19]  B.V.K. Vijaya Kumar,et al.  Arrhythmia detection and classification using morphological and dynamic features of ECG signals , 2010, 2010 Annual International Conference of the IEEE Engineering in Medicine and Biology.

[20]  David Menotti,et al.  Inter-Patient ECG Heartbeat Classification with Temporal VCG Optimized by PSO , 2017, Scientific Reports.

[21]  Chun-Cheng Lin,et al.  Heartbeat Classification Using Normalized RR Intervals and Morphological Features , 2014 .

[22]  S. Yoo,et al.  Support Vector Machine Based Arrhythmia Classification Using Reduced Features , 2005 .

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

[24]  黄亚明.  PhysioBank , 2009 .

[25]  Zhengya Sun,et al.  Inter-Patient ECG Classification With Symbolic Representations and Multi-Perspective Convolutional Neural Networks , 2020, IEEE Journal of Biomedical and Health Informatics.

[26]  Masoumeh Haghpanahi,et al.  Cardiologist-level arrhythmia detection and classification in ambulatory electrocardiograms using a deep neural network , 2019, Nature Medicine.

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

[28]  Quoc V. Le,et al.  Sequence to Sequence Learning with Neural Networks , 2014, NIPS.

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

[30]  Majid Sarrafzadeh,et al.  ECG Heartbeat Classification: A Deep Transferable Representation , 2018, 2018 IEEE International Conference on Healthcare Informatics (ICHI).