English

Unsupervised Source-Free Ranking of Biomedical Segmentation Models Under Distribution Shift

Computer Vision and Pattern Recognition 2026-04-09 v4

Abstract

Model reuse offers a solution to the challenges of segmentation in biomedical imaging, where high data annotation costs remain a major bottleneck for deep learning. However, although many pretrained models are released through challenges, model zoos, and repositories, selecting the most suitable model for a new dataset remains difficult due to the lack of reliable model ranking methods. We introduce the first black-box-compatible framework for unsupervised and source-free ranking of semantic and instance segmentation models based on the consistency of predictions under perturbations. While ranking methods have been studied for classification and a few segmentation-related approaches exist, most target related tasks such as transferability estimation or model validation and typically rely on labelled data, feature-space access, or specific training assumptions. In contrast, our method directly addresses the repository setting and applies to both semantic and instance segmentation, for zero-shot reuse or after unsupervised domain adaptation. We evaluate the approach across a wide range of biomedical segmentation tasks in both 2D and 3D imaging, showing that our estimated rankings strongly correlate with true target-domain model performance rankings.

Keywords

Cite

@article{arxiv.2503.00450,
  title  = {Unsupervised Source-Free Ranking of Biomedical Segmentation Models Under Distribution Shift},
  author = {Joshua Talks and Kevin Marchesini and Luca Lumetti and Federico Bolelli and Anna Kreshuk},
  journal= {arXiv preprint arXiv:2503.00450},
  year   = {2026}
}

Comments

15 pages, 4 figures

R2 v1 2026-06-28T22:03:00.815Z