arXiv:2508.20398cs.LGcs.AI2025-08

用时频联合损失提升心电图去噪精度,保障数字心脏建模可靠性。

TF-TransUNet1D: Time-Frequency Guided Transformer U-Net for Robust ECG Denoising in Digital Twin

  • 结合U-Net与Transformer,捕捉心电信号局部形态与长程依赖。
  • 在MIT-BIH和NSTDB数据集上实现0.1285的均方误差与0.9540相关系数。
  • 适合心电数字孪生、实时监测等对信号质量要求高的场景。

心电图(ECG)是心脏数字孪生的基础数据源,但常受噪声和伪影影响而降低诊断价值。为此,我们提出TF-TransUNet1D,一种新型一维深度神经网络,融合基于U-Net的编码器-解码器结构与Transformer编码器,并采用时频域联合损失函数进行优化。该模型旨在同时捕捉局部波形特征与长程时间依赖性,以保持心电信号的诊断完整性。为增强去噪鲁棒性,引入双域损失函数,联合优化时域波形重建与频域谱保真度。其中频域部分能有效抑制高频噪声,同时保留信号谱结构,恢复细微但临床重要的波形成分。我们在合成污染的MIT-BIH心律失常数据库与噪声压力测试数据库(NSTDB)上评估模型,对比当前最优基线方法,在信噪比提升与误差指标上均表现一致优势,平均绝对误差达0.1285,皮尔逊相关系数为0.9540。本工作填补了心脏数字孪生预处理流程中的关键空白,推动更可靠的真实动态监测与个性化建模。

原文摘要 · Abstract (English)

Electrocardiogram (ECG) signals serve as a foundational data source for cardiac digital twins, yet their diagnostic utility is frequently compromised by noise and artifacts. To address this issue, we propose TF-TransUNet1D, a novel one-dimensional deep neural network that integrates a U-Net-based encoder-decoder architecture with a Transformer encoder, guided by a hybrid time-frequency domain loss. The model is designed to simultaneously capture local morphological features and long-range temporal dependencies, which are critical for preserving the diagnostic integrity of ECG signals. To enhance denoising robustness, we introduce a dual-domain loss function that jointly optimizes waveform reconstruction in the time domain and spectral fidelity in the frequency domain. In particular, the frequency-domain component effectively suppresses high-frequency noise while maintaining the spectral structure of the signal, enabling recovery of subtle but clinically significant waveform components. We evaluate TF-TransUNet1D using synthetically corrupted signals from the MIT-BIH Arrhythmia Database and the Noise Stress Test Database (NSTDB). Comparative experiments against state-of-the-art baselines demonstrate consistent superiority of our model in terms of SNR improvement and error metrics, achieving a mean absolute error of 0.1285 and Pearson correlation coefficient of 0.9540. By delivering high-precision denoising, this work bridges a critical gap in pre-processing pipelines for cardiac digital twins, enabling more reliable real-time monitoring and personalized modeling.

心电图去噪数字孪生时频分析Transformer

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。