ECG beat classification using neural classifier based on deep autoencoder and decomposition techniques

This paper proposes an ECG beat classification system based on deep autoencoder as feature extractor and a system of multiple neural networks as classifier. The objectives are as follows: First is simplifying the feature extraction step by applying the deep autoencoder, which permits defining high level features without neither pre-processing stage nor expert intervention. Second is enhancing the classification performance by decomposing the original multi-class problem into simpler binary subproblems and solving them using independent classifiers. Third is overcoming the problem of imbalanced data, by applying an oversampling method after the decomposition of the original problem. This allows adding synthetic samples according the number of training instances in each subproblem. To evaluate the proposed system, we conduct experiments on MIT-BIH arrhythmia dataset and we consider the recommendations of the Association for the Advancement of Medical Instrumentation, which defines five classes of interest. Furthermore, we perform two types of tests, i.e. intra- and inter-patient, and compare the obtained results with some of the state-of-the-art methods. We show that solving each subproblem independently can enhance the accuracy, sensitivity and specificity.

[1]  Wendong Xiao,et al.  Electrocardiogram Classification Based on Faster Regions with Convolutional Neural Network , 2019, Sensors.

[2]  Chih-Jen Lin,et al.  Probability Estimates for Multi-class Classification by Pairwise Coupling , 2003, J. Mach. Learn. Res..

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

[4]  B. Fei,et al.  Binary tree of SVM: a new fast multiclass training and classification algorithm , 2006, IEEE Transactions on Neural Networks.

[5]  Manuel G. Penedo,et al.  Heartbeat classification fusing temporal and morphological information of ECGs via ensemble of classifiers , 2019, Biomed. Signal Process. Control..

[6]  G. Pazour,et al.  Ror2 signaling regulates Golgi structure and transport through IFT20 for tumor invasiveness , 2017, Scientific Reports.

[7]  Yang Bai,et al.  A novel electrocardiogram arrhythmia classification method based on stacked sparse auto-encoders and softmax regression , 2017, International Journal of Machine Learning and Cybernetics.

[8]  Eric C. C. Tsang,et al.  Nesting One-Against-One Algorithm Based on SVMs for Pattern Classification , 2008, IEEE Transactions on Neural Networks.

[9]  Mikel Galar,et al.  Addressing the Overlapping Data Problem in Classification Using the One-vs-One Decomposition Strategy , 2019, IEEE Access.

[10]  Jamil Salem Barbar,et al.  Multilayer Perceptron and Stacked Autoencoder for Internet Traffic Prediction , 2014, NPC.

[11]  Eyke Hüllermeier,et al.  Combining predictions in pairwise classification: An optimal adaptive voting strategy and its relation to weighted voting , 2010, Pattern Recognit..

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

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

[14]  Francisco Herrera,et al.  Solving multi-class problems with linguistic fuzzy rule based classification systems based on pairwise learning and preference relations , 2010, Fuzzy Sets Syst..

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

[16]  Thomas Arnold,et al.  Self-Assembly and Anti-Amyloid Cytotoxicity Activity of Amyloid beta Peptide Derivatives , 2017, Scientific Reports.

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

[18]  Yoshua. Bengio,et al.  Learning Deep Architectures for AI , 2007, Found. Trends Mach. Learn..

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

[20]  Eyke Hüllermeier,et al.  FR3: A Fuzzy Rule Learner for Inducing Reliable Classifiers , 2009, IEEE Transactions on Fuzzy Systems.

[21]  Yixiang Huang,et al.  A high-precision arrhythmia classification method based on dual fully connected neural network , 2020, Biomed. Signal Process. Control..

[22]  Herman Akdag,et al.  The Labeled Systems of Multiple Neural Networks , 2008, Int. J. Neural Syst..

[23]  Francisco Herrera,et al.  An overview of ensemble methods for binary classifiers in multi-class problems: Experimental study on one-vs-one and one-vs-all schemes , 2011, Pattern Recognit..

[24]  Masatoshi Okutomi,et al.  Improving transparency of deep neural inference process , 2019, Progress in Artificial Intelligence.

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

[26]  Mihir Narayan Mohanty,et al.  Detection of Arrhythmia using Neural Network , 2017, ICITKM.

[27]  Bo Tan,et al.  Electrocardiogram soft computing using hybrid deep learning CNN-ELM , 2020, Appl. Soft Comput..

[28]  Di Wang,et al.  Automatic recognition of arrhythmia based on principal component analysis network and linear support vector machine , 2018, Comput. Biol. Medicine.

[29]  Jiye Liang,et al.  A multi-view OVA model based on decision tree for multi-classification tasks , 2017, Knowl. Based Syst..

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

[31]  Herman Akdag,et al.  One-against-all and one-against-one based neuro-fuzzy classifiers , 2014, J. Intell. Fuzzy Syst..

[32]  Mihir Narayan Mohanty,et al.  Classification of Arrhythmia Using Artificial Neural Network with Grey Wolf Optimization , 2019 .

[33]  Francisco Herrera,et al.  Exploring the effectiveness of dynamic ensemble selection in the one-versus-one scheme , 2017, Knowl. Based Syst..

[34]  Gyanendra K. Verma,et al.  Convolutional neural network: a review of models, methodologies and applications to object detection , 2019, Progress in Artificial Intelligence.

[35]  Mahdi Jafari Siavoshani,et al.  Deep packet: a novel approach for encrypted traffic classification using deep learning , 2017, Soft Computing.

[36]  Ligang Zhou,et al.  One versus one multi-class classification fusion using optimizing decision directed acyclic graph for predicting listing status of companies , 2017, Inf. Fusion.

[37]  Annisa Darmawahyuni,et al.  Deep Learning-Based Stacked Denoising and Autoencoder for ECG Heartbeat Classification , 2020, Electronics.

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

[39]  Özal Yildirim,et al.  A novel wavelet sequence based on deep bidirectional LSTM network model for ECG signal classification , 2018, Comput. Biol. Medicine.

[40]  Jian Li,et al.  Heartbeat classification using projected and dynamic features of ECG signal , 2017, Biomed. Signal Process. Control..

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

[42]  U. Rajendra Acharya,et al.  A new approach for arrhythmia classification using deep coded features and LSTM networks , 2019, Comput. Methods Programs Biomed..

[43]  Siouda Roguia,et al.  An Optimized RBF-Neural Network for Breast Cancer Classification , 2018 .

[44]  Fatemeh Afghah,et al.  Inter- and Intra- Patient ECG Heartbeat Classification for Arrhythmia Detection: A Sequence to Sequence Deep Learning Approach , 2018, ICASSP 2019 - 2019 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP).

[45]  U. Rajendra Acharya,et al.  An efficient compression of ECG signals using deep convolutional autoencoders , 2018, Cognitive Systems Research.

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

[47]  Nello Cristianini,et al.  Large Margin DAGs for Multiclass Classification , 1999, NIPS.

[48]  Nitesh V. Chawla,et al.  SMOTE: Synthetic Minority Over-sampling Technique , 2002, J. Artif. Intell. Res..

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

[50]  Chengjin Qin,et al.  Automated heartbeat classification based on deep neural network with multiple input layers , 2020, Knowl. Based Syst..

[51]  Shalin Savalia,et al.  Cardiac Arrhythmia Classification by Multi-Layer Perceptron and Convolution Neural Networks , 2018, Bioengineering.