Deep Learning Based Atrial Fibrillation Detection Using Combination of Dimensionality Reduction Techniques and RR Interval Features

Abstract

Purpose: Accurate detection of Atrial Fibrillation (AF) has great significance in the field of medical science which can reduce the rate of mortality and morbidity. The present study focuses on Electrocardiography (ECG) signal classification using dimensionality reduction techniques combined with R wave to R wave interval (RR interval) features.

Materials and Methods: In the first approach, Principal Component Analysis (PCA), Linear Discriminant Analysis (LDA), Independent Component Analysis (ICA), and Probabilistic Principal Component Analysis (PPCA) are performed independently on denoised ECG signal using Discrete Wavelet Transform (DWT) for the classification of ECG signal. In the second approach, the dimensionality reduction techniques combined with RR interval features are used for the classification of ECG signal.

Results: Machine Learning (ML) algorithms such as Decision Tree (DT), Support Vector Machine (SVM), and Deep Learning (DL) algorithms such as Long Short Term Memory (LSTM) and Bi-Directional LSTM (BiLSTM) are used for classification purposes.

Conclusion: The proposed methodology provided an overall accuracy of 93.65% with PCA and LSTM classifier and an overall accuracy of 99.45% with PCA combined with RR interval features and LSTM classifier. The developed technology has potential applications in many practical solutions.