用扩散模型生成心脏MRI合成数据,解决跨中心图像差异问题。
Can Diffusion Models Bridge the Domain Gap in Cardiac MR Imaging?
- 用扩散模型生成结构一致的合成心脏MRI图像,保持与源域相似性。
- 在未见目标域上,分割性能显著提升(p < 0.01)。
- 适用于数据少的临床场景,减少对迁移学习的依赖。
磁共振成像,包括心脏MRI,因设备和采集协议差异易出现领域偏移,导致训练好的AI模型在真实场景中性能下降。传统方法如数据增强或在线微调存在局限。合成数据是替代方案,但生成模型难以保证解剖结构一致性。为此,我们提出在源域上训练的扩散模型(DM),生成与参考图像相似的合成心脏MRI。该方法保持空间和结构保真度,确保与源域一致并兼容分割掩码。我们在多中心心脏MRI分割任务中评估了2D nnU-Net、3D nnU-Net和普通U-Net的表现。通过领域泛化(在合成源域数据上训练领域不变模型)和领域自适应(用扩散模型将目标域数据映射到源域)两种策略,均显著提升了在未见目标域上的分割性能(表面评估指标,Welch's t检验,p < 0.01)。该方法减轻了对迁移学习或在线训练的需求,尤其适用于数据稀缺的场景。
原文摘要 · Abstract (English)
Magnetic resonance (MR) imaging, including cardiac MR, is prone to domain shift due to variations in imaging devices and acquisition protocols. This challenge limits the deployment of trained AI models in real-world scenarios, where performance degrades on unseen domains. Traditional solutions involve increasing the size of the dataset through ad-hoc image augmentation or additional online training/transfer learning, which have several limitations. Synthetic data offers a promising alternative, but anatomical/structural consistency constraints limit the effectiveness of generative models in creating image-label pairs. To address this, we propose a diffusion model (DM) trained on a source domain that generates synthetic cardiac MR images that resemble a given reference. The synthetic data maintains spatial and structural fidelity, ensuring similarity to the source domain and compatibility with the segmentation mask. We assess the utility of our generative approach in multi-centre cardiac MR segmentation, using the 2D nnU-Net, 3D nnU-Net and vanilla U-Net segmentation networks. We explore domain generalisation, where, domain-invariant segmentation models are trained on synthetic source domain data, and domain adaptation, where, we shift target domain data towards the source domain using the DM. Both strategies significantly improved segmentation performance on data from an unseen target domain, in terms of surface-based metrics (Welch's t-test, p < 0.01), compared to training segmentation models on real data alone. The proposed method ameliorates the need for transfer learning or online training to address domain shift challenges in cardiac MR image analysis, especially useful in data-scarce settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。