用解剖先验控制生成4D心脏MRI,提升数据质量和模型泛化能力。
Anatomy-Guided Residual Motion Diffusion for Controllable 4D Cardiac MRI Synthesis

- 分步扩散模型分离解剖结构与运动,实现可控生成
- 生成数据使分割性能提升1.4%(Dice)和3.0mm(Hausdorff)
- 适合缺乏标注数据或跨设备场景的医学影像研究者
构建4D(3D+时间)医学影像的鲁棒人工智能模型受限于标注数据少、设备间域偏移和隐私问题。为此,提出一种可控制的4D生成框架,用于解剖一致的数据增强。基于半监督变分自编码器学习紧凑的解剖体积潜在表示,并统一预测对齐的分割掩膜。通过级联潜空间扩散模型(LDM),将解剖结构与时间动态解耦:静态LDM在临床先验(诊断、体积测量)条件下生成个体化解剖,后续运动LDM估计残差潜变量运动,确保4D序列严格时序一致性。在心肌电影MRI上评估该方法。多数据集实验显示,静态解剖可控性高(皮尔逊相关系数r > 0.8),时序一致性强(FVD = 288.08)。跨厂商泛化实验中,用合成4D序列增强训练集显著提升下游分割性能:使用nnU-Net,平均Dice提升1.4%,左心室Dice提升2.8%,边界误差减少5.4mm,Hausdorff距离降低3.0mm。整体框架为4D医学图像合成提供可扩展、可控方案,支持在有限标注和跨设备差异下开发更鲁棒模型。代码已开源:https://github.com/cyiheng/4DCardiacMRISynthesis。
原文摘要 · Abstract (English)
Developing robust artificial intelligence models for 4D (3D + time) medical imaging is constrained by limited annotated data, inter-device domain shifts, and privacy restrictions. To address this, we propose a 4D controllable generative framework for anatomically consistent data augmentation. A semi-supervised variational autoencoder learns a compact latent representation of anatomical volumes while jointly predicting aligned segmentation masks in a unified framework. Anatomical structure is then disentangled from temporal dynamics through a cascaded latent diffusion model (LDM). A static LDM generates subject-specific anatomy conditioned on clinical priors (diagnosis and volumes measures) and a subsequent motion LDM estimates residual latent motions, ensuring strict temporal coherence across the 4D sequence. The proposed approach was evaluated on cine cardiac MRI as a representative 4D imaging application. Experiments across multiple datasets demonstrate high controllability of static anatomy (Pearson r > 0.8) and strong temporal coherence (FVD = 288.08). In cross-vendor generalization experiments, augmenting training sets with synthetic 4D sequences significantly improves downstream segmentation performance. Using nnU-Net, the proposed augmentation strategy improves the average Dice score by 1.4% and reduces the Hausdorff Distance by 3.0mm compared to training on real data alone, for the left ventricle, Dice improves by 2.8% with a 5.4mm reduction in boundary error. Overall, this framework provides a scalable and controllable solution for 4D medical image synthesis, supporting the development of more robust models with limited annotations and cross-vendor variability. Code available on https://github.com/cyiheng/4DCardiacMRISynthesis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。