Classification of Atrial Fibrillation and Congestive Heart Failure using Convolutional Neural Network with Electrocardiogram

Abstract

Delayed diagnosis of atrial fibrillation (AF) and congestive heart failure (CHF) can lead to death. Early diagnosis of these cardiac conditions is possible by manually analyzing electrocardiogram (ECG) signals. However, manual diagnosis is complex, owing to the various characteristics of ECG signals. Several studies have reported promising results using the automatic classification of ECG signals. The performance accuracy needs to be improved considering that an accurate classification system of AF and CHF has the potential to save a patient’s life. An optimal ECG signal classification system for AF and CHF has been proposed in this study using a one-dimensional convolutional neural network (1-D CNN) to improve the performance. A total of 150 datasets of ECG signals were modeled using the1-D CNN. The proposed 1-D CNN algorithm, provided precision values, recall, f1-score, accuracy of 100%, and successfully classified raw data of ECG signals into three conditions, which are normal sinus rhythm (NSR), AF, and CHF. The results showed that the proposed method outperformed the previous methods. This approach can be considered as an adjunct for medical personnel to diagnose AF, CHF, and NSR.