用可证明的统计方法控制图像分割误报,保障医疗决策安全
Controlling False Positives in Image Segmentation via Conformal Prediction
- 通过调整阈值或形态学腐蚀生成嵌套置信掩码
- 在肠息肉数据集上实现目标误报率控制,实测验证有效
- 无需重训练,适用于任何预训练模型,适合高风险医疗场景
可靠的语义分割对临床决策至关重要,但深度模型通常无法提供误差的明确统计保证。本文提出一种简单的后处理框架,构建无需分布假设、基于图像级别的误报控制置信掩码。给定任意预训练分割模型,通过提高分数阈值或应用形态学腐蚀,生成一系列收缩掩码。利用标注校准集通过置信预测选择单一收缩参数,确保在与校准数据同分布的新图像上,置信掩码中保留的误报比例以高概率低于用户指定容差。该方法模型无关,无需重新训练,且在有限样本下提供严格保证。在肠息肉分割基准测试中,实验验证了目标水平的实证有效性。本框架使在过度分割可能带来临床后果的场景中实现风险感知的实用分割成为可能。代码见 https://github.com/deel-ai-papers/conseco。
原文摘要 · Abstract (English)
Reliable semantic segmentation is essential for clinical decision making, yet deep models rarely provide explicit statistical guarantees on their errors. We introduce a simple post-hoc framework that constructs confidence masks with distribution-free, image-level control of false-positive predictions. Given any pretrained segmentation model, we define a nested family of shrunken masks obtained either by increasing the score threshold or by applying morphological erosion. A labeled calibration set is used to select a single shrink parameter via conformal prediction, ensuring that, for new images that are exchangeable with the calibration data, the proportion of false positives retained in the confidence mask stays below a user-specified tolerance with high probability. The method is model-agnostic, requires no retraining, and provides finite-sample guarantees regardless of the underlying predictor. Experiments on a polyp-segmentation benchmark demonstrate target-level empirical validity. Our framework enables practical, risk-aware segmentation in settings where over-segmentation can have clinical consequences. Code at https://github.com/deel-ai-papers/conseco.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。