SEResUTer: a deep learning approach for accurate ECG signal delineation and atrial fibrillation detection

Abstract

Objective.Accurate detection of electrocardiogram (ECG) waveforms is crucial for computer-aided diagnosis of cardiac abnormalities. This study introduces SEResUTer, an enhanced deep learning model designed for ECG delineation and atrial fibrillation (AF) detection.Approach. Built upon a U-Net architecture, SEResUTer incorporates ResNet modules and Transformer encoders to replace convolution blocks, resulting in improved optimization and encoding capabilities. A novel masking strategy is proposed to handle incomplete expert annotations. The model is trained on the QT database (QTDB) and evaluated on the Lobachevsky University Electrocardiography Database (LUDB) to assess its generalization performance. Additionally, the model's scope is extended to AF detection using the the China Physiological Signal Challenge 2021 (CPSC2021) and the China Physiological Signal Challenge 2018 (CPSC2018) datasets.Main results.The proposed model surpasses existing traditional and deep learning approaches in ECG waveform delineation on the QTDB. It achieves remarkable average F1 scores of 99.14%, 98.48%, and 98.46% for P wave, QRS wave, and T wave delineation, respectively. Moreover, the model demonstrates exceptional generalization ability on the LUDB, achieving average SE, positive prediction rate, and F1 scores of 99.05%, 94.59%, and 94.62%, respectively. By analyzing RR interval differences and the existence of P waves, our method achieves AF identification with 99.20% accuracy on the CPSC2021 test set and demonstrates strong generalization on CPSC2018 dataset.Significance.The proposed approach enables highly accurate ECG waveform delineation and AF detection, facilitating automated analysis of large-scale ECG recordings and improving the diagnosis of cardiac abnormalities.

Keywords: ECG delineation; atrial fibrillation detection; data augmentation; transformer.