Espis Andrea, Marzi Chiara, Diciotti Stefano
Department of Electrical, Electronic, and Information Engineering "Guglielmo Marconi" - DEI, University of Bologna, Via dell'Università 50, 47521, Cesena, Italy.
Department of Statistics, Computer Science and Applications "Giuseppe Parenti", University of Florence, 50134, Florence, Italy.
Sci Rep. 2025 Sep 2;15(1):32345. doi: 10.1038/s41598-025-99000-0.
Self-supervised learning (SSL) in computer vision has shown its potential to reduce reliance on labeled data. However, most studies focused on balanced, large, broad-domain datasets like ImageNet, whereas, in real-world medical applications, dataset size is typically limited. This study compares the performance of SSL versus supervised learning (SL) on small, imbalanced medical imaging datasets. We experimented with four binary classification tasks: age prediction and diagnosis of Alzheimer's disease from brain magnetic resonance imaging scans, pneumonia from chest radiograms, and retinal diseases associated with choroidal neovascularization from optical coherence tomography with a mean size of training sets of 843 images, 771 images, 1,214 images, and 33,484 images, respectively. We tested various combinations of label availability and class frequency distribution, repeating the training with different random seeds to assess result uncertainty. In most experiments involving small training sets, SL outperformed the selected SSL paradigms, even when a limited portion of labeled data was available. Our findings highlight the importance of carefully selecting learning paradigms based on specific application requirements, which are influenced by factors such as training set size, label availability, and class frequency distribution.
Med Image Anal. 2024-5
Cochrane Database Syst Rev. 2024-10-17
IEEE Trans Pattern Anal Mach Intell. 2024-9-13
IEEE Trans Pattern Anal Mach Intell. 2024-12
Med Image Comput Comput Assist Interv. 2023-10
Med Image Comput Comput Assist Interv. 2023-10
Adv Neural Inf Process Syst. 2023-12
Comput Struct Biotechnol J. 2024-4-12
Adv Neural Inf Process Syst. 2022-12
Med Image Anal. 2023-10