Right Regions, Wrong Labels: Semantic Label Flips in Segmentation under Correlation Shift
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 24 of its 33 references that are in the map. First posted on arXiv 14 Apr 2026.
Builds on
24 of 33 references.
References · 24
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2014Microsoft COCO: Common Objects in Contextcited by 17,608
- 2021SegFormer: Simple and Efficient Design for Semantic Segmentation with Transformerscited by 2,940
- 2019CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Featurescited by 2,884
- 2016V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentationcited by 2,292
- 2020Shortcut Learning in Deep Neural Networkscited by 1,969
- 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,247
- 2015Bayesian SegNet: Model Uncertainty in Deep Convolutional Encoder-Decoder Architectures for Scene Understandingcited by 218
- 2022MetaShift: A Dataset of Datasets for Evaluating Contextual Distribution Shifts and Training Conflictscited by 81
- 2021Fairness in Cardiac MR Image Analysis: An Investigation of Bias Due to Data Imbalance in Deep Learning Based Segmentationcited by 57
12 more
- 2018Not Using the Car to See the Sidewalk: Quantifying and Controlling the Effects of Context in Classification and Segmentationcited by 49
- 2019Grid Saliency for Context Explanations of Semantic Segmentationcited by 32
- 2022RBC: Rectifying the Biased Context in Continual Semantic Segmentationcited by 27
- 2022On Calibrating Semantic Segmentation Models: Analyses and An Algorithmcited by 25
- 2022A Study of Demographic Bias in CNN-based Brain MR Segmentationcited by 16
- 2024Shortcut Learning in Medical Image Segmentationcited by 16
- 2024Separate and Conquer: Decoupling Co-occurrence via Decomposition and Representation for Weakly Supervised Semantic Segmentationcited by 14
- 2023MARS: Model-agnostic Biased Object Removal without Additional Supervision for Weakly-Supervised Semantic Segmentationcited by 13
- 2025Are We Done with Object-Centric Learning?cited by 11
- 2021SegMix: Co-occurrence Driven Mixup for Semantic Segmentation and Adversarial Robustnesscited by 7
- 2024Learning to Detour: Shortcut Mitigating Augmentation for Weakly Supervised Semantic Segmentationcited by 4
- 2025Mitigating Spurious Correlations in Weakly Supervised Semantic Segmentation via Cross-architecture Consistency Regularizationcited by 1
Built on by
Not cited yet.