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

Xinyue Li, Wenjie Cai, Bolin Xu, Yupeng Jiang, Mengdi Qi, Mingjie Wang
Author Information
  1. Xinyue Li: School of Health Science and Engineering, University of Shanghai for Science and Technology, Shanghai, 200093, People's Republic of China.
  2. Wenjie Cai: School of Health Science and Engineering, University of Shanghai for Science and Technology, Shanghai, 200093, People's Republic of China. ORCID
  3. Bolin Xu: School of Health Science and Engineering, University of Shanghai for Science and Technology, Shanghai, 200093, People's Republic of China.
  4. Yupeng Jiang: School of Health Science and Engineering, University of Shanghai for Science and Technology, Shanghai, 200093, People's Republic of China.
  5. Mengdi Qi: School of Health Science and Engineering, University of Shanghai for Science and Technology, Shanghai, 200093, People's Republic of China.
  6. Mingjie Wang: Shanghai Key Laboratory of Bioactive Small Molecules, School of Basic Medical Science, Fudan University, Shanghai, 200032, People's Republic of China.

Abstract

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.. 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.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.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

MeSH Term

Humans
Atrial Fibrillation
Deep Learning
Algorithms
Signal Processing, Computer-Assisted
Electrocardiography

Word Cloud

Created with Highcharts 10.0.0ECGdetectiondelineationmodelAFdeeplearningatrialfibrillationproposedgeneralization99wavediagnosiscardiacabnormalitiesSEResUTerQTDBLUDBChinaPhysiologicalSignalChallengeCPSC2021CPSC2018ThewaveformachievesaverageF1scores98Prespectivelydemonstrates94approachaccurateAccurateelectrocardiogramwaveformscrucialcomputer-aidedstudyintroducesenhanceddesignedBuiltuponU-NetarchitectureincorporatesResNetmodulesTransformerencodersreplaceconvolutionblocksresultingimprovedoptimizationencodingcapabilitiesnovelmaskingstrategyhandleincompleteexpertannotationstrainedQTdatabaseevaluatedLobachevskyUniversityElectrocardiographyDatabaseassessperformanceAdditionallymodel'sscopeextendedusing20212018datasetssurpassesexistingtraditionalapproachesremarkable14%48%46%QRSTMoreoverexceptionalabilityachievingSEpositivepredictionrate05%59%62%analyzingRRintervaldifferencesexistencewavesmethodidentification20%accuracytestsetstrongdatasetenableshighlyfacilitatingautomatedanalysislarge-scalerecordingsimprovingSEResUTer:signaldataaugmentationtransformer

Similar Articles

Cited By