Automatic heart disease class detection using convolutional neural network architecture-based various optimizers-networks

Marwa Fradi, Lazhar Khriji*, Mohsen Machhout, Abdulnasir Hossen

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

3 Citations (Scopus)

Abstract

Early heart disease class detection is of great interest to reduce the mortality rate. In this context, computational techniques have been proposed to solve this issue. Thus, here, a deep learning architecture is proposed to automatically classify the patient’s Electrocardiogram (ECG) signal into a specific class according to the ANSI–AAMI standards. The proposed methodology is a multi-stage technique. The first stage combines an R–R peak extraction with a low-pass filter applied on the ECG data for noise removal. The second stage shows the proposed convolutional neural network-based fully connected layers architecture, using different network optimizers. Different ECG databases, including challenging tasks, have been used for validation purpose. The whole system is implemented on both CPU and GPU for complexity analysis. For the predicted improved PTB data set, the classification accuracy results achieve 99.37%, 99.15% and 99.31% for training, validation and testing, respectively. Besides, for the MIT-BIH-database, the training, validation and testing accuracies are 99.5%, 99.06% and 99.34%, respectively. A top F1-score of 0.99 is obtained. Experimental results show a high achievement compared to the state-of-the-art models where the implementation of GPU confirms the low computational complexity of the system.

Original languageEnglish
Pages (from-to)3-15
Number of pages13
JournalIET Smart Cities
Volume3
Issue number1
DOIs
Publication statusPublished - Mar 2021
Externally publishedYes

ASJC Scopus subject areas

  • Urban Studies
  • Electrical and Electronic Engineering
  • Control and Systems Engineering
  • Computer Science Applications
  • Artificial Intelligence
  • Computer Networks and Communications
  • Software

Fingerprint

Dive into the research topics of 'Automatic heart disease class detection using convolutional neural network architecture-based various optimizers-networks'. Together they form a unique fingerprint.

Cite this