用噪声模型精准分割乳腺癌影像,提升低分辨率图像的肿瘤定位能力
PGDiffSeg: Prior-Guided Denoising Diffusion Model with Parameter-Shared Attention for Breast Cancer Segmentation
- 设计并行噪声与语义处理管道,引入参数共享注意力模块融合多层语义信息
- 在自编码器上实现94.3%的Dice分数,优于现有最佳方法,显著提升分割精度
- 首次探讨生成式扩散模型在乳腺癌分割中的可解释性,适合医学影像研究者
早期通过影像检测和准确诊断对降低乳腺癌高死亡率至关重要。然而,在低分辨率和高噪声医学图像中定位肿瘤极为困难。为此,本文提出一种新型的PGDiffSeg(先验引导的去噪扩散模型,采用参数共享注意力)方法,将扩散去噪技术应用于乳腺癌医学图像分割,从高斯噪声中精确恢复病灶区域。首先,设计并行的噪声处理与语义信息处理管道,并在多层中引入参数共享注意力模块(PSA),无缝整合两者,使模型在去噪过程中融入多层次语义细节,生成高精度分割图。其次,提出基于先验知识的引导策略,模拟临床医生决策过程,增强肿瘤位置定位能力。最后,首次系统讨论生成式扩散模型在乳腺癌分割任务中的可解释性。大量实验表明,本模型优于当前最先进方法,证实其作为灵活去噪扩散模型在医学影像研究中的有效性。代码将于后续公开。
原文摘要 · Abstract (English)
Early detection through imaging and accurate diagnosis is crucial in mitigating the high mortality rate associated with breast cancer. However, locating tumors from low-resolution and high-noise medical images is extremely challenging. Therefore, this paper proposes a novel PGDiffSeg (Prior-Guided Diffusion Denoising Model with Parameter-Shared Attention) that applies diffusion denoising methods to breast cancer medical image segmentation, accurately recovering the affected areas from Gaussian noise. Firstly, we design a parallel pipeline for noise processing and semantic information processing and propose a parameter-shared attention module (PSA) in multi-layer that seamlessly integrates these two pipelines. This integration empowers PGDiffSeg to incorporate semantic details at multiple levels during the denoising process, producing highly accurate segmentation maps. Secondly, we introduce a guided strategy that leverages prior knowledge to simulate the decision-making process of medical professionals, thereby enhancing the model's ability to locate tumor positions precisely. Finally, we provide the first-ever discussion on the interpretability of the generative diffusion model in the context of breast cancer segmentation. Extensive experiments have demonstrated the superiority of our model over the current state-of-the-art approaches, confirming its effectiveness as a flexible diffusion denoising method suitable for medical image research. Our code will be publicly available later.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。