Self-supervised pretraining for dermoscopic lesion segmentation with limited labels: A controlled comparison of initialisation strategies
Hamda, H. K. H.; Mustafa, S.; Khalili Moghaddam, G.
Show abstract
Background: Self-supervised learning may improve segmentation when labelled data are limited, but its benefit may depend on model and initialisation strategy. This study evaluated in-domain masked-reconstruction pretraining across limited annotation budgets. Methods: The International Skin Imaging Collaboration (ISIC) 2016 development set was partitioned into 765 images for self-supervised pretraining, 94 for fine-tuning and 41 for validation. Five nested annotation budgets (12-94 images) and five seeds were evaluated on the 379-image test set. A custom U-Net compared random and in-domain self-supervised initialisation; a ResNet-34 U-Net compared random, ImageNet, in-domain self-supervised and sequential ImageNet-to-self-supervised initialisation. Models trained with 94 labelled images were evaluated without adaptation on ISIC 2017. A separate experiment examined restoration-based pretraining. Results: In the custom U-Net, self-supervised pretraining achieved higher mean Dice than random initialisation at all five budgets, including 0.664 versus 0.615 with 12 labels and 0.811 versus 0.768 with 94 labels. Seed-level 95% confidence intervals for paired differences excluded zero at 47 and 94 labels but not at the other budgets. With 47 labels, the self-supervised model achieved mean Dice of 0.789, compared with 0.768 for random initialisation with 94 labels. In ResNet-34, ImageNet initialisation achieved higher mean Dice than in-domain self-supervised pretraining at all five budgets, and adding self-supervised pretraining after ImageNet reduced mean Dice at four of five budgets. All six 94-label configurations achieved lower mean Dice on ISIC 2017 than ISIC 2016. In the restoration experiment, mean Dice was 0.686 with random initialisation and 0.638-0.642 following pretraining. Conclusions: In-domain masked-reconstruction pretraining improved mean Dice relative to random initialisation in the custom U-Net, but not consistently in the ResNet-34 U-Net, where ImageNet initialisation performed better at every budget. Adding self-supervised pretraining after ImageNet provided no consistent gain. These findings are specific to the dermoscopic lesion-segmentation experiments and do not establish melanoma-specific diagnostic or clinical performance.
Matching journals
The top 7 journals account for 50% of the predicted probability mass.
Similar papers in this journal
Similar papers in this journal
Similar papers in this journal
- Deep neural frameworks improve the accuracy of general practitioners in the classification of pigmented skin lesions 93%
- A Machine Learning Ensemble Based on Radiomics to Predict BI-RADS Category and Reduce the Biopsy Rate of Ultrasound-Detected Suspicious Breast Masses 91%
- Deep learning models for poorly differentiated colorectal adenocarcinoma classification in whole slide images using transfer learning 88%
Similar papers in this journal
- Enhancing Semantic Segmentation in Chest X-Ray Images through Image Preprocessing: ps-KDE for Pixel-wise Substitution by Kernel Density Estimation 91%
- Fibre tracing in biomedical images: An objective comparison between seven algorithms 91%
- Classification performance bias between training and test sets in a limited mammography dataset 91%
Similar papers in this journal
- An Inherently Interpretable AI model improves Screening Speed and Accuracy for Early Diabetic Retinopathy 92%
- Self-supervised contrastive learning improves machine learning discrimination of full thickness macular holes from epiretinal membranes in retinal OCT scans 92%
- Assessing generalizability of an AI-based visual test for cervical cancer screening 92%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.