ECG classification using 1-D convolutional deep residual neural network

PLoS One. 2023 Apr 25;18(4):e0284791. doi: 10.1371/journal.pone.0284791. eCollection 2023.

Abstract

An electrocardiograph (ECG) is widely used in diagnosis and prediction of cardiovascular diseases (CVDs). The traditional ECG classification methods have complex signal processing phases that leads to expensive designs. This paper provides a deep learning (DL) based system that employs the convolutional neural networks (CNNs) for classification of ECG signals present in PhysioNet MIT-BIH Arrhythmia database. The proposed system implements 1-D convolutional deep residual neural network (ResNet) model that performs feature extraction by directly using the input heartbeats. We have used synthetic minority oversampling technique (SMOTE) that process class-imbalance problem in the training dataset and effectively classifies the five heartbeat types in the test dataset. The classifier's performance is evaluated with ten-fold cross validation (CV) using accuracy, precision, sensitivity, F1-score, and kappa. We have obtained an average accuracy of 98.63%, precision of 92.86%, sensitivity of 92.41%, and specificity of 99.06%. The average F1-score and Kappa obtained were 92.63% and 95.5% respectively. The study shows that proposed ResNet performs well with deep layers compared to other 1-D CNNs.

MeSH terms

  • Algorithms*
  • Arrhythmias, Cardiac / diagnosis
  • Cardiovascular Diseases*
  • Electrocardiography
  • Humans
  • Neural Networks, Computer
  • Signal Processing, Computer-Assisted

Grants and funding

The author(s) received no specific funding for this work.