DGRNet: Disagreement-Guided Refinement for Uncertainty-Aware Brain Tumor Segmentation
Authors: Bahram Mohammadi, Yanqiu Wu, Vu Minh Hieu Phan, Sam White, Minh-Son To, Jian Yang, Michael Sheng, Yang Song, +1 more
Organizations: Macquarie University, Sydney, NSW, Australia · Adelaide University, Adelaide, SA, Australia · Flinders University, Adelaide, SA, Australia · University of New South Wales, Sydney, NSW, Australia
Accurate brain tumor segmentation from MRI scans is critical for diagnosis and treatment planning. Despite the strong performance of recent deep learning approaches, two fundamental limitations remain: (1) the lack of reliable uncertainty quantification in single-model predictions, which is essential for clinical deployment because the level of uncertainty may impact treatment decision-making, and (2) the under-utilization of rich information in radiology reports that can guide segmentation in ambiguous regions. In this paper, we propose the Disagreement-Guided Refinement Network (DGRNet), a novel framework that addresses both limitations through multi-view disagreement-based uncertainty estimation and text-conditioned refinement. DGRNet generates diverse predictions via four lightweight view-specific adapters attached to a shared encoder-decoder, enabling efficient uncertainty quantification within a single forward pass. Afterward, we build disagreement maps to identify regions of high segmentation uncertainty, which are then selectively refined according to clinical reports. Moreover, we introduce a diversity-preserving training strategy that combines pairwise similarity penalties and gradient isolation to prevent view collapse. The experimental results on the TextBraTS dataset show that DGRNet favorably improves state-of-the-art segmentation accuracy by 2.4% and 11% in main metrics Dice and HD95, respectively, while providing meaningful uncertainty estimates.
Figures & tables
Figure 1: Overview of the DGRNet architecture. The model consists of four main components: (1) backbone extracts visual features via the Swin Transformer encoder, and textual features via the frozen BioBERT [ 10 ] , fused at the bottleneck, (2) view-specific adapters generate diverse predictions Z(1) , Z(2) , and Z(3) using learnable FiLM modulation with shared decoder and view-specific output heads (Sec. 2.1 ), (3) disagreement module computes an uncertainty map U from variance, pairwise disagreement, and entropy of mean predictions with learnable weights (Sec. 2.2 ), and (4) refinement module uses disagreement-aware spatial attention and text-guided FiLM conditioning to produce a gated residual refinement ΔZ , yielding the final prediction for all three sub-regions Z~=Z^+g⋅ΔZ . (Sec. 2.3 )
Methods
Dice (%) ↑
HD95 ↓
ET
WT
TC
Avg.
ET
WT
TC
Avg.
3D-UNet [ 18 ]
80.4
87.3
81.6
83.1
6.11
10.51
8.93
8.17
nnU-Net [ 9 ]
82.2
87.5
82.6
84.1
4.27
11.90
8.52
8.23
SegResNet [ 7 ]
80.9
88.4
82.3
83.8
6.18
7.28
7.41
6.95
Swin UNETR [ 6 ]
81.0
89.5
80.8
83.8
5.95
8.23
7.03
7.07
Nestedformer [ 26 ]
82.6
89.5
80.2
84.1
5.08
10.51
8.93
8.17
Table 1: Comparison of DGRNet with the state-of-the-art methods on the TextBraTS dataset. † shows the results reproduced on the same platform as ours.
Multi-View
View Div.
Disagree Attn
Refinement
Text Cond.
Dice (%) ↑
HD95 ↓
×
×
×
×
×
84.9
6.88
✓
×
×
×
×
85.6
6.41
✓
✓
×
×
×
86.2
5.94
✓
✓
✓
×
×
86.6
5.48
✓
✓
✓
✓
×
87.1
4.95
✓
✓
✓
✓
✓
87.6
4.56
Table 2: Ablation study of main components of our method.
Table 4
Figure 2: Qualitative comparison with the baseline, using a representative sample that contains all of the sub-regions (TC, WT, and ET).
Figure 3: The strong correlation between segmentation errors and the uncertainty map.
Multimodal MRI is essential for accurate brain tumor segmentation. However, acquiring all modalities at inference is often challenging in practice, which causes intrinsic uncertainty due to unavoidable information loss. Without modeling this uncertainty, existing methods encode incomplete evidence into deterministic representations that appear plausible but lack reliability. In this regime, we propose a probabilistic representation framework that models representations as Gaussian distributions, where their mean captures task information and their variance measures uncertainty from missing evidence. To make variance reflect information deficiency, we regularize the mean from each partial configuration toward its full-modality counterpart, while scaling the variance with the discrepancy between their aligned means. We further introduce a set-inclusive strategy that exploits the hierarchical structure of modality subsets and enforces an ordering constraint to maintain their consistent uncertainty relationships. Extensive experiments on BraTS 2018 and 2020 demonstrate that our approach offers superior performance over baselines across diverse missing-modality scenarios. Code and model checkpoint are available at https://github.com/atlas-sky/SIUM.
Seunghun Baek, Jihwan Park, Jaeyoon Sim +3
Pohang University of Science and Technology, Pohang, South Korea
Medical image segmentation models often report high benchmark accuracy under ideal imaging conditions, yet their failures under clinical degradation can be quiet: sensor noise, patient motion, low- resolution acquisition, and contrast variability may all alter model behavior without producing an obvious warning. We present a reproducible framework for evaluating uncertainty-aware segmentation under con- trolled clinical degradation. Our experiments use a synthetic multimodal brain tumor MRI cohort generated with a biophysical phantom simulator that follows the BraTS protocol. We train U-Net and Attention U-Net baselines for multi-class tumor sub-region segmentation and augment both models with Monte Carlo dropout to estimate per-voxel uncertainty. Across eight clinically motivated corruption types at five severity levels, we measure segmentation accuracy, calibration, failure detection, and selective prediction coverage. On clean data, Attention U-Net achieves a whole-tumor Dice of 0.990; under severe Gaussian noise, its performance falls to 0.089. Predictive uncertainty rises with degradation and tracks segmentation error (Pearson r = 0.53 under severity-3 Gaussian noise), allowing us to flag failures with an AUROC of 0.843. These results argue for uncertainty-aware inference as a practical safety layer in physician-in-the-loop radiology workflows. We release the code, trained models, and evaluation protocol to support direct reproduction.
Pranav Kaliaperumal, Manisha Kaliaperumal
Department of Computer Science University of Colorado Denver Aurora, CO, USA · Department of Biology (Pre-Medical) Creighton University Omaha, NE, USA
Glioma segmentation in multiparametric MRI is a critical component of treatment planning. A segmentation model that fails silently on treatment-critical sub-regions represents a patient safety risk that overlap-based metrics such as Dice scores cannot expose. We ask whether voxel-level uncertainty estimation via Monte Carlo (MC) Dropout can reliably identify segmentation errors in clinically critical sub-regions, and whether calibration failure modes are detectable from standard reporting metrics alone. In an empirical two-model case study on 126 BraTS21 patients, we evaluate a high-performance pretrained SegResNet and a locally trained UNet with residual units (UNet-Res). MC dropout preserved segmentation accuracy (∣ΔDice∣<0.01) while achieving strong uncertainty-error alignment (AUROC for entropy (H) ≈0.97), indicating uncertainty correctly ranks erroneous voxels above correct ones. Entropy-based patient stratification identified a high-uncertainty subgroup with substantially lower segmentation performance (median whole-tumour Dice 0.835 vs. 0.925), supporting uncertainty as a practical triage signal. However, global alignment can mask important region-specific differences. Despite similar AUROC, UNet-Res exhibited near-zero enhancing tumour entropy (0.054) and Expected Calibration Error (ECE) of 0.915, with a Dice of only 0.714, indicating severely miscalibrated confidence on the most clinically critical sub-region, a failure mode invisible to standard Dice and AUROC reporting. These findings demonstrate that strong uncertainty-error alignment is necessary but insufficient for clinical safety: sub-region-specific calibration assessment must accompany AUROC evaluation when selecting models for clinical deployment.
Xin Ci Wong, Duygu Sarikaya, Kieran Zucker +2
Centre for Doctoral Training in AI for Medical Diagnosis and Care, School of Computing, University of Leeds · School of Computer Science, University of Leeds · Leeds Cancer Centre, St James’s University Hospital, Leeds, UK