Heartbeat classification by using a convolutional neural network trained with Walsh functions

From recent studies, it is observed that convolutional neural networks are proved to be extremely successful in classification problems. Accurate and fast classification of electrocardiogram (ECG) beats is a crucial step in the implementation of real-time arrhythmia diagnosis systems. In this study, convolutional neural networks are employed to classify eleven different ECG beat types in the MIT-BIH arrhythmia database. We aimed to implement a computer-aided mobile diagnosis system equipped with artificial intelligence that detects the classes of heartbeats by visual inspection of the ECG records in the manner as the cardiologists do. Since doctors make their decisions basing heavily on the 2D visual appearances of the ECG signals without doing numerical calculations on 1D time samples, 2D images of 1D ECG records were given to the classifier as the input data. It would not be surprising that the structure of the network classifying 2D ECG data has to be larger than the one used to classify 1D ECG signals. The small size of a neural network is an important property for real-time use of the system. In this study, smaller network structures that provide high performances using the Walsh functions (WF), and drawbacks of converting 1D signals to 2D images have been investigated. The network structures using the WF during the training stage have been applied to different databases and successful results have been obtained. Classification results and sizes of the network structures are compared for the ECG beats in one dimension and in the form of 2D visuals. The training of ECG signals in the form of 2D visuals takes much longer time than that of the 1D signals. However, it is observed that training and testing times of both networks were quite fast. Moreover, average success rates of 99% were achieved for all beat types by using small-size networks.

[1]  Kumar Sricharan,et al.  Classifying heart sound recordings using deep convolutional neural networks and mel-frequency cepstral coefficients , 2016, 2016 Computing in Cardiology Conference (CinC).

[2]  Lei Wang,et al.  Deep learning-based classification of massive electrocardiography data , 2016, 2016 IEEE Advanced Information Management, Communicates, Electronic and Automation Control Conference (IMCEC).

[3]  B Pyakillya,et al.  Deep Learning for ECG Classification , 2017 .

[4]  Song Han,et al.  Learning both Weights and Connections for Efficient Neural Network , 2015, NIPS.

[5]  Reza Ebrahimpour,et al.  Classification of ECG arrhythmia by a modular neural network based on Mixture of Experts and Negatively Correlated Learning , 2013, Biomed. Signal Process. Control..

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

[7]  Alex Krizhevsky,et al.  Learning Multiple Layers of Features from Tiny Images , 2009 .

[8]  Sangdeok Kim,et al.  Epileptic seizure detection for multi-channel EEG with deep convolutional neural network , 2018, 2018 International Conference on Electronics, Information, and Communication (ICEIC).

[9]  F. Melgani,et al.  Detecting premature ventricular contractions in ECG signals with Gaussian processes , 2008, 2008 Computers in Cardiology.

[10]  Paul Honeine,et al.  Multi-class SVM classification combined with kernel PCA feature extraction of ECG signals , 2012, 2012 19th International Conference on Telecommunications (ICT).

[11]  Yang Song,et al.  Class-Balanced Loss Based on Effective Number of Samples , 2019, 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[12]  Pornchai Phukpattaranont,et al.  QRS detection algorithm based on the quadratic filter , 2015, Expert Syst. Appl..

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

[14]  Yixin Chen,et al.  Compressing Neural Networks with the Hashing Trick , 2015, ICML.

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

[16]  Sungyoung Lee,et al.  Iss2Image: A Novel Signal-Encoding Technique for CNN-Based Human Activity Recognition , 2018, Sensors.

[17]  Arnab Chakraborty,et al.  Hybrid shallow and deep learned feature mixture model for arrhythmia classification , 2018, 2018 Electric Electronics, Computer Science, Biomedical Engineerings' Meeting (EBBT).

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

[19]  Yoshua Bengio,et al.  Gradient-based learning applied to document recognition , 1998, Proc. IEEE.

[20]  Shan Gao,et al.  Respiratory Sounds Feature Learning with Deep Convolutional Neural Networks , 2017, 2017 IEEE 15th Intl Conf on Dependable, Autonomic and Secure Computing, 15th Intl Conf on Pervasive Intelligence and Computing, 3rd Intl Conf on Big Data Intelligence and Computing and Cyber Science and Technology Congress(DASC/PiCom/DataCom/CyberSciTech).

[21]  Amit Acharyya,et al.  Development of an Automated Updated Selvester QRS Scoring System Using SWT-Based QRS Fractionation Detection and Classification , 2014, IEEE Journal of Biomedical and Health Informatics.

[22]  Yann LeCun,et al.  Regularization of Neural Networks using DropConnect , 2013, ICML.

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

[24]  Daeyoung Kim,et al.  Premature Ventricular Contraction Beat Detection with Deep Neural Networks , 2016, 2016 15th IEEE International Conference on Machine Learning and Applications (ICMLA).

[25]  Yakup Kutlu,et al.  Feature extraction for ECG heartbeats using higher order statistics of WPD coefficients , 2012, Comput. Methods Programs Biomed..

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

[27]  Hui Yang,et al.  Spatiotemporal Differentiation of Myocardial Infarctions , 2013, IEEE Transactions on Automation Science and Engineering.

[28]  Khashayar Khorasani,et al.  Deep Convolutional Neural Networks and Learning ECG Features for Screening Paroxysmal Atrial Fibrillation Patients , 2018, IEEE Transactions on Systems, Man, and Cybernetics: Systems.

[29]  Min Zhou,et al.  ECG Classification Using Wavelet Packet Entropy and Random Forests , 2016, Entropy.

[30]  Yongqiang Ye,et al.  Parallel-type fractional zero-phase filtering for ECG signal denoising , 2015, Biomed. Signal Process. Control..

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

[32]  David Menotti,et al.  ECG arrhythmia classification based on optimum-path forest , 2013, Expert Syst. Appl..

[33]  U. Rajendra Acharya,et al.  Computer aided diagnosis of atrial arrhythmia using dimensionality reduction methods on transform domain representation , 2014, Biomed. Signal Process. Control..

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

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

[36]  Prabin Kumar Bora,et al.  Electrocardiogram signal denoising using non-local wavelet transform domain filtering , 2015, IET Signal Process..

[37]  José Luis Rojo-Álvarez,et al.  Detection of Life-Threatening Arrhythmias Using Feature Selection and Support Vector Machines , 2014, IEEE Transactions on Biomedical Engineering.

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

[39]  Ying Liu,et al.  A Comparison of 1-D and 2-D Deep Convolutional Neural Networks in ECG Classification , 2018, Conference proceedings : ... Annual International Conference of the IEEE Engineering in Medicine and Biology Society. IEEE Engineering in Medicine and Biology Society. Annual Conference.

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

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

[42]  Song Han,et al.  Deep Compression: Compressing Deep Neural Network with Pruning, Trained Quantization and Huffman Coding , 2015, ICLR.

[43]  Pei-Chann Chang,et al.  Myocardial infarction classification with multi-lead ECG using hidden Markov models and Gaussian mixture models , 2012, Appl. Soft Comput..

[44]  Shuo Li,et al.  An Automatic Cardiac Arrhythmia Classification System With Wearable Electrocardiogram , 2018, IEEE Access.

[45]  Iwan Syarif,et al.  Deep learning algorithm for arrhythmia detection , 2017, 2017 International Electronics Symposium on Knowledge Creation and Intelligent Computing (IES-KCIC).

[46]  Sugata Munshi,et al.  Correlation technique and least square support vector machine combine for frequency domain based ECG beat classification. , 2010, Medical engineering & physics.

[47]  Ming Yang,et al.  Compressing Deep Convolutional Networks using Vector Quantization , 2014, ArXiv.

[48]  Daeyoung Kim,et al.  ECG arrhythmia classification using a 2-D convolutional neural network , 2018, ArXiv.

[49]  Benjamin Graham,et al.  Fractional Max-Pooling , 2014, ArXiv.

[50]  Rich Caruana,et al.  Do Deep Nets Really Need to be Deep? , 2013, NIPS.

[51]  Christos Panagiotou,et al.  On the Detection of Myocadial Scar Based on ECG/VCG Analysis , 2013, IEEE Transactions on Biomedical Engineering.

[52]  Naif Alajlan,et al.  Detection of premature ventricular contraction arrhythmias in electrocardiogram signals with kernel methods , 2012, Signal, Image and Video Processing.

[53]  Nadeem Ahmad Khan,et al.  DeepMI: Deep Learning for Multiclass Motor Imagery Classification , 2018, 2018 40th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC).

[54]  Herbert F. Voigt,et al.  IEEE Engineering in Medicine and Biology Society , 2019, IEEE Transactions on Biomedical Engineering.

[55]  Mohammadreza Balouchestani,et al.  Fast clustering algorithm for large ECG data sets based on CS theory in combination with PCA and K-NN methods , 2014, 2014 36th Annual International Conference of the IEEE Engineering in Medicine and Biology Society.

[56]  Xuhui Chen,et al.  A deep multi-task learning approach for ECG data analysis , 2018, 2018 IEEE EMBS International Conference on Biomedical & Health Informatics (BHI).

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