提升医学图像分割的不确定性量化,让误判率更稳定可靠。
Conditional Conformal Risk Adaptation
- 设计自适应预测集得分函数,动态调整风险控制
- 三种方法均实现有效边缘风险控制,条件风险更均匀
- 适合高风险医疗场景的个性化分割应用
不确定性量化在图像分割中愈发重要,尤其在医疗影像等高风险场景。传统符合风险控制虽可处理多种损失函数(如漏检率),但在分割任务中常导致条件风险控制不佳:部分图像漏检率极高,另一些则几乎为零。本文提出符合风险自适应(CRA),引入新得分函数生成自适应预测集,显著改善条件风险控制。建立新的理论框架,通过加权分位数揭示符合风险控制与符合预测间的深层联系,适用于任意得分函数。针对分割模型概率校准不足的问题,提出专用概率校准框架,提升像素级包含估计可靠性。基于校准概率,提出校准符合风险自适应(CCRA)及其分层变体(CCRA-S),按图像特征分组并应用特定阈值以进一步增强条件风险控制。在息肉分割实验中,CRA、CCRA 和 CCRA-S 均实现有效的边缘风险控制,并在不同图像间提供更一致的条件风险表现,为高风险与个性化分割提供系统性不确定性量化方案。
原文摘要 · Abstract (English)
Uncertainty quantification is becoming increasingly important in image segmentation, especially for high-stakes applications like medical imaging. While conformal risk control generalizes conformal prediction beyond standard miscoverage to handle various loss functions such as false negative rate, its application to segmentation often yields inadequate conditional risk control: some images experience very high false negative rates while others have negligibly small ones. We develop Conformal Risk Adaptation (CRA), which introduces a new score function for creating adaptive prediction sets that significantly improve conditional risk control for segmentation tasks. We establish a novel theoretical framework that demonstrates a fundamental connection between conformal risk control and conformal prediction through a weighted quantile approach, applicable to any score function. To address the challenge of poorly calibrated probabilities in segmentation models, we introduce a specialized probability calibration framework that enhances the reliability of pixel-wise inclusion estimates. Using these calibrated probabilities, we propose Calibrated Conformal Risk Adaptation (CCRA) and a stratified variant (CCRA-S) that partitions images based on their characteristics and applies group-specific thresholds to further enhance conditional risk control. Our experiments on polyp segmentation demonstrate that all three methods (CRA, CCRA, and CCRA-S) provide valid marginal risk control and deliver more consistent conditional risk control across diverse images compared to standard approaches, offering a principled approach to uncertainty quantification that is particularly valuable for high-stakes and personalized segmentation applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。