Atrial Fibrillation (Af) Detection Using Deep Learning With Gan-Based Data Augmentation Publisher



Akhoondkazemi A¹ ; Vashagh A¹ ; Zahabi SJ¹ ; Shafie D² Show Affiliations
Source: 2023 31st International Conference on Electrical Engineering, ICEE 2023 Published:2023

Abstract

Atrial Fibrillation (AF) is the most common cardiac arrhythmia that may lead to stroke and heart failure and because of this, a lot of research has gone into detecting AF from the electrocardiogram (ECG) signal. In this paper, we propose an AF detection pipeline, which first transforms the ECG data into an informative 2-D image by the aid Poincare recurrence plot, and then addresses the imbalance of the data by augmenting AF samples using a form of generative adversarial network (GAN). The augmented dataset is then used within the training set to train a 5-layer convolutional neural network (CNN) as a classifier. The performance of the proposed classifier is finally evaluated based on a 4-fold cross-validation scheme. The performance metrics suggest that the proposed method provides acceptable sensitivity and specificity. © 2023 IEEE.