Automated detection of atrial fibrillation using long short-term memory network with RR interval signals

Comput Biol Med. 2018 Nov 1:102:327-335. doi: 10.1016/j.compbiomed.2018.07.001. Epub 2018 Jul 17.

Abstract

Atrial Fibrillation (AF), either permanent or intermittent (paroxysnal AF), increases the risk of cardioembolic stroke. Accurate diagnosis of AF is obligatory for initiation of effective treatment to prevent stroke. Long term cardiac monitoring improves the likelihood of diagnosing paroxysmal AF. We used a deep learning system to detect AF beats in Heart Rate (HR) signals. The data was partitioned with a sliding window of 100 beats. The resulting signal blocks were directly fed into a deep Recurrent Neural Network (RNN) with Long Short-Term Memory (LSTM). The system was validated and tested with data from the MIT-BIH Atrial Fibrillation Database. It achieved 98.51% accuracy with 10-fold cross-validation (20 subjects) and 99.77% with blindfold validation (3 subjects). The proposed system structure is straight forward, because there is no need for information reduction through feature extraction. All the complexity resides in the deep learning system, which gets the entire information from a signal block. This setup leads to the robust performance for unknown data, as measured with the blind fold validation. The proposed Computer-Aided Diagnosis (CAD) system can be used for long-term monitoring of the human heart. To the best of our knowledge, the proposed system is the first to incorporate deep learning for AF beat detection.

Keywords: Atrial fibrillation; Deep learning; Heart rate; Recurrent neural network.

MeSH terms

  • Algorithms
  • Atrial Fibrillation / diagnosis*
  • Data Collection
  • Databases, Factual
  • Deep Learning
  • Diagnosis, Computer-Assisted / methods*
  • Electrocardiography*
  • Electronic Data Processing*
  • Heart Rate
  • Humans
  • Monitoring, Physiologic
  • Neural Networks, Computer
  • Reproducibility of Results
  • Risk
  • Sensitivity and Specificity
  • Signal Processing, Computer-Assisted*
  • Software
  • Support Vector Machine