用心脏表型指导生成高质量心脏磁共振影像,解决数据少难题。
Phenotype-Guided Generative Model for High-Fidelity Cardiac MRI Synthesis: Advancing Pretraining and Clinical Applications
- 根据心脏表型训练生成模型,分两阶段合成影像。
- 生成数据显著提升下游诊断与表型预测性能。
- 适合需要扩充医学影像数据的研究者使用。
心脏磁共振(CMR)是诊断心脏病和评估心脏健康的重要无创工具,但大规模高质量CMR数据集的稀缺严重制约了人工智能在该领域的应用。现有未标注数据量及覆盖的健康状态范围难以满足模型预训练需求,影响下游任务表现。本文提出心脏表型引导的CMR生成方法(CPGG),通过两阶段框架生成涵盖多种心脏健康状态的多样化CMR数据。第一阶段基于CMR数据提取的心脏表型训练生成模型;第二阶段采用条件于表型的掩码自回归扩散模型,生成高保真心动电影序列,精细捕捉心脏结构与功能特征。我们合成大量CMR数据以扩充预训练集。实验表明,CPGG生成的数据能显著提升多种下游任务性能,包括疾病诊断与心脏表型预测,在公开与私有数据集上均验证有效。代码已公开于https://anonymous.4open.science/r/CPGG。
原文摘要 · Abstract (English)
Cardiac Magnetic Resonance (CMR) imaging is a vital non-invasive tool for diagnosing heart diseases and evaluating cardiac health. However, the limited availability of large-scale, high-quality CMR datasets poses a major challenge to the effective application of artificial intelligence (AI) in this domain. Even the amount of unlabeled data and the health status it covers are difficult to meet the needs of model pretraining, which hinders the performance of AI models on downstream tasks. In this study, we present Cardiac Phenotype-Guided CMR Generation (CPGG), a novel approach for generating diverse CMR data that covers a wide spectrum of cardiac health status. The CPGG framework consists of two stages: in the first stage, a generative model is trained using cardiac phenotypes derived from CMR data; in the second stage, a masked autoregressive diffusion model, conditioned on these phenotypes, generates high-fidelity CMR cine sequences that capture both structural and functional features of the heart in a fine-grained manner. We synthesized a massive amount of CMR to expand the pretraining data. Experimental results show that CPGG generates high-quality synthetic CMR data, significantly improving performance on various downstream tasks, including diagnosis and cardiac phenotypes prediction. These gains are demonstrated across both public and private datasets, highlighting the effectiveness of our approach. Code is availabel at https://anonymous.4open.science/r/CPGG.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。