arXiv:2604.15271cs.CVcs.AI2026-04中稿 · UNSURE@MICCAI2026

用扰动能量建模不确定性,单次前向传播实现医疗图像分割的可靠性评估。

SegWithU: Uncertainty as Perturbation Energy for Single-Forward-Pass Risk-Aware Medical Image Segmentation

  • 基于骨干网络中间特征,以秩一后验探针构建扰动能量模型。
  • 在三个数据集上分别达到0.9838/2.4885、0.9946/0.2660、0.9925/0.8193的AUROC/AURC。
  • 无需重复推理,适合临床部署的高可靠分割场景。

可靠的不确定性估计对医疗图像分割至关重要,其自动轮廓用于下游量化和临床决策支持。许多强效不确定性方法需多次推断,而高效的单次前向传播替代方案往往表现较弱或依赖严苛的特征空间假设。本文提出\textbf{SegWithU},一个后处理框架,将冻结的预训练分割骨干网络扩展为轻量级不确定性头。该方法利用骨干网络中间特征,通过秩一后验探针在紧凑探测空间中将不确定性建模为扰动能量,生成两个体素级不确定性图:面向校准的概率调节图与面向错误检测及选择性预测的排序图。在ACDC、BraTS2024和LiTS三个数据集上,SegWithU作为单次前向传播基线表现最强且最稳定,分别取得AUROC/AURC为0.9838/2.4885、0.9946/0.2660、0.9925/0.8193,同时保持分割质量。结果表明,基于扰动的不确定性建模是实现可靠性感知医疗分割的有效且可行路径。源代码见https://github.com/ProjectNeura/SegWithU。

原文摘要 · Abstract (English)

Reliable uncertainty estimation is critical for medical image segmentation, where automated contours feed downstream quantification and clinical decision support. Many strong uncertainty methods require repeated inference, while efficient single-forward-pass alternatives often provide weaker failure ranking or rely on restrictive feature-space assumptions. We present $\textbf{SegWithU}$, a post-hoc framework that augments a frozen pretrained segmentation backbone with a lightweight uncertainty head. SegWithU taps intermediate backbone features and models uncertainty as perturbation energy in a compact probe space using rank-1 posterior probes. It produces two voxel-wise uncertainty maps: a calibration-oriented map for probability tempering and a ranking-oriented map for error detection and selective prediction. Across ACDC, BraTS2024, and LiTS, SegWithU is the strongest and most consistent single-forward-pass baseline, achieving AUROC/AURC of $0.9838/2.4885$, $0.9946/0.2660$, and $0.9925/0.8193$, respectively, while preserving segmentation quality. These results suggest that perturbation-based uncertainty modeling is an effective and practical route to reliability-aware medical segmentation. Source code is available at https://github.com/ProjectNeura/SegWithU.

医学分割不确定性建模单次前向扰动能量

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