English

Pixel-level Counterfactual Contrastive Learning for Medical Image Segmentation

Computer Vision and Pattern Recognition 2026-03-19 v1 Machine Learning

Abstract

Image segmentation relies on large annotated datasets, which are expensive and slow to produce. Silver-standard (AI-generated) labels are easier to obtain, but they risk introducing bias. Self-supervised learning, needing only images, has become key for pre-training. Recent work combining contrastive learning with counterfactual generation improves representation learning for classification but does not readily extend to pixel-level tasks. We propose a pipeline combining counterfactual generation with dense contrastive learning via Dual-View (DVD-CL) and Multi-View (MVD-CL) methods, along with supervised variants that utilize available silver-standard annotations. A new visualisation algorithm, the Color-coded High Resolution Overlay map (CHRO-map) is also introduced. Experiments show annotation-free DVD-CL outperforms other dense contrastive learning methods, while supervised variants using silver-standard labels outperform training on the silver-standard labeled data directly, achieving \sim94% DSC on challenging data. These results highlight that pixel-level contrastive learning, enhanced by counterfactuals and silver-standard annotations, improves robustness to acquisition and pathological variations.

Keywords

Cite

@article{arxiv.2603.17110,
  title  = {Pixel-level Counterfactual Contrastive Learning for Medical Image Segmentation},
  author = {Marceau Lafargue-Hauret and Raghav Mehta and Fabio De Sousa Ribeiro and Mélanie Roschewitz and Ben Glocker},
  journal= {arXiv preprint arXiv:2603.17110},
  year   = {2026}
}

Comments

Accepted at ISBI-2026 (oral presentation)

R2 v1 2026-07-01T11:25:08.347Z