用语义标签生成高质量左心房LGE-MRI,提升分割精度
3D Conditional Image Synthesis of Left Atrial LGE MRI from Composite Semantic Masks
- 基于语义标签图生成3D左心房LGE-MRI,采用SPADE-LDM模型
- 合成数据使心腔分割Dice系数从0.908提升至0.936
- 适合心血管影像数据稀缺场景下的模型训练增强
从晚钆增强(LGE)MRI中分割左心房(LA)壁与心内膜对评估心房纤维化至关重要。由于数据有限且解剖结构复杂,构建准确的机器学习分割模型仍具挑战。本文探索3D条件生成模型作为扩充稀缺训练数据的解决方案,并提出一个从复合语义标签图(融合专家标注与无监督组织聚类)生成高保真3D LGE-MRI体积的流程。采用三种3D条件生成器:Pix2Pix GAN、SPADE-GAN 和 SPADE-LDM。合成图像在真实性和下游分割任务中进行评估。SPADE-LDM生成的图像最逼真且结构准确,其FID为4.063,显著优于Pix2Pix(FID=40.821)和SPADE-GAN(FID=7.652)。使用合成数据增强后,3D U-Net模型的左心房腔分割Dice分数从0.908提升至0.936(p<0.05),具有统计学显著性。结果表明,标签条件化的3D生成可有效提升欠代表性心脏结构的分割性能。
原文摘要 · Abstract (English)
Segmentation of the left atrial (LA) wall and endocardium from late gadolinium-enhanced (LGE) MRI is essential for quantifying atrial fibrosis in patients with atrial fibrillation. The development of accurate machine learning-based segmentation models remains challenging due to the limited availability of data and the complexity of anatomical structures. In this work, we investigate 3D conditional generative models as potential solution for augmenting scarce LGE training data and improving LA segmentation performance. We develop a pipeline to synthesize high-fidelity 3D LGE MRI volumes from composite semantic label maps combining anatomical expert annotations with unsupervised tissue clusters, using three 3D conditional generators (Pix2Pix GAN, SPADE-GAN, and SPADE-LDM). The synthetic images are evaluated for realism and their impact on downstream LA segmentation. SPADE-LDM generates the most realistic and structurally accurate images, achieving an FID of 4.063 and surpassing GAN models, which have FIDs of 40.821 and 7.652 for Pix2Pix and SPADE-GAN, respectively. When augmented with synthetic LGE images, the Dice score for LA cavity segmentation with a 3D U-Net model improved from 0.908 to 0.936, showing a statistically significant improvement (p < 0.05) over the baseline.These findings demonstrate the potential of label-conditioned 3D synthesis to enhance the segmentation of under-represented cardiac structures.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。