학술논문

Self-supervised learning for atrial fibrillation detection with ECG using CNNTransformer
Document Type
Conference
Source
2023 IEEE International Conference on Bioinformatics and Biomedicine (BIBM) Bioinformatics and Biomedicine (BIBM), 2023 IEEE International Conference on. :807-812 Dec, 2023
Subject
Bioengineering
Computing and Processing
Engineering Profession
Robotics and Control Systems
Signal Processing and Analysis
Soft sensors
Atrial fibrillation
Self-supervised learning
Electrocardiography
Predictive models
Feature extraction
Data models
ECG classification
Deep learning
Language
ISSN
2156-1133
Abstract
Cardiovascular diseases are a significant cause of mortality worldwide, and the accurate diagnosis of these conditions is essential for effective treatment and management. Electrocardiograms (ECGs) are a common diagnostic tool used by cardiologists, but the manual interpretation of ECGs can be relatively time-consuming and challenging, particularly in cases of atrial fibrillation (AF), which is associated with an increased risk of stroke, heart failure, and other complications. To address the need for reliable and automatic ECG classifiers, we propose a new method using self-supervised learning with a CNNTransformer architecture to improve the ECG classification performance. The proposed model is pre-trained on the China Physiological Signal Challenge 2018 dataset and part of the Physikalisch-Technische Bundesanstalt (PTB) XL dataset using a novel ’nextclip’ prediction task, which asks the model to predict the next small segment of ECG, followed by finetuning on the ECG classification task. Our experimental results demonstrate that our proposed method achieves state-of-the-art results for ECG classification, with an average F1-score of 0.84 and 0.96 for AF detection on the CPSC2018 dataset. The proposed CNNTransformer architecture has shown to be an effective and efficient solution for ECG classification, especially on AF.