Evaluating Self-Supervised Learning in Medical Imaging: A Benchmark for Robustness, Generalizability, and Multi-Domain Impact
Not on the map. Why?
Abstract (arXiv)
Self-supervised learning (SSL) has emerged as a promising paradigm in medical imaging, addressing the chronic challenge of limited labeled data in healthcare settings. While SSL has shown impressive results, existing studies in the medical domain are often limited in scope, focusing on specific datasets or modalities, or evaluating only isolated aspects of model performance. This fragmented evaluation approach poses a significant challenge, as models deployed in critical medical settings must not only achieve high accuracy but also demonstrate robust performance and generalizability across diverse datasets and varying conditions. To address this gap, we present a comprehensive evaluation of SSL methods within the medical domain, with a particular focus on robustness and generalizability. Using the MedMNIST dataset collection as a standardized benchmark, we evaluate 8 major SSL methods across 11 different medical datasets. Our study provides an in-depth analysis of model performance in both in-domain scenarios and the detection of out-of-distribution (OOD) samples, while exploring the effect of various initialization strategies, model architectures, and multi-domain pre-training. We further assess the generalizability of SSL methods through cross-dataset evaluations and the in-domain performance with varying label proportions (1%, 10%, and 100%) to simulate real-world scenarios with limited supervision. We hope this comprehensive benchmark helps practitioners and researchers make more informed decisions when applying SSL methods to medical applications.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 47 of its 76 references that are in the map. First posted on arXiv 26 Dec 2024.
Builds on
47 of 76 references.
References · 47
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,482
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,712
- 2016A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networkscited by 2,362
- 2020FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidencecited by 2,080
- 2021An Empirical Study of Training Self-Supervised Vision Transformerscited by 1,610
- 2020Don't Stop Pretraining: Adapt Language Models to Domains and Taskscited by 1,548
- 2018A Simple Unified Framework for Detecting Out-of-Distribution Samples and Adversarial Attackscited by 1,472
- 2020Energy-based Out-of-distribution Detectioncited by 1,296
- 2021MedMNIST v2 -- A large-scale lightweight benchmark for 2D and 3D biomedical image classificationcited by 682
35 more
- 2019Skin Lesion Analysis Toward Melanoma Detection 2018: A Challenge Hosted by the International Skin Imaging Collaboration (ISIC)cited by 657
- 2019The Liver Tumor Segmentation Benchmark (LiTS)cited by 504
- 2021VATT: Transformers for Multimodal Self-Supervised Learning from Raw Video, Audio and Textcited by 427
- 2021With a Little Help from My Friends: Nearest-Neighbor Contrastive Learning of Visual Representationscited by 324
- 2021SSD: A Unified Framework for Self-Supervised Outlier Detectioncited by 264
- 2017Convolutional Neural Networks for Medical Image Analysis: Full Training or Fine Tuning?cited by 220
- 2023A Survey on Self-supervised Learning: Algorithms, Applications, and Future Trendscited by 205
- 2019Scaling and Benchmarking Self-Supervised Visual Representation Learningcited by 201
- 2022Benchmarking Self-Supervised Learning on Diverse Pathology Datasetscited by 148
- 2021Self-Supervised Representation Learning: Introduction, Advances and Challengescited by 147
- 2020Domain Generalization for Medical Imaging Classification with Linear-Dependency Regularizationcited by 122
- 2021Delving Deep into the Generalization of Vision Transformers under Distribution Shiftscited by 85
- 2022Delving into Masked Autoencoders for Multi-Label Thorax Disease Classificationcited by 65
- 2021Is it Time to Replace CNNs with Transformers for Medical Images?cited by 60
- 2020A scoping review of transfer learning research on medical image analysis using ImageNetcited by 54
- 2023Rethinking Out-of-distribution (OOD) Detection: Masked Image Modeling is All You Needcited by 46
- 2023AIROGS: Artificial Intelligence for RObust Glaucoma Screening Challengecited by 35
- 2021Confidence-based Out-of-Distribution Detection: A Comparative Study and Analysiscited by 34
- 2021A Systematic Benchmarking Analysis of Transfer Learning for Medical Image Analysiscited by 31
- 2023A framework for benchmarking class-out-of-distribution detection and its application to ImageNetcited by 31
- 2023Understanding Why ViT Trains Badly on Small Datasets: An Intuitive Perspectivecited by 28
- 2024MedIAnomaly: A comparative study of anomaly detection in medical imagescited by 25
- 2022No Reason for No Supervision: Improved Generalization in Supervised Modelscited by 23
- 2024A Closer Look at Benchmarking Self-Supervised Pre-training with Image Classificationcited by 19
- 2022Is Self-Supervised Learning More Robust Than Supervised Learning?cited by 17
- 2022RODD: A Self-Supervised Approach for Robust Out-of-Distribution Detectioncited by 13
- 2024A comprehensive and easy-to-use multi-domain multi-task medical imaging meta-datasetcited by 12
- 2022CADet: Fully Self-Supervised Out-Of-Distribution Detection With Contrastive Learningcited by 11
- 2023Systematic comparison of semi-supervised and self-supervised learning for medical image classificationcited by 8
- 2024Prompt-driven Latent Domain Generalization for Medical Image Classificationcited by 8
- 2021Tasting the cake: evaluating self-supervised generalization on out-of-distribution multimodal MRI datacited by 5
- 2024Using Self-supervised Learning Can Improve Model Fairnesscited by 5
- 2021Label scarcity in biomedicine: Data-rich latent factor discovery enhances phenotype predictioncited by 2
- 2022An unobtrusive quality supervision approach for medical image annotationcited by 2
- 2023Multi-domain improves out-of-distribution and data-limited scenarios for medical image analysiscited by 1
Built on by
Citing papers per year: 2025: 1; 2026: 1.
2 papers.