CTRhythm: Accurate Atrial Fibrillation Detection from Single-Lead ECG by Convolutional Neural Network and Transformer Integration
Liang, Z.; Yang, C.; Yu, Z.; Fu, Y.; Ren, B.; Lin, M.; Li, Q.; Liu, X.; Chen, Y.; Xia, L. C.
Show abstract
Atrial Fibrillation (AF) is a common supraventricular arrhythmia that affects about 30 million people globally. Electrocardiogram (ECG) analysis is the primary diagnostic approach. The widespread adoption of wearable devices monitoring heart rhythm prompted the development of AF detection models for single-lead ECGs, benefitting real-time early diagnosis. Current state-of-the-art methods for AF detection are convolutional neural network (CNN) and convolutional recurrent neural network (CRNN) based models, which only focus on capturing local patterns despite heart rhythms exhibiting rich long-range dependencies. To address this limitation, we propose a novel method for single-lead ECG rhythm classification, termed CNN-Transformer Rhythm Classifier (CTRhythm), which integrates CNN with a Transformer encoder to capture local and global patterns effectively. CTRhythm achieved an overall F1 score of 0.831, outperforming the baseline deep learning models on the golden standard CINC2017 dataset. Moreover, pre-training with additional data improved the overall F1 score to 0.840. In two external validation datasets, CTRhythm showed its strong generalization capabilities. CTRhythm is freely available at https://github.com/labxscut/CTRhythm.
Matching journals
The top 7 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- A recurrent neural network and parallel hidden Markov model algorithm to segment and detect heart murmurs in phonocardiograms 96%
- Multiple Instance Learning Framework can Facilitate Explainability in Murmur Detection 96%
- QRS detection in single-lead, telehealth electrocardiogram signals: benchmarking open-source algorithms 95%
Similar papers in this journal
Similar papers in this journal
- A methodology of phenotyping ICU patients from EHR data: high-fidelity, personalized, and interpretable phenotypes estimation 91%
- Hypothesis-driven modeling of the human lung-ventilator system for Acute Respiratory Distress Syndrome research 90%
- Creating a computer assisted ICD coding system: performance metric choice and use of the ICD hierarchy 90%
Similar papers in this journal
- BRAVEHEART: Open-source software for automated electrocardiographic and vectorcardiographic analysis 95%
- Digitizing ECG image: new fully automated method and open-source software code 95%
- An algorithm to detect dicrotic notch in arterial blood pressure and photoplethysmography waveforms using the iterative envelope mean method 93%
Similar papers in this journal
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.