arXiv:2409.00807cs.CVcs.AI2024-09被引 7

用扩散模型统一处理多中心脑影像差异,保留解剖细节且效果优于传统方法。

Diffusion based multi-domain neuroimaging harmonization method with preservation of anatomical details

  • 基于扩散模型实现跨多个中心的脑影像统一,单模型处理多域数据
  • 在ADNI1和ABIDE II数据集上保持解剖结构清晰,FID得分显著更优
  • 可有效提升脑血管周围间隙分割一致性,适合多中心神经影像研究

多中心脑影像研究因不同扫描站点间的批次差异导致技术变异性,影响数据整合与研究可靠性。现有神经影像标准化方法试图缩小此类技术差距,但生成对抗网络(GAN)虽被广泛应用,其生成图像常出现伪影或解剖结构扭曲。鉴于去噪扩散概率模型能生成高保真图像,本文评估其在神经影像标准化中的有效性。实验表明,扩散模型具备同时处理多域影像的能力,而传统GAN方法通常仅限于两域间映射。所提方法通过学习域不变解剖条件,在每一步扩散过程中既准确保留解剖细节,又能区分不同批次差异。在公开数据集ADNI1与ABIDE II上验证,该方法生成结果在解剖结构一致性方面表现优异,且相比GAN基线模型取得更低的FID分数。进一步定量与定性分析、消融实验以及对脑血管周围间隙(PVS)分割一致性的改善,均证实了该方法的有效性。

原文摘要 · Abstract (English)

Multi-center neuroimaging studies face technical variability due to batch differences across sites, which potentially hinders data aggregation and impacts study reliability.Recent efforts in neuroimaging harmonization have aimed to minimize these technical gaps and reduce technical variability across batches. While Generative Adversarial Networks (GAN) has been a prominent method for addressing image harmonization tasks, GAN-harmonized images suffer from artifacts or anatomical distortions. Given the advancements of denoising diffusion probabilistic model which produces high-fidelity images, we have assessed the efficacy of the diffusion model for neuroimaging harmonization. we have demonstrated the diffusion model's superior capability in harmonizing images from multiple domains, while GAN-based methods are limited to harmonizing images between two domains per model. Our experiments highlight that the learned domain invariant anatomical condition reinforces the model to accurately preserve the anatomical details while differentiating batch differences at each diffusion step. Our proposed method has been tested on two public neuroimaging dataset ADNI1 and ABIDE II, yielding harmonization results with consistent anatomy preservation and superior FID score compared to the GAN-based methods. We have conducted multiple analysis including extensive quantitative and qualitative evaluations against the baseline models, ablation study showcasing the benefits of the learned conditions, and improvements in the consistency of perivascular spaces (PVS) segmentation through harmonization.

脑影像扩散模型多中心解剖保真

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。