Classification of electrocardiogram signals using deep learning based on genetic algorithm feature extraction.

Journal: Biomedical physics & engineering express
PMID:

Abstract

Arrhythmias using electrocardiogram (ECG) signal is important in medical and computer research due to the timely diagnosis of dangerous cardiac conditions. The current study used the ECG to classify cardiac signals into normal heartbeats, congestive heart failure, ventricular arrhythmias, atrial fibrillation arrhythmias, atrial flutter, malignant ventricular arrhythmias, and premature atrial fibrillation. A deep learning algorithm was used to identify and diagnose cardiac arrhythmias. We proposed a new ECG signal classification method to increase signal classification sensitivity. We smoothed the ECG signal with noise removal filters. A discrete wavelet transform based on an arrhythmic database was applied to extract ECG features. Feature vectors were obtained based on wavelet decomposition energy properties and calculated values of PQRS morphological features. We used the genetic algorithm to reduce the feature vector and determine the input layer weights of the artificial neural network (ANN) and adaptive neuro-fuzzy inference system (ANFIS). Proposed methods for classifying ECG signals were in different classes of rhythm to diagnose heart rhythm diseases. Training data was with 80% of the data set and test data was with 20% for the whole data set. The learning accuracy for the results of training and test data in the ANN classifier was calculated as 99.9% and 88.92% and in ANFIS as 99.8% and 88.83% respectively. Based on these results, good accuracy was observed.

Authors

  • Hossein Khezripour
    Department of Computer Engineering, Science and Research Branch, Islamic Azad University, Tehran, Iran.
  • Saadat Pour Mozaffari
    Computer Engineering and Information Technology Department, Amirkabir University of Technology, Tehran, Iran.
  • Midia Reshadi
    Department of Computer Engineering, Science and Research Branch, Islamic Azad University, Tehran, Iran.
  • Houman Zarrabi
    ICT Research Center, Tehran, Iran.