arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2505.03426cs.CVcs.AI

表型引导的生成模型用于高保真心肌磁共振成像合成:推进预训练与临床应用

Phenotype-Guided Generative Model for High-Fidelity Cardiac MRI Synthesis: Advancing Pretraining and Clinical Applications

Zhejiang University(浙江大学);The First Affiliated Hospital of Zhejiang University School of Medicine(浙江大学医学院附属第一医院)

Ziyu Li, Yujian Hu, Zhengyao Ding, Yiheng Mao, Haitao Li, Fan Yi, Hongkun Zhang, Zhengxing Huang

AI总结:

本文提出CPGG框架,利用心脏表型引导掩码自回归扩散模型生成高保真CMR数据以扩展预训练,显著提升诊断和表型预测等下游任务性能。

摘要:

心肌磁共振(CMR)成像是诊断心脏疾病和评估心脏健康的重要非侵入性工具。然而,大规模高质量CMR数据集的有限可用性对人工智能(AI)在该领域的有效应用构成了重大挑战。即使是无标签数据的数量及其所覆盖的健康状态也难以满足模型预训练的需求,这阻碍了AI模型在下游任务上的性能。在本研究中,我们提出了Cardiac Phenotype-Guided CMR Generation(CPGG),这是一种用于生成多样化CMR数据的新方法,覆盖广泛的心脏健康状态谱。CPGG框架由两个阶段组成:在第一阶段,使用从CMR数据中提取的心脏表型训练一个生成模型;在第二阶段,一个以这些表型为条件的掩码自回归扩散模型生成高保真CMR电影序列,以细粒度方式捕捉心脏的结构和功能特征。我们合成了大量CMR数据以扩展预训练数据。实验结果表明,CPGG生成了高质量的合成CMR数据,显著提升了多个下游任务的性能,包括诊断和心脏表型预测。这些提升在公共和私有数据集上均得到验证,凸显了我们方法的有效性。代码可在https://anonymous.4open.science/r/CPGG获取。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑