RECOGNITION OF ATRIAL FIBRILLATION BASED ON CNN-LSTM AND LAPLACIAN SUPPORT VECTOR MACHINE

Abstract

Atrial fibrillation (AF) is a common arrhythmia associated with cardiac death and stroke. Therefore, the early detection of AF is of critical importance and the wearable long-term ECG monitoring system is one promising way. In order to assist the cardiologists in identifying potential AF in a tremendous amount of long-term ECG data, this study proposed an automatic detector combining deep learning and semi-supervised learning in view of the difficulty of obtaining a large number of labeled data in clinical practice. Three R-R interval features and two nonlinear features extracted from ECG samples, combined with 16-dimension deep learning features extracted by CNN-LSTM, are put into the semi-supervised machine learning model Laplacian Support Vector Machine (LapSVM) for AF detection. The proposed method showed very promising performance, with an accuracy of 99.63%, a sensitivity of 99.70%, a specificity of 99.57% and an F_score of 99.59% on the AFDB dataset. It still achieved an accuracy of 98% when the proportion of the training set was reduced, and achieved an accuracy of 96% on the SPHD collected clinically. The results show that the proposed method can classify AF and non-AF with a higher accuracy, and has excellent generalization performance in different categories of subjects, which is in line with clinical scenarios. The proposed method is also conducive to solving the clinical cases with little labeled data.