Automatic detection of arrhythmias from an ECG signal using an auto-encoder and SVM classifier

Phys Eng Sci Med. 2022 Jun;45(2):665-674. doi: 10.1007/s13246-022-01119-1. Epub 2022 Mar 18.

Abstract

Millions of people around the world are affected by arrhythmias, which are abnormal activities of the functioning of the heart. Most arrhythmias are harmful to the heart and can suddenly become life-threatening. The electrocardiogram (ECG) is an important non-invasive tool in cardiology for the diagnosis of arrhythmias. This work proposes a computer-aided diagnosis (CAD) system to automatically classify different types of arrhythmias from ECG signals. First, the auto-encoder convolutional network (ACN) model is used, which is based on a one-dimensional convolutional neural network (1D-CNN) that automatically learns the best features from the raw ECG signals. After that, the support vector machine (SVM) classifier is applied to the features learned by the ACN model to improve the detection of arrhythmic beats. This classifier detects four different types of arrhythmias, namely the left bundle branch block (LBBB), right bundle branch block (RBBB), paced beat (PB), and premature ventricular contractions (PVC), along with the normal sinus rhythms (NSR). Among these arrhythmias, PVC is particularly a dangerous type of heartbeat in ECG signals. The performance of the model is measured in terms of accuracy, sensitivity, and precision using a tenfold cross-validation strategy on the MIT-BIH arrhythmia database. The obtained overall accuracy of the SVM classifier was 98.84%. The result of this model is portrayed as a better performance than in other literary works. Thus, this approach may also help in further clinical studies of cardiac cases.

Keywords: Arrhythmia; Auto-encoder; ECG; SVM classifier.

MeSH terms

  • Bundle-Branch Block / diagnosis
  • Electrocardiography / methods
  • Humans
  • Support Vector Machine*
  • Ventricular Premature Complexes* / diagnosis