用深度学习精准提取重叠心电图信号,提升数字化准确率。
Deep Learning-Based Digitization of Overlapping ECG Images with Open-Source Python Code
- 两阶段流程:先用U-Net分割重叠信号,再通过自适应网格转为时间序列
- 重叠样本下误差仅0.0029,相关系数达0.9641,显著优于基线
- 代码开源,适合医疗影像数字化与心电数据分析研究者使用
本文针对纸质心电图(ECG)数字化中信号重叠这一常见但未被充分解决的问题,提出一种两阶段方法。第一阶段采用基于U-Net的分割网络,在含重叠信号的数据集上训练,并结合定制增强技术,精准分离主波形;第二阶段利用经典数字化方法,结合自适应网格检测模块,将二值掩码转换为时序信号。实验表明,该方法在细粒度分割任务中达到0.87的交并比(IoU)。在非重叠样本上,均方误差(MSE)为0.0010,皮尔逊相关系数(rho)为0.9644,优于基线(MSE: 0.0015, rho: 0.9366)。在重叠样本上,其性能显著超越基线(MSE: 0.0029 vs. 0.0178, rho: 0.9641 vs. 0.8676)。该方法有效提升了重叠场景下的数字化精度,为模拟心电记录向数字数据的可靠转化提供支持。代码已公开于https://github.com/masoudrahimi39/ECG-code。
原文摘要 · Abstract (English)
This paper addresses the persistent challenge of accurately digitizing paper-based electrocardiogram (ECG) recordings, with a particular focus on robustly handling single leads compromised by signal overlaps-a common yet under-addressed issue in existing methodologies. We propose a two-stage pipeline designed to overcome this limitation. The first stage employs a U-Net based segmentation network, trained on a dataset enriched with overlapping signals and fortified with custom data augmentations, to accurately isolate the primary ECG trace. The subsequent stage converts this refined binary mask into a time-series signal using established digitization techniques, enhanced by an adaptive grid detection module for improved versatility across different ECG formats and scales. Our experimental results demonstrate the efficacy of our approach. The U-Net architecture achieves an IoU of 0.87 for the fine-grained segmentation task. Crucially, our proposed digitization method yields superior performance compared to a well-established baseline technique across both non-overlapping and challenging overlapping ECG samples. For non-overlapping signals, our method achieved a Mean Squared Error (MSE) of 0.0010 and a Pearson Correlation Coefficient (rho) of 0.9644, compared to 0.0015 and 0.9366, respectively, for the baseline. On samples with signal overlap, our method achieved an MSE of 0.0029 and a rho of 0.9641, significantly improving upon the baseline's 0.0178 and 0.8676. This work demonstrates an effective strategy to significantly enhance digitization accuracy, especially in the presence of signal overlaps, thereby laying a strong foundation for the reliable conversion of analog ECG records into analyzable digital data for contemporary research and clinical applications. The implementation is publicly available at this GitHub repository: https://github.com/masoudrahimi39/ECG-code.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。