Knowing When Not to Predict: Self Supervised Learning and Abstention for Safer DR Screening
Not on the map. Why?
Abstract (arXiv)
Self-supervised learning (SSL) is now a standard way to pretrain medical image models, but performance is still mostly judged by downstream accuracy. For safety-critical screening tasks such as diabetic retinopathy grading, this is not enough: a model must also know when its predictions are unreliable and defer uncertain cases for clinical review. In this work, we examine how the length of SSL pretraining influences calibrated confidence and confidence-based abstention. We evaluate multiple SSL checkpoints under a fixed fine-tuning protocol and assess calibrated confidence, coverage, selective accuracy, and selective macro-F1. Across datasets and data regimes, SSL pretraining improves selective prediction compared to training from scratch. Unlike prior SSL studies that primarily evaluate downstream accuracy or AUROC, we analyze how SSL pretraining duration influences confidence behavior under calibrated confidence-based abstention. However, once accuracy saturates, selective performance can still change markedly across checkpoints, and longer pretraining does not consistently improve reliability. These results underscore the importance of abstention-aware evaluation and suggest that pretraining length should be treated as an important reliability-related design choice rather than only a computational detail. Code is available at GitHub.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 19 of its 30 references that are in the map. First posted on arXiv 18 May 2026.
Builds on
19 of 30 references.
References · 19
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
- 2017On Calibration of Modern Neural Networkscited by 4,960
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,750
- 2015Learning Deep Features for Discriminative Localizationcited by 2,436
- 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,838
- 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,057
- 2017Selective Classification for Deep Neural Networkscited by 852
- 2017In Defense of the Triplet Loss for Person Re-Identificationcited by 786
- 2019SelectiveNet: A Deep Neural Network with an Integrated Reject Optioncited by 350
7 more
- 2020Self supervised contrastive learning for digital histopathologycited by 146
- 2021Optimal strategies for reject option classifierscited by 61
- 2017Grader variability and the importance of reference standards for evaluating machine learning models for diabetic retinopathycited by 44
- 2021How Transferable Are Self-supervised Features in Medical Image Classification Tasks?cited by 33
- 2021A novel multiple instance learning framework for COVID-19 severity assessment via data augmentation and self-supervised learningcited by 19
- 2023How to Fix a Broken Confidence Estimator: Evaluating Post-hoc Methods for Selective Classification with Deep Neural Networkscited by 16
- 2023Fine-Grained Self-Supervised Learning with Jigsaw Puzzles for Medical Image Classificationcited by 7
Built on by
Not cited yet.