کنفرانس مهندسی برق ایران

صفحه اصلی / سی و یکمین کنفرانس بین المللی مهندسی برق

Atrial Fibrillation (AF) Detection Using Deep Learning with GAN-based Data Augmentation

نویسندگان :

Amirhossein Akhoondkazemi¹ Arash Vashagh² Sayed Jalal Zahabi³ Davood Shafie⁴

1- دانشگاه صنعتی امیرکبیر 2- دانشگاه صنعتی اصفهان 3- دانشگاه صنعتی اصفهان 4- دانشگاه علوم پزشکی اصفهان

کلمات کلیدی :

ECG،Atrial Fibrillation،Deep Learning،Poincare recurrence plot،Generative Adversarial Network

چکیده :

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