arXiv:2510.09681cs.CV2025-10

用扩散模型优化脑肿瘤分割,提升边界精度和泛化能力。

NNDM: NN_UNet Diffusion Model for Brain Tumor Segmentation

  • 将NN-UNet与扩散模型结合,通过迭代去噪修正分割结果。
  • 在BraTS 2021上Dice系数提升,豪斯多夫距离降低。
  • 适合需要高精度分割的医学影像分析场景。

磁共振成像中脑肿瘤的精准检测与分割对诊断和治疗规划至关重要。尽管卷积神经网络(如U-Net)已取得进展,现有模型仍面临泛化性差、边界精度不足及数据多样性有限的问题。为此,我们提出NNDM(NN_UNet Diffusion Model),一种融合NN-UNet强特征提取能力与扩散概率模型生成能力的混合框架。该方法通过学习预测掩码与真实掩码之间的残差误差分布,逐步精炼由NN-UNet生成的分割结果。这一迭代去噪过程可有效纠正细微结构不一致,增强肿瘤边界的描绘。在BraTS 2021数据集上的实验表明,NNDM优于传统U-Net及基于Transformer的基线模型,在Dice系数与豪斯多夫距离指标上均有提升。此外,扩散引导的精修增强了跨模态和不同肿瘤亚区的鲁棒性。该工作为确定性分割网络与随机扩散模型的结合提供了新方向,推动了自动化脑肿瘤分析的最新进展。

原文摘要 · Abstract (English)

Accurate detection and segmentation of brain tumors in magnetic resonance imaging (MRI) are critical for effective diagnosis and treatment planning. Despite advances in convolutional neural networks (CNNs) such as U-Net, existing models often struggle with generalization, boundary precision, and limited data diversity. To address these challenges, we propose NNDM (NN\_UNet Diffusion Model)a hybrid framework that integrates the robust feature extraction of NN-UNet with the generative capabilities of diffusion probabilistic models. In our approach, the diffusion model progressively refines the segmentation masks generated by NN-UNet by learning the residual error distribution between predicted and ground-truth masks. This iterative denoising process enables the model to correct fine structural inconsistencies and enhance tumor boundary delineation. Experiments conducted on the BraTS 2021 datasets demonstrate that NNDM achieves superior performance compared to conventional U-Net and transformer-based baselines, yielding improvements in Dice coefficient and Hausdorff distance metrics. Moreover, the diffusion-guided refinement enhances robustness across modalities and tumor subregions. The proposed NNDM establishes a new direction for combining deterministic segmentation networks with stochastic diffusion models, advancing the state of the art in automated brain tumor analysis.

脑肿瘤分割扩散模型医学图像

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