arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2505.03494cs.CV
收藏

UPMAD-Net:一种具有不确定性引导和自适应多模态特征融合的脑肿瘤分割网络

UPMAD-Net: A Brain Tumor Segmentation Network with Uncertainty Guidance and Adaptive Multimodal Feature Fusion

School of Computer Science and Technology, Zhengzhou University of Light Industry(郑州轻工业大学计算机与通信工程学院);Kennesaw State University(肯尼索州立大学);Michigan Technological University(密歇根理工大学)

Zhanyuan Jia, Ni Yao, Danyang Sun, Chuang Han, Yanting Li, Jiaofen Nan, Fubao Zhu, Chen Zhao, Weihua Zhou

AI总结:

本文提出UPMAD-Net,一种结合区域生长先验知识、多尺度特征融合与自适应注意力机制,并利用MC Dropout进行不确定性估计的3D脑肿瘤分割网络,在BraTS数据集上取得优于现有方法的性能。

摘要:

背景:脑肿瘤分割对脑肿瘤的诊断和治疗具有重要影响。由于脑肿瘤形状不规则、边界模糊且变异性高,准确的脑肿瘤分割仍然具有挑战性。目标:我们提出了一种将深度学习与来自区域生长算法的先验知识相结合的脑肿瘤分割方法。方法:所提出的方法利用多尺度特征融合(MSFF)模块和自适应注意力机制(AAM)来提取多尺度特征并捕获全局上下文信息。为了增强模型在低置信度区域的鲁棒性,采用蒙特卡洛丢弃(MC Dropout)策略进行不确定性估计。结果:大量实验表明,所提出的方法在脑肿瘤分割(BraTS)数据集上取得了优越的性能,显著优于各种最先进的方法。在BraTS2021数据集上,测试Dice分数分别为:增强肿瘤(ET)分割89.18%,全肿瘤(WT)分割93.67%,肿瘤核心(TC)分割91.23%。在BraTS2019验证集上,ET、WT和TC分割的验证Dice分数分别为87.43%、90.92%和90.40%。消融研究进一步证实了每个模块对分割精度的贡献,表明每个组件在整体性能提升中都发挥了重要作用。结论:本研究提出了一种基于U-Net架构的新型3D脑肿瘤分割网络。通过引入先验知识并采用不确定性估计方法,提高了鲁棒性和性能。所提出方法的代码可在https://github.com/chenzhao2023/UPMAD_Net_BrainSeg获取。

英文摘要:

Background: Brain tumor segmentation has a significant impact on the diagnosis and treatment of brain tumors. Accurate brain tumor segmentation remains challenging due to their irregular shapes, vague boundaries, and high variability. Objective: We propose a brain tumor segmentation method that combines deep learning with prior knowledge derived from a region-growing algorithm. Methods: The proposed method utilizes a multi-scale feature fusion (MSFF) module and adaptive attention mechanisms (AAM) to extract multi-scale features and capture global contextual information. To enhance the model's robustness in low-confidence regions, the Monte Carlo Dropout (MC Dropout) strategy is employed for uncertainty estimation. Results: Extensive experiments demonstrate that the proposed method achieves superior performance on Brain Tumor Segmentation (BraTS) datasets, significantly outperforming various state-of-the-art methods. On the BraTS2021 dataset, the test Dice scores are 89.18% for Enhancing Tumor (ET) segmentation, 93.67% for Whole Tumor (WT) segmentation, and 91.23% for Tumor Core (TC) segmentation. On the BraTS2019 validation set, the validation Dice scores are 87.43%, 90.92%, and 90.40% for ET, WT, and TC segmentation, respectively. Ablation studies further confirmed the contribution of each module to segmentation accuracy, indicating that each component played a vital role in overall performance improvement. Conclusion: This study proposed a novel 3D brain tumor segmentation network based on the U-Net architecture. By incorporating the prior knowledge and employing the uncertainty estimation method, the robustness and performance were improved. The code for the proposed method is available at https://github.com/chenzhao2023/UPMAD_Net_BrainSeg.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑