A Hybrid Deep CNN Model for Abnormal Arrhythmia Detection Based on Cardiac ECG Signal

Amin Ullah, Sadaqat Ur Rehman, Shanshan Tu, Raja Majid Mehmood, Fawad Fawad, Muhammad Ehatisham-ul-Haq

Sensors · 2021 · 161 citations · 58 references

DOIFull text

Open access

Concepts

TL;DR

ECG signals are essential for diagnosing and monitoring cardiovascular diseases. The study aims to create a robust algorithm that accurately classifies ECG signals even amid environmental noise. A hybrid approach uses a one‑dimensional CNN with two convolutional and two down‑sampling layers followed by a fully connected layer, and a two‑dimensional CNN that processes 1D ECG data converted to images, comprising three 2D convolutional and three down‑sampling layers plus a fully connected layer. The proposed 1D and 2D models achieved 97.38 % and 99.02 % accuracy on the MIT‑BIH arrhythmia database, outperforming existing state‑of‑the‑art classifiers.

Abstract

Electrocardiogram (ECG) signals play a vital role in diagnosing and monitoring patients suffering from various cardiovascular diseases (CVDs). This research aims to develop a robust algorithm that can accurately classify the electrocardiogram signal even in the presence of environmental noise. A one-dimensional convolutional neural network (CNN) with two convolutional layers, two down-sampling layers, and a fully connected layer is proposed in this work. The same 1D data was transformed into two-dimensional (2D) images to improve the model’s classification accuracy. Then, we applied the 2D CNN model consisting of input and output layers, three 2D-convolutional layers, three down-sampling layers, and a fully connected layer. The classification accuracy of 97.38% and 99.02% is achieved with the proposed 1D and 2D model when tested on the publicly available Massachusetts Institute of Technology-Beth Israel Hospital (MIT-BIH) arrhythmia database. Both proposed 1D and 2D CNN models outperformed the corresponding state-of-the-art classification algorithms for the same data, which validates the proposed models’ effectiveness.

References

58