用图像处理与深度学习结合的方法,从纸质心电图还原出高质量信号。
Combining Hough Transform and Deep Learning Approaches to Reconstruct ECG Signals From Printouts
- 先用霍夫变换校正图像角度,再用U-Net分割信号线段
- 在21799份数据上测试,信噪比达17.02,挑战赛得分12.15第一
- 适合需要将纸质医疗记录数字化的研究者和开发者
本文介绍我们团队(SignalSavants)在2024年George B. Moody PhysioNet挑战赛中的获奖贡献。挑战赛包含两个目标:从纸质打印心电图中重建信号并进行心脏病分类。本研究聚焦于第一个任务。尽管如今多数心电图已数字化,但纸质心电图在全球仍广泛使用。将其数字化有助于构建更丰富的数据集并支持自动化分析。然而,不同记录标准及图像质量差的问题,要求采用以数据为中心的方法,开发具备强泛化能力的鲁棒模型。我们的方法包括构建多样化训练集、使用霍夫变换旋转图像、基于U-Net的分割模型识别单个信号,以及掩码向量化来重构信号。采用PTB-XL数据集提出的10折分层交叉验证划分(共21,799条记录)评估模型性能。在数字化任务中,模型平均交叉验证信噪比为17.02,挑战赛隐藏集官方得分为12.15,获得第一名。研究揭示了构建鲁棒、可泛化的数字化方法所面临的挑战,这类模型虽需大量资源(数据、时间、算力),却具有显著提升数据多样性的潜力。
原文摘要 · Abstract (English)
This work presents our team's (SignalSavants) winning contribution to the 2024 George B. Moody PhysioNet Challenge. The Challenge had two goals: reconstruct ECG signals from printouts and classify them for cardiac diseases. Our focus was the first task. Despite many ECGs being digitally recorded today, paper ECGs remain common throughout the world. Digitising them could help build more diverse datasets and enable automated analyses. However, the presence of varying recording standards and poor image quality requires a data-centric approach for developing robust models that can generalise effectively. Our approach combines the creation of a diverse training set, Hough transform to rotate images, a U-Net based segmentation model to identify individual signals, and mask vectorisation to reconstruct the signals. We assessed the performance of our models using the 10-fold stratified cross-validation (CV) split of 21,799 recordings proposed by the PTB-XL dataset. On the digitisation task, our model achieved an average CV signal-to-noise ratio of 17.02 and an official Challenge score of 12.15 on the hidden set, securing first place in the competition. Our study shows the challenges of building robust, generalisable, digitisation approaches. Such models require large amounts of resources (data, time, and computational power) but have great potential in diversifying the data available.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。