用肿瘤空间先验提升脑部MRI分割精度,减少无效计算。
PGR-Net: Prior-Guided ROI Reasoning Network for Brain Tumor MRI Segmentation
- 引入数据驱动的空间先验和分层候选区域选择机制
- 在BraTS2019/2023上达89.02%~91.82%的骰子分数
- 参数仅864万,适合临床部署的轻量级模型
脑肿瘤MRI分割对临床诊断与治疗规划至关重要,可实现病灶精准检测与放疗靶区勾画。然而,肿瘤病灶仅占体积的一小部分,导致空间稀疏性严重,现有分割网络常忽略临床观察到的肿瘤分布先验,造成大量背景区域冗余特征计算。为此,本文提出PGR-Net(先验引导的感兴趣区域推理网络)——一种显式关注区域的框架,通过数据驱动的空间先验集捕捉肿瘤病灶的分布与尺度特性,提供全局指导以增强分割稳定性。PGR-Net引入分层Top-K ROI决策机制,在编码器各层逐步筛选最可信的病灶候选区域,提升定位精度。进一步设计了WinGS-ROI(窗口化高斯-空间衰减区域)模块,利用多窗口高斯模板结合空间衰减函数生成中心增强引导图,引导全网特征学习。结合窗口化RetNet主干网络,增强定位可靠性。在BraTS-2019/2023与MSD Task01上实验表明,PGR-Net始终优于现有方法,仅使用864万参数,全肿瘤区域骰子分数达89.02%、91.82%与89.67%。代码已开源。
原文摘要 · Abstract (English)
Brain tumor MRI segmentation is essential for clinical diagnosis and treatment planning, enabling accurate lesion detection and radiotherapy target delineation. However, tumor lesions occupy only a small fraction of the volumetric space, resulting in severe spatial sparsity, while existing segmentation networks often overlook clinically observed spatial priors of tumor occurrence, leading to redundant feature computation over extensive background regions. To address this issue, we propose PGR-Net (Prior-Guided ROI Reasoning Network) - an explicit ROI-aware framework that incorporates a data-driven spatial prior set to capture the distribution and scale characteristics of tumor lesions, providing global guidance for more stable segmentation. Leveraging these priors, PGR-Net introduces a hierarchical Top-K ROI decision mechanism that progressively selects the most confident lesion candidate regions across encoder layers to improve localization precision. We further develop the WinGS-ROI (Windowed Gaussian-Spatial Decay ROI) module, which uses multi-window Gaussian templates with a spatial decay function to produce center-enhanced guidance maps, thus directing feature learning throughout the network. With these ROI features, a windowed RetNet backbone is adopted to enhance localization reliability. Experiments on BraTS-2019/2023 and MSD Task01 show that PGR-Net consistently outperforms existing approaches while using only 8.64M Params, achieving Dice scores of 89.02%, 91.82%, and 89.67% on the Whole Tumor region. Code is available at https://github.com/CNU-MedAI-Lab/PGR-Net.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。