发现分割模型在边界正确时会错标语义,提出诊断与预警方法。
Right Regions, Wrong Labels: Semantic Label Flips in Segmentation under Correlation Shift
- 设计新诊断指标Flip,量化正确边界但标签错误的像素
- 相关性越强,罕见场景中标签错换现象越严重,差距扩大27%以上
- 无需真实标签,用熵值可提前识别易出错的预测案例
机器学习模型的鲁棒性可能受输入数据中非因果特征与标签之间的虚假相关性影响。现有方法常通过训练时强关联某非因果线索与标签,测试时移除该关联来检验模型可靠性,此思路在分类任务中成熟,但在语义分割中尚不明确。我们发现,模型可能在保持较高重叠度的同时,将一个合理的前景类别错误地替换为另一个,即使物体边界基本正确。为此,我们提出简单诊断指标Flip,用于统计真实前景像素被错误分配前景身份但仍被预测为前景的频率。在类别与场景相关性训练下,增加相关性会显著拉大常见与罕见测试条件间的差距,并提升反事实组内的内部标签错换比例。结果表明,评估分割鲁棒性应超越重叠率,需分解前景错误为:正确像素、标签错换像素、误判为背景像素三类。此外,我们提出一种基于熵的、无需真实标签的‘翻转风险’评分,在推理时可有效识别高翻转风险样本。代码已开源:https://github.com/acharaakshit/label-flips。
原文摘要 · Abstract (English)
The robustness of machine learning models can be compromised by spurious correlations between non-causal features in the input data and target labels. A common way to test for such correlations is to train on data where the label is strongly tied to some non-causal cue, then evaluate on examples where that tie no longer holds. This idea is well established for classification tasks, but for semantic segmentation the specific failure modes are not well understood. We show that a model may achieve reasonable overlap while assigning the wrong semantic label, swapping one plausible foreground class for another, even when object boundaries are largely correct. We focus on this semantic label-flip behaviour and quantify it with a simple diagnostic (Flip) that counts how often ground truth foreground pixels are assigned the wrong foreground identity while remaining predicted as foreground. In a setting where category and scene are correlated during training, increasing the correlation consistently widens the gap between common and rare test conditions and increases these within-object label swaps on counterfactual groups. Overall, our results motivate assessing segmentation robustness under distribution shift beyond overlap by decomposing foreground errors into correct pixels, flipped-identity pixels, and missed-to-background pixels. We also propose an entropy-based, ground truth label-free `flip-risk' score, which is computed from foreground identity uncertainty, and show that it can flag flip-prone cases at inference time. Code is available at https://github.com/acharaakshit/label-flips.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。