arXiv:2602.10045cs.CVcs.LG2026-02被引 1

为实例分割生成可证明置信度的预测集,提升不确定性量化能力

Conformal Prediction Sets for Instance Segmentation

  • 基于共形预测构建动态置信集,覆盖真实掩码的概率有保证
  • 预测集大小随查询难度自适应变化,实测达到目标覆盖率
  • 首次在实例分割中实现结构化不确定性建模,适合高风险场景

当前实例分割模型虽平均性能优异,但缺乏可靠的不确定性量化:其输出未校准,也无法保证预测掩码与真实掩码接近。为此,本文提出一种共形预测算法,为图像中任意像素坐标生成包含多个实例预测的置信集,确保至少一个预测与真实物体掩码的交并比(IoU)较高,具有可证明的概率保证。该方法应用于农业田块划分、细胞分割和车辆检测任务。实验表明,预测集大小随查询难易程度自适应调整,且达到目标覆盖水平,优于基线方法(朴素最佳参数法与形态学膨胀法)。算法提供渐近和有限样本两种保证版本。本工作首次通过生成多样化的分割预测置信集,捕捉实例分割中的结构不确定性。

原文摘要 · Abstract (English)

Current instance segmentation models achieve high performance on average predictions, but lack principled uncertainty quantification: their outputs are not calibrated, and there is no guarantee that a predicted mask is close to the ground truth. To address this limitation, we introduce a conformal prediction algorithm to generate adaptive confidence sets for instance segmentation. Given an image and a pixel coordinate query, our algorithm generates a confidence set of instance predictions for that pixel, with a provable guarantee for the probability that at least one of the predictions has high Intersection-Over-Union (IoU) with the true object instance mask. We apply our algorithm to instance segmentation examples in agricultural field delineation, cell segmentation, and vehicle detection. Empirically, we find that our prediction sets vary in size based on query difficulty and attain the target coverage, outperforming baselines (naive best parameter and morphological dilation-based methods). We provide versions of the algorithm with asymptotic and finite sample guarantees. Our work is the first to capture structural uncertainty in instance segmentation by constructing confidence sets of diverse segmentation predictions.

实例分割共形预测不确定性量化

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