mlsanityData to 14 Oct 2026

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

arXiv 2503.00450 ↗First posted 1 Mar 2025cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

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.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 49 of its 100 references that are in the map. First posted on arXiv 1 Mar 2025.

Builds on

49 of 100 references.

References · 49

  1. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  2. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
  3. 2014Microsoft COCO: Common Objects in Contextcited by 17,608
  4. 2016Densely Connected Convolutional Networkscited by 6,720
  5. 2018MobileNetV2: Inverted Residuals and Linear Bottleneckscited by 4,795
  6. 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
  7. 2019Can You Trust Your Model's Uncertainty? Evaluating Predictive Uncertainty Under Dataset Shiftcited by 1,287
  8. 2024VMamba: Visual State Space Modelcited by 969
  9. 2021UNETR: Transformers for 3D Medical Image Segmentationcited by 959
  10. 2022Swin UNETR: Swin Transformers for Semantic Segmentation of Brain Tumors in MRI Imagescited by 830
  11. 2018Deep Visual Domain Adaptation: A Surveycited by 635
  12. 2024U-Mamba: Enhancing Long-range Dependency for Biomedical Image Segmentationcited by 419
37 more
  1. 2020Semi-Supervised Semantic Segmentation with Cross-Consistency Trainingcited by 376
  2. 2018Aleatoric uncertainty estimation with test-time augmentation for medical image segmentation with convolutional neural networkscited by 280
  3. 2018Characterizing and Avoiding Negative Transfercited by 229
  4. 2014Efficient Object Localization Using Convolutional Networkscited by 227
  5. 2020LEEP: A New Measure to Evaluate Transferability of Learned Representationscited by 188
  6. 2021LogME: Practical Assessment of Pre-trained Models for Transfer Learningcited by 172
  7. 2019Confidence Calibration and Predictive Uncertainty Estimation for Deep Medical Image Segmentationcited by 163
  8. 2020Understanding Self-Training for Gradual Domain Adaptationcited by 149
  9. 2022Leveraging Unlabeled Data to Predict Out-of-Distribution Performancecited by 145
  10. 2019A General Framework for Uncertainty Estimation in Deep Learningcited by 119
  11. 2018Predicting the Generalization Gap in Deep Networks with Margin Distributionscited by 113
  12. 2022An Information-Theoretic Approach to Transferability in Task Transfer Learningcited by 81
  13. 2017Minimal-Entropy Correlation Alignment for Unsupervised Deep Domain Adaptationcited by 71
  14. 2017Superhuman Accuracy on the SNEMI3D Connectomics Challengecited by 64
  15. 2021Transferability Estimation using Bhattacharyya Class Separabilitycited by 54
  16. 2020Ranking Neural Checkpointscited by 51
  17. 2018Leveraging Uncertainty Estimates for Predicting Segmentation Qualitycited by 44
  18. 2021Tune it the Right Way: Unsupervised Validation of Domain Adaptation via Soft Neighborhood Densitycited by 44
  19. 2018Bayesian QuickNAT: Model Uncertainty in Deep Whole-Brain Segmentation for Structure-wise Quality Controlcited by 39
  20. 2021Supervised Transfer Learning at Scale for Medical Imagingcited by 38
  21. 2017Reverse Classification Accuracy: Predicting Segmentation Performance in the Absence of Ground Truthcited by 35
  22. 2022How stable are Transferability Metrics evaluations?cited by 30
  23. 2019Training-Free Uncertainty Estimation for Dense Regression: Sensitivity as a Surrogatecited by 28
  24. 2023Confidence and Dispersity Speak: Characterising Prediction Matrix for Unsupervised Accuracy Estimationcited by 22
  25. 2020Which Model to Transfer? Finding the Needle in the Growing Haystackcited by 21
  26. 2021Predicting Deep Neural Network Generalization with Perturbation Response Curvescited by 14
  27. 2023Dropout Injection at Test Time for Post Hoc Uncertainty Quantification in Neural Networkscited by 14
  28. 2023Can We Evaluate Domain Adaptation Models Without Target-Domain Labels?cited by 14
  29. 2021Robustness to Augmentations as a Generalization metriccited by 13
  30. 2024Which Model to Transfer? A Survey on Transferability Estimationcited by 13
  31. 2024Comparative Benchmarking of Failure Detection Methods in Medical Image Segmentation: Unveiling the Role of Confidence Aggregationcited by 12
  32. 2023Pick the Best Pre-trained Model: Towards Transferability Estimation for Medical Image Segmentationcited by 11
  33. 2024Epistemic Uncertainty Quantification For Pre-trained Neural Networkcited by 10
  34. 2024Estimating Model Performance Under Covariate Shift Without Labelscited by 9
  35. 2019GASP, a generalized framework for agglomerative clustering of signed graphs and its application to Instance Segmentationcited by 8
  36. 2024Confidence-based Estimators for Predictive Performance in Model Monitoringcited by 8
  37. 2023The Performance of Transferability Metrics does not Translate to Medical Taskscited by 7

Built on by

Not cited yet.