Robust and Efficient Medical Imaging with Self-Supervision
Not on the map. Why?
Abstract (arXiv)
Recent progress in Medical Artificial Intelligence (AI) has delivered systems that can reach clinical expert level performance. However, such systems tend to demonstrate sub-optimal "out-of-distribution" performance when evaluated in clinical settings different from the training environment. A common mitigation strategy is to develop separate systems for each clinical setting using site-specific data [1]. However, this quickly becomes impractical as medical data is time-consuming to acquire and expensive to annotate [2]. Thus, the problem of "data-efficient generalization" presents an ongoing difficulty for Medical AI development. Although progress in representation learning shows promise, their benefits have not been rigorously studied, specifically for out-of-distribution settings. To meet these challenges, we present REMEDIS, a unified representation learning strategy to improve robustness and data-efficiency of medical imaging AI. REMEDIS uses a generic combination of large-scale supervised transfer learning with self-supervised learning and requires little task-specific customization. We study a diverse range of medical imaging tasks and simulate three realistic application scenarios using retrospective data. REMEDIS exhibits significantly improved in-distribution performance with up to 11.5% relative improvement in diagnostic accuracy over a strong supervised baseline. More importantly, our strategy leads to strong data-efficient generalization of medical imaging AI, matching strong supervised baselines using between 1% to 33% of retraining data across tasks. These results suggest that REMEDIS can significantly accelerate the life-cycle of medical imaging AI development thereby presenting an important step forward for medical imaging AI to deliver broad impact.
Key facts. Cited by 40 papers in the map (data through 7 Oct 2026). Builds on 79 of its 136 references that are in the map. First posted on arXiv 19 May 2022.
Builds on
79 of 136 references.
References · 79
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 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
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,482
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,712
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,560
- 2020Exploring Simple Siamese Representation Learningcited by 2,853
- 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,845
- 2018Unsupervised Feature Learning via Non-Parametric Instance-level Discriminationcited by 1,838
38 more
- 2018Unsupervised Representation Learning by Predicting Image Rotationscited by 1,586
- 2016Context Encoders: Feature Learning by Inpaintingcited by 1,424
- 2020Big Self-Supervised Models are Strong Semi-Supervised Learnerscited by 1,375
- 2018Attention-based Deep Multiple Instance Learningcited by 1,077
- 2015Unsupervised Visual Representation Learning by Context Predictioncited by 1,062
- 2019EfficientDet: Scalable and Efficient Object Detectioncited by 1,018
- 2017Revisiting Unreasonable Effectiveness of Data in Deep Learning Eracited by 916
- 2019Self-Supervised Learning of Pretext-Invariant Representationscited by 806
- 2021Robust fine-tuning of zero-shot modelscited by 785
- 2021Generalizing to Unseen Domains: A Survey on Domain Generalizationcited by 762
- 2021Domain Generalization: A Surveycited by 760
- 2019Learning Representations by Maximizing Mutual Information Across Viewscited by 733
- 2018Deep Visual Domain Adaptation: A Surveycited by 634
- 2013Domain Generalization via Invariant Feature Representationcited by 622
- 2018Exploring the Limits of Weakly Supervised Pretrainingcited by 604
- 2017Large Batch Training of Convolutional Networkscited by 529
- 2020Underspecification Presents Challenges for Credibility in Modern Machine Learningcited by 488
- 2018Confounding variables can degrade generalization performance of radiological deep learning modelscited by 374
- 2019Using Pre-Training Can Improve Model Robustness and Uncertaintycited by 373
- 2019Transfusion: Understanding Transfer Learning for Medical Imagingcited by 366
- 2020What is being transferred in transfer learning?cited by 365
- 2017Unified Deep Supervised Domain Adaptation and Generalizationcited by 322
- 2021Big Self-Supervised Models Advance Medical Image Classificationcited by 320
- 2020Contrastive learning of global and local features for medical image segmentation with limited annotationscited by 309
- 2020Pretrained Transformers Improve Out-of-Distribution Robustnesscited by 293
- 2020Do Wide and Deep Networks Learn the Same Things? Uncovering How Neural Network Representations Vary with Width and Depthcited by 241
- 2019Unsupervised Embedding Learning via Invariant and Spreading Instance Featurecited by 231
- 2020A review of deep learning in medical imaging: Imaging traits, technology trends, case studies with progress highlights, and future promisescited by 230
- 2018Characterizing and Avoiding Negative Transfercited by 226
- 2020How Well Do Self-Supervised Models Transfer?cited by 213
- 2021Towards Causal Representation Learningcited by 207
- 2017Multi-task Self-Supervised Visual Learningcited by 200
- 2020Representation Learning via Invariant Causal Mechanismscited by 197
- 2021Self-supervised Pretraining of Visual Features in the Wildcited by 197
- 2016What makes ImageNet good for transfer learning?cited by 192
- 2020On the Theory of Transfer Learning: The Importance of Task Diversitycited by 186
- 2020Contrastive Training for Improved Out-of-Distribution Detectioncited by 186
- 2017Colorization as a Proxy Task for Visual Understandingcited by 175
50 most cited of 79 All 79 (JSON)
Built on by
Citing papers per year: 2022: 9; 2023: 14; 2024: 11; 2025: 3; 2026: 3.
40 papers.
Sort:
By area:
- 2023Generative models improve fairness of medical classifiers under distribution shiftscited by 90
- 2023MOTOR: A Time-To-Event Foundation Model For Structured Medical Recordscited by 75
- 2022MEDFAIR: Benchmarking Fairness for Medical Imagingcited by 66
- 2023Parameter-Efficient Fine-Tuning for Medical Image Analysis: The Missed Opportunitycited by 54
- 2024Towards a clinically accessible radiology foundation model: open-access and lightweight, with automated evaluationcited by 51
- 2022SVL-Adapter: Self-Supervised Adapter for Vision-Language Pretrained Modelscited by 34
- 2024How to build the best medical image segmentation algorithm using foundation models: a comprehensive empirical study with Segment Anything Modelcited by 32
- 2024Review of multimodal machine learning approaches in healthcarecited by 30
- 2022Risk of Bias in Chest Radiography Deep Learning Foundation Modelscited by 26
- 2022How Robust is Unsupervised Representation Learning to Distribution Shift?cited by 24
- 2023PECoP: Parameter Efficient Continual Pretraining for Action Quality Assessmentcited by 19
- 2023Exploring Image Augmentations for Siamese Representation Learning with Chest X-Rayscited by 12
28 more
- 2022Self-Supervised Learning with Limited Labeled Data for Prostate Cancer Detection in High Frequency Ultrasoundcited by 7
- 2023Large-scale pretraining on pathological images for fine-tuning of small pathological benchmarkscited by 7
- 2023Evaluating AI systems under uncertain ground truth: a case study in dermatologycited by 7
- 2024Time-to-Event Pretraining for 3D Medical Imagingcited by 7
- 2023Video Pretraining Advances 3D Deep Learning on Chest CT Taskscited by 4
- 2024Learning Low-Rank Feature for Thorax Disease Classificationcited by 4
- 2023Cross-Shaped Windows Transformer with Self-supervised Pretraining for Clinically Significant Prostate Cancer Detection in Bi-parametric MRIcited by 3
- 2024ConPro: Learning Severity Representation for Medical Images using Contrastive Learning and Preference Optimizationcited by 3
- 2025Towards disentangling the contributions of articulation and acoustics in multimodal phoneme recognitioncited by 3
- 2026MLLM-HWSI: A Multimodal Large Language Model for Hierarchical Whole Slide Image Understandingcited by 3
- 2022Federated Training of Dual Encoding Models on Small Non-IID Client Datasetscited by 2
- 2023BenchMD: A Benchmark for Unified Learning on Medical Images and Sensorscited by 2
- 2023FBA-Net: Foreground and Background Aware Contrastive Learning for Semi-Supervised Atrium Segmentationcited by 2
- 2023Revisiting Fine-Tuning Strategies for Self-supervised Medical Imaging Analysiscited by 2
- 2024Embedding-based Multimodal Learning on Pan-Squamous Cell Carcinomas for Improved Survival Outcomescited by 2
- 2022Robust Alzheimer's Progression Modeling using Cross-Domain Self-Supervised Deep Learningcited by 1
- 2023MedLSAM: Localize and Segment Anything Model for 3D CT Imagescited by 1
- 2026From pre-training to downstream performance: Does domain-specific pre-training make sense?cited by 1
- 2022Joint Debiased Representation and Image Clustering Learning with Self-Supervisioncited by 0
- 2022Calibrating Ensembles for Scalable Uncertainty Quantification in Deep Learning-based Medical Segmentationcited by 0
- 2023How to Train Your CheXDragon: Training Chest X-Ray Models for Transfer to Novel Tasks and Healthcare Systemscited by 0
- 2024Exploiting Liver CT scans in Colorectal Carcinoma genomics mutation classificationcited by 0
- 2024Self-Supervised Learning for Building Robust Pediatric Chest X-ray Classification Modelscited by 0
- 2024Deep Learning for Classification of Inflammatory Bowel Disease Activity in Whole Slide Images of Colonic Histopathologycited by 0
- 2024Enhancing Skin Lesion Classification Generalization with Active Domain Adaptationcited by 0
- 2025The Next Layer: Augmenting Foundation Models with Structure-Preserving and Attention-Guided Learning for Local Patches to Global Context Awareness in Computational Pathologycited by 0
- 2025Harmonizing Generalization and Specialization: Uncertainty-Informed Collaborative Learning for Semi-supervised Medical Image Segmentationcited by 0
- 2026From Multi-Resolution Cells to Gigapixel Whole Slide Images Foundation Model for Computational Pathologycited by 0
- 2026From Multi-Resolution Cells to Gigapixel Whole Slide Images Foundation Model for Computational Pathology4 Aug 2026
- 2026From pre-training to downstream performance: Does domain-specific pre-training make sense?9 May 2026
- 2026MLLM-HWSI: A Multimodal Large Language Model for Hierarchical Whole Slide Image Understanding24 Mar 2026
- 2025Harmonizing Generalization and Specialization: Uncertainty-Informed Collaborative Learning for Semi-supervised Medical Image Segmentation15 Dec 2025
- 2025The Next Layer: Augmenting Foundation Models with Structure-Preserving and Attention-Guided Learning for Local Patches to Global Context Awareness in Computational Pathology27 Aug 2025
- 2025Towards disentangling the contributions of articulation and acoustics in multimodal phoneme recognition29 May 2025
- 2024Enhancing Skin Lesion Classification Generalization with Active Domain Adaptation1 Dec 2024
- 2024Time-to-Event Pretraining for 3D Medical Imaging14 Nov 2024
- 2024Deep Learning for Classification of Inflammatory Bowel Disease Activity in Whole Slide Images of Colonic Histopathology25 Oct 2024
- 2024Self-Supervised Learning for Building Robust Pediatric Chest X-ray Classification Models30 Aug 2024
- 2024Embedding-based Multimodal Learning on Pan-Squamous Cell Carcinomas for Improved Survival Outcomes11 Jun 2024
- 2024ConPro: Learning Severity Representation for Medical Images using Contrastive Learning and Preference Optimization29 Apr 2024
28 more
- 2024How to build the best medical image segmentation algorithm using foundation models: a comprehensive empirical study with Segment Anything Model15 Apr 2024
- 2024Towards a clinically accessible radiology foundation model: open-access and lightweight, with automated evaluation12 Mar 2024
- 2024Learning Low-Rank Feature for Thorax Disease Classification14 Feb 2024
- 2024Review of multimodal machine learning approaches in healthcare4 Feb 2024
- 2024Exploiting Liver CT scans in Colorectal Carcinoma genomics mutation classification25 Jan 2024
- 2023PECoP: Parameter Efficient Continual Pretraining for Action Quality Assessment11 Nov 2023
- 2023Revisiting Fine-Tuning Strategies for Self-supervised Medical Imaging Analysis20 Jul 2023
- 2023Evaluating AI systems under uncertain ground truth: a case study in dermatology5 Jul 2023
- 2023FBA-Net: Foreground and Background Aware Contrastive Learning for Semi-Supervised Atrium Segmentation27 Jun 2023
- 2023MedLSAM: Localize and Segment Anything Model for 3D CT Images26 Jun 2023
- 2023Parameter-Efficient Fine-Tuning for Medical Image Analysis: The Missed Opportunity14 May 2023
- 2023How to Train Your CheXDragon: Training Chest X-Ray Models for Transfer to Novel Tasks and Healthcare Systems13 May 2023
- 2023Cross-Shaped Windows Transformer with Self-supervised Pretraining for Clinically Significant Prostate Cancer Detection in Bi-parametric MRI30 Apr 2023
- 2023Generative models improve fairness of medical classifiers under distribution shifts18 Apr 2023
- 2023BenchMD: A Benchmark for Unified Learning on Medical Images and Sensors17 Apr 2023
- 2023Video Pretraining Advances 3D Deep Learning on Chest CT Tasks2 Apr 2023
- 2023Large-scale pretraining on pathological images for fine-tuning of small pathological benchmarks28 Mar 2023
- 2023Exploring Image Augmentations for Siamese Representation Learning with Chest X-Rays30 Jan 2023
- 2023MOTOR: A Time-To-Event Foundation Model For Structured Medical Records9 Jan 2023
- 2022Robust Alzheimer's Progression Modeling using Cross-Domain Self-Supervised Deep Learning15 Nov 2022
- 2022Self-Supervised Learning with Limited Labeled Data for Prostate Cancer Detection in High Frequency Ultrasound1 Nov 2022
- 2022SVL-Adapter: Self-Supervised Adapter for Vision-Language Pretrained Models7 Oct 2022
- 2022MEDFAIR: Benchmarking Fairness for Medical Imaging4 Oct 2022
- 2022Federated Training of Dual Encoding Models on Small Non-IID Client Datasets30 Sep 2022
- 2022Calibrating Ensembles for Scalable Uncertainty Quantification in Deep Learning-based Medical Segmentation20 Sep 2022
- 2022Joint Debiased Representation and Image Clustering Learning with Self-Supervision14 Sep 2022
- 2022Risk of Bias in Chest Radiography Deep Learning Foundation Models7 Sep 2022
- 2022How Robust is Unsupervised Representation Learning to Distribution Shift?17 Jun 2022
- 2023Generative models improve fairness of medical classifiers under distribution shiftscited by 90
- 2023MOTOR: A Time-To-Event Foundation Model For Structured Medical Recordscited by 75
- 2022MEDFAIR: Benchmarking Fairness for Medical Imagingcited by 66
- 2023Parameter-Efficient Fine-Tuning for Medical Image Analysis: The Missed Opportunitycited by 54
- 2024Towards a clinically accessible radiology foundation model: open-access and lightweight, with automated evaluationcited by 51
- 2024How to build the best medical image segmentation algorithm using foundation models: a comprehensive empirical study with Segment Anything Modelcited by 32
- 2024Review of multimodal machine learning approaches in healthcarecited by 30
- 2022Risk of Bias in Chest Radiography Deep Learning Foundation Modelscited by 26
- 2022How Robust is Unsupervised Representation Learning to Distribution Shift?cited by 24
- 2023PECoP: Parameter Efficient Continual Pretraining for Action Quality Assessmentcited by 19
- 2023Exploring Image Augmentations for Siamese Representation Learning with Chest X-Rayscited by 12
- 2022Self-Supervised Learning with Limited Labeled Data for Prostate Cancer Detection in High Frequency Ultrasoundcited by 7
12 most cited of 38 All 38 (JSON)
- 2026From Multi-Resolution Cells to Gigapixel Whole Slide Images Foundation Model for Computational Pathology4 Aug 2026
- 2026From pre-training to downstream performance: Does domain-specific pre-training make sense?9 May 2026
- 2025Harmonizing Generalization and Specialization: Uncertainty-Informed Collaborative Learning for Semi-supervised Medical Image Segmentation15 Dec 2025
- 2025The Next Layer: Augmenting Foundation Models with Structure-Preserving and Attention-Guided Learning for Local Patches to Global Context Awareness in Computational Pathology27 Aug 2025
- 2025Towards disentangling the contributions of articulation and acoustics in multimodal phoneme recognition29 May 2025
- 2024Enhancing Skin Lesion Classification Generalization with Active Domain Adaptation1 Dec 2024
- 2024Time-to-Event Pretraining for 3D Medical Imaging14 Nov 2024
- 2024Deep Learning for Classification of Inflammatory Bowel Disease Activity in Whole Slide Images of Colonic Histopathology25 Oct 2024
- 2024Self-Supervised Learning for Building Robust Pediatric Chest X-ray Classification Models30 Aug 2024
- 2024Embedding-based Multimodal Learning on Pan-Squamous Cell Carcinomas for Improved Survival Outcomes11 Jun 2024
- 2024ConPro: Learning Severity Representation for Medical Images using Contrastive Learning and Preference Optimization29 Apr 2024
- 2024How to build the best medical image segmentation algorithm using foundation models: a comprehensive empirical study with Segment Anything Model15 Apr 2024
12 newest of 38 All 38 (JSON)