Application of artificial intelligence ensemble learning model in early prediction of atrial fibrillation

Cai Wu1, Maxwell Hwang2, Tian-Hsiang Huang3

  • 1Department of Hematology, The Fourth Affiliated Hospital of Zhejiang University School of Medicine, No. 1, Shangcheng Road, Yiwu, Zhejiang, China.

BMC Bioinformatics
|November 9, 2021
PubMed

Insights

Early detection of atrial fibrillation (AF) is crucial for stroke prevention. This study used P-wave and heart rate variability parameters with artificial intelligence to improve AF diagnosis, achieving 92% accuracy.

Area of Science:

  • Cardiology
  • Biomedical Engineering
  • Artificial Intelligence

Background:

  • Atrial fibrillation (AF) is often asymptomatic during onset, making early detection via electrocardiogram (ECG) challenging.
  • Delayed diagnosis and treatment of AF increase the risk of stroke.
  • ECGs of individuals with AF may appear normal between episodes.

Purpose of the Study:

  • To develop an improved artificial intelligence model for early atrial fibrillation detection.
  • To evaluate the efficacy of combining P-wave morphology and heart rate variability parameters for AF prediction.
  • To compare the performance of different artificial intelligence ensemble learning methods for AF diagnosis.

Main Methods:

  • Extracted 31 parameters including P-wave morphology and heart rate variability features from ECG data.
  • Employed artificial intelligence ensemble learning methods: Bagging, AdaBoost, and Stacking.
  • Utilized a hybrid Taguchi-genetic algorithm for accurate Gaussian function fitting in P-wave parameter calculation.

Main Results:

  • The Stacking ensemble learning method achieved the highest prediction accuracy (92%).
  • Key performance metrics included 88% sensitivity, 96% specificity, and a 0.911 area under the ROC curve.
  • The combined parameter approach significantly improved the model's predictive performance.

Conclusions:

  • Combining P-wave morphology and heart rate variability parameters enhances AF prediction model accuracy.
  • The Stacking ensemble learning method provides superior results for early AF detection.
  • This approach holds promise for improving early identification and management of atrial fibrillation.
Abstract

Related Concept Videos