Home > Published Issues > 2023 > Volume 11, No. 1, March 2023 >
JOIG 2023 Vol.11(1): 9-14
doi: 10.18178/joig.11.1.9-14

ImECGnet: Cardiovascular Disease Classification from Image-Based ECG Data Using a Multibranch Convolutional Neural Network

Amir Ghahremani and Christoph Lofi*
Delft University of Technology, Delft, the Netherlands; Email: a.ghahremani@tudelft.nl (A.G.)
*Correspondence: c.lofi@tudelft.nl (C.L.)

Manuscript received July 1, 2022; revised August 1, 2022; accepted October 9, 2022.

Abstract—Reliable Cardiovascular Disease (CVD) classification performed by a smart system can assist medical doctors in recognizing heart illnesses in patients more efficiently and effectively. Electrocardiogram (ECG) signals are an important diagnostic tool as they are already available early in the patients’ health diagnosis process and contain valuable indicators for various CVDs. Most ECG processing methods represent ECG data as a time series, often as a matrix with each row containing the measurements of a sensor lead; and/or the transforms of such time series like wavelet power spectrums. While methods processing such time-series data have been shown to work well in benchmarks, they are still highly dependent on factors like input noise and sequence length, and cannot always correlate lead data from different sensors well. In this paper, we propose to represent ECG signals incorporating all lead data plotted as a single image, an approach not yet explored by literature. We will show that such an image representation combined with our newly proposed convolutional neural network specifically designed for CVD classification can overcome the aforementioned shortcomings. The proposed (Convolutional Neural Network) CNN is designed to extract features representing both the proportional relationships of different leads to each other and the characteristics of each lead separately. Empirical validation on the publicly available PTB, MIT-BIH, and St.-Petersburg benchmark databases shows that the proposed method outperforms time seriesbased state-of-the-art approaches, yielding classification accuracy of 97.91%, 99.62%, and 98.70%, respectively.

Keywords—Convolutional Neural Network (CNN), classification, Electrocardiogram (ECG)

Cite: Amir Ghahremani and Christoph Lofi, "ImECGnet: Cardiovascular Disease Classification from Image-Based ECG Data Using a Multibranch Convolutional Neural Network," Journal of Image and Graphics, Vol. 11, No. 1, pp. 9-14, March 2023.

Copyright © 2023 by the authors. This is an open access article distributed under the Creative Commons Attribution License (CC BY-NC-ND 4.0), which permits use, distribution and reproduction in any medium, provided that the article is properly cited, the use is non-commercial and no modifications or adaptations are made.