用扩散模型合成脑瘤影像,无标注也能精准分割
ReCoSeg++:Extended Residual-Guided Cross-Modal Diffusion for Brain Tumor Segmentation
- 用残差引导扩散模型从多模态MRI重建增强扫描图
- 在BraTS 2021上实现93.02%的骰子系数,超越旧版方法
- 适合缺乏标注数据的临床多中心MRI分析场景
准确分割MRI中的脑肿瘤对临床诊断和治疗规划至关重要。本文提出一种半监督、两阶段框架,将ReCoSeg方法扩展至更大更异质的BraTS 2021数据集,且无需真实标签进行分割训练。第一阶段采用残差引导的去噪扩散概率模型(DDPM),通过FLAIR、T1和T2扫描重建T1ce模态。残差图(预测与实际T1ce的差异)作为空间先验,提升下游分割性能。第二阶段使用轻量级U-Net,将残差图与原始T1、T2、FLAIR图像拼接输入,优化整体肿瘤分割。为应对BraTS 2021的数据规模与多样性,引入切片级过滤排除无效样本,并优化阈值策略以平衡精确率与召回率。该方法在BraTS 2021上实现整体肿瘤分割的骰子系数93.02%、交并比86.7%,优于ReCoSeg在BraTS 2020上的表现(骰子:91.7%,交并比:85.3%),展现出更强的准确性和可扩展性。
原文摘要 · Abstract (English)
Accurate segmentation of brain tumors in MRI scans is critical for clinical diagnosis and treatment planning. We propose a semi-supervised, two-stage framework that extends the ReCoSeg approach to the larger and more heterogeneous BraTS 2021 dataset, while eliminating the need for ground-truth masks for the segmentation objective. In the first stage, a residual-guided denoising diffusion probabilistic model (DDPM) performs cross-modal synthesis by reconstructing the T1ce modality from FLAIR, T1, and T2 scans. The residual maps, capturing differences between predicted and actual T1ce images, serve as spatial priors to enhance downstream segmentation. In the second stage, a lightweight U-Net takes as input the concatenation of residual maps, computed as the difference between real T1ce and synthesized T1ce, with T1, T2, and FLAIR modalities to improve whole tumor segmentation. To address the increased scale and variability of BraTS 2021, we apply slice-level filtering to exclude non-informative samples and optimize thresholding strategies to balance precision and recall. Our method achieves a Dice score of $93.02\%$ and an IoU of $86.7\%$ for whole tumor segmentation on the BraTS 2021 dataset, outperforming the ReCoSeg baseline on BraTS 2020 (Dice: $91.7\%$, IoU: $85.3\%$), and demonstrating improved accuracy and scalability for real-world, multi-center MRI datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。