用可控扩散模型生成个性化心电图,保持患者特征并支持多种疾病条件控制。
ECGTwin: Personalized ECG Generation Using Controllable Diffusion Model
- 通过对比学习提取个体心电特征,无需真实标签。
- 生成心电图保真度高、多样性好,且能精准保留个人特征。
- 适合个性化医疗诊断与数字孪生系统研究者使用。
个性化心电图(ECG)生成旨在为特定患者创建数字孪生心电数据,推动医疗向精准个体化转型,同时保留传统群体级合成的优势。但该任务面临两大挑战:在无真实标签情况下提取个体特征,以及在不干扰生成模型的前提下注入多种心脏疾病条件。本文提出ECGTwin,一种两阶段框架:第一阶段采用对比学习训练的个体基线提取器,从参考心电图中稳健捕捉个人特征;第二阶段通过新颖的AdaX条件注入器,将提取的个体特征与目标心脏条件分别通过两条专用路径注入基于扩散模型的生成过程。定性和定量实验表明,本模型不仅能生成高保真、多样化的心电图,实现细粒度控制,还能有效保留个体特异性特征。此外,ECGTwin在下游自动诊断任务中表现优异,验证了其在精准个性化医疗中的应用潜力。
原文摘要 · Abstract (English)
Personalized electrocardiogram (ECG) generation is to simulate a patient's ECG digital twins tailored to specific conditions. It has the potential to transform traditional healthcare into a more accurate individualized paradigm, while preserving the key benefits of conventional population-level ECG synthesis. However, this promising task presents two fundamental challenges: extracting individual features without ground truth and injecting various types of conditions without confusing generative model. In this paper, we present ECGTwin, a two-stage framework designed to address these challenges. In the first stage, an Individual Base Extractor trained via contrastive learning robustly captures personal features from a reference ECG. In the second stage, the extracted individual features, along with a target cardiac condition, are integrated into the diffusion-based generation process through our novel AdaX Condition Injector, which injects these signals via two dedicated and specialized pathways. Both qualitative and quantitative experiments have demonstrated that our model can not only generate ECG signals of high fidelity and diversity by offering a fine-grained generation controllability, but also preserving individual-specific features. Furthermore, ECGTwin shows the potential to enhance ECG auto-diagnosis in downstream application, confirming the possibility of precise personalized healthcare solutions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。