为医学分割的度量结果提供精准不确定性估计
COMPASS: Robust Feature Conformal Prediction for Medical Segmentation Metrics
- 通过扰动中间特征空间提升度量预测的置信区间精度
- 在4个任务中置信区间比传统方法更紧致,覆盖率更高
- 适合临床决策支持中的可靠性评估,尤其在数据分布变化时
在临床应用中,分割模型的价值常取决于衍生下游指标(如器官大小)的准确性,而非像素级分割掩码的精度。因此,对这些指标进行不确定性量化对决策至关重要。分位数预测(CP)是一种可提供严谨不确定性保证的框架,但直接对最终标量指标应用CP效率低下,因其将复杂的非线性分割到指标的转换过程视为黑箱。我们提出COMPASS,一种实用框架,通过利用深层神经网络的归纳偏置,生成针对图像分割模型的高效、基于度量的CP区间。COMPASS在模型表示空间中通过沿对目标度量最敏感的低维子空间扰动中间特征进行校准。我们证明,在可交换性假设下,COMPASS能实现有效的边际覆盖。实证表明,在四个医学图像分割任务中(皮肤病变与解剖结构面积估计),COMPASS产生的置信区间显著优于传统CP基线。此外,利用学习到的内部特征估计重要性权重,使COMPASS在协变量偏移下仍能恢复目标覆盖率。COMPASS为医学图像分割的实用化、基于度量的不确定性量化开辟了道路。
原文摘要 · Abstract (English)
In clinical applications, the utility of segmentation models is often based on the accuracy of derived downstream metrics such as organ size, rather than by the pixel-level accuracy of the segmentation masks themselves. Thus, uncertainty quantification for such metrics is crucial for decision-making. Conformal prediction (CP) is a popular framework to derive such principled uncertainty guarantees, but applying CP naively to the final scalar metric is inefficient because it treats the complex, non-linear segmentation-to-metric pipeline as a black box. We introduce COMPASS, a practical framework that generates efficient, metric-based CP intervals for image segmentation models by leveraging the inductive biases of their underlying deep neural networks. COMPASS performs calibration directly in the model's representation space by perturbing intermediate features along low-dimensional subspaces maximally sensitive to the target metric. We prove that COMPASS achieves valid marginal coverage under the assumption of exchangeability. Empirically, we demonstrate that COMPASS produces significantly tighter intervals than traditional CP baselines on four medical image segmentation tasks for area estimation of skin lesions and anatomical structures. Furthermore, we show that leveraging learned internal features to estimate importance weights allows COMPASS to also recover target coverage under covariate shifts. COMPASS paves the way for practical, metric-based uncertainty quantification for medical image segmentation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。