Detection of atrial fibrillation from ECG recordings using decision tree ensemble with multi-level features

Physiol Meas. 2018 Sep 27;39(9):094008. doi: 10.1088/1361-6579/aadf48.

Abstract

Objective: Detecting atrial fibrillation (AF) from electrocardiogram (ECG) recordings remains a challenging task. In this paper, a new AF detection method was proposed to classify the ECG recordings into one of four classes: Normal rhythm, AF, Other rhythm, and Noisy recordings.

Approach: The proposed method comprised preprocessing, feature extraction, and classification. In preprocessing, R-peaks were detected, and RR intervals and delta RR intervals were extracted. In feature extraction, 30 multi-level features were extracted, including AF features (n = 4), morphology features (n = 20), RR interval features (n = 2), and features of similarity index between beats (n = 4). In classification, these features were used to train an AdaBoosted decision tree ensemble for classifying ECG recordings into the four classes. The decision tree ensemble was trained with 100-fold cross-validation on the training dataset (n = 8528) provided by the 2017 PhysioNet/Computing in Cardiology (CinC) Challenge.

Main results: The trained classifier was submitted to the Challenge for testing on the unavailable test dataset (n = 3658); the official F 1 scores for 'Normal', 'AF', 'Other' were 0.91, 0.82, and 0.73, respectively, and the overall F 1 score was 0.82 (ranking equal 5th with eight other algorithms in the 2017 PhysioNet/CinC Challenge).

Significance: The proposed algorithm may be used as a new method for AF detection.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Algorithms*
  • Artifacts
  • Atrial Fibrillation / diagnosis*
  • Decision Trees
  • Diagnosis, Computer-Assisted / methods*
  • Electrocardiography / methods*
  • Humans
  • Multilevel Analysis
  • Pattern Recognition, Automated / methods