用扩散模型生成带标签脑部MRI,提升脑室扩大分割精度。
Guided Synthesis of Labeled Brain MRI Data Using Latent Diffusion Models for Segmentation of Enlarged Ventricles
- 用扩散模型按脑室体积生成带标注的合成MRI数据。
- 合成数据训练模型使误差降低至6.23mL,优于现有方法。
- 适合医疗影像数据稀缺场景,对脑室分割研究有参考价值。
医学深度学习面临数据稀缺、图像异质性和隐私问题。本研究通过合成数据提升脑部MRI中脑室分割效果。采用两个潜在扩散模型(LDM):基于10,000张掩码训练的掩码生成器,和基于6,881例扫描优化的SPADE图像生成器,实现以3D脑掩码为条件生成MRI。通过将掩码生成器与脑室体积联合控制,并结合无分类器引导,可调节生成图像的脑室体积分布。随后在真实数据、增强数据和全合成数据上分别训练三个nnU-Net模型,测试集为独立患者数据,手动勾画为真值。真实数据训练模型平均绝对误差(MAE)为9.09±12.18 mL,合成数据训练模型为7.52±4.81 mL,增强数据训练模型为6.23±4.33 mL。后者还达到最高Dice分数0.892±0.05,略优于SynthSeg,与真实数据模型相当。合成模型表现接近SynthSeg。结果表明,基于引导的合成方法能有效提升脑室扩大分割性能,优于现有最先进模型。
原文摘要 · Abstract (English)
Deep learning models in medical contexts face challenges like data scarcity, inhomogeneity, and privacy concerns. This study focuses on improving ventricular segmentation in brain MRI images using synthetic data. We employed two latent diffusion models (LDMs): a mask generator trained using 10,000 masks, and a corresponding SPADE image generator optimized using 6,881 scans to create an MRI conditioned on a 3D brain mask. Conditioning the mask generator on ventricular volume in combination with classifier-free guidance enabled the control of the ventricular volume distribution of the generated synthetic images. Next, the performance of the synthetic data was tested using three nnU-Net segmentation models trained on a real, augmented and entirely synthetic data, respectively. The resulting models were tested on a completely independent hold-out dataset of patients with enlarged ventricles, with manual delineation of the ventricles used as ground truth. The model trained on real data showed a mean absolute error (MAE) of 9.09 \pm 12.18 mL in predicted ventricular volume, while the models trained on synthetic and augmented data showed MAEs of 7.52 \pm 4.81 mL and 6.23 \pm 4.33 mL, respectively. Both the synthetic and augmented model also outperformed the state-of-the-art model SynthSeg, which due to limited performance in cases of large ventricular volumes, showed an MAE of 7.73 \pm 12.12 mL with a factor of 3 higher standard deviation. The model trained on augmented data showed the highest Dice score of 0.892 \pm 0.05, slightly outperforming SynthSeg and on par with the model trained on real data. The synthetic model performed similar to SynthSeg. In summary, we provide evidence that guided synthesis of labeled brain MRI data using LDMs improves the segmentation of enlarged ventricles and outperforms existing state-of-the-art segmentation models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。