An improved method to detect arrhythmia using ensemble learning-based model in multi lead electrocardiogram (ECG)

PLoS One. 2024 Apr 9;19(4):e0297551. doi: 10.1371/journal.pone.0297551. eCollection 2024.

Abstract

Arrhythmia is a life-threatening cardiac condition characterized by irregular heart rhythm. Early and accurate detection is crucial for effective treatment. However, single-lead electrocardiogram (ECG) methods have limited sensitivity and specificity. This study propose an improved ensemble learning approach for arrhythmia detection using multi-lead ECG data. Proposed method, based on a boosting algorithm, namely Fine Tuned Boosting (FTBO) model detects multiple arrhythmia classes. For the feature extraction, introduce a new technique that utilizes a sliding window with a window size of 5 R-peaks. This study compared it with other models, including bagging and stacking, and assessed the impact of parameter tuning. Rigorous experiments on the MIT-BIH arrhythmia database focused on Premature Ventricular Contraction (PVC), Atrial Premature Contraction (PAC), and Atrial Fibrillation (AF) have been performed. The results showed that the proposed method achieved high sensitivity, specificity, and accuracy for all three classes of arrhythmia. It accurately detected Atrial Fibrillation (AF) with 100% sensitivity and specificity. For Premature Ventricular Contraction (PVC) detection, it achieved 99% sensitivity and specificity in both leads. Similarly, for Atrial Premature Contraction (PAC) detection, proposed method achieved almost 96% sensitivity and specificity in both leads. The proposed method shows great potential for early arrhythmia detection using multi-lead ECG data.

MeSH terms

  • Algorithms
  • Atrial Fibrillation* / diagnosis
  • Atrial Premature Complexes* / diagnosis
  • Electrocardiography / methods
  • Humans
  • Machine Learning
  • Ventricular Premature Complexes* / diagnosis

Grants and funding

Satria Mandala received a research grant from the Ministry of Education, Culture, Research, and Technology of Indonesia under the main contract number 074/E5/PG.02.00.PL/2023; Subcontract both "No. 021/SP2H/RT-JAMAK/LL4/2023 dated May 2, 2023" and "No. 109/PNLT2/PPM/2003 dated May 3, 2023". The URL of the funder is https://dikti.kemdikbud.go.id/category/pengumuman/drtpm/. The funders had no role in study design, data collection and analysis, decision to publish, or preparation of the manuscript.