A novel atrial fibrillation automatic detection algorithm based on ensemble learning and multi-feature discrimination

Med Biol Eng Comput. 2024 Feb 23. doi: 10.1007/s11517-024-03046-7. Online ahead of print.

Abstract

Atrial fibrillation (AF) is a prevalent cardiac arrhythmia disorder that necessitates long-time electrocardiogram (ECG) data for clinical diagnosis, leading to low detection efficiency. Automatic detection of AF signals within short-time ECG recordings is challenging. To address these issues, this paper proposes a novel algorithm called Ensemble Learning and Multi-Feature Discrimination (ELMD) for the identification and detection of AF signals. Firstly, a robust classifier, BSK-Model, is constructed using ensemble learning. Subsequently, the ECG R-waves are detected, and the ECG signals are segmented into consecutive RR intervals. Time domain, frequency domain, and nonlinear features are extracted from these intervals. Finally, these features are fed into the BSK-Model to discriminate AF. The proposed methodology is evaluated using the MIT-BIH AF database. The results demonstrate that when RR intervals are employed as classification units, the specificity and accuracy of AF detection in long-time ECG data exceed 99%, showcasing a significant improvement over traditional single-model classification. Additionally, the sensitivity and accuracy achieved by testing cardiac segments are both above 96%. With a minimum requirement of only four cardiac segments, AF events can be accurately identified, thereby enabling rapid discrimination of short-time single-lead ECG AF events. Consequently, this approach is suitable for real-time and accurate AF detection using low-computational-power ECG diagnostic analysis devices, such as wearable devices.

Keywords: Atrial fibrillation detection; Electrocardiogram; Ensemble learning; Multi-feature extraction.