Realistic Performance Estimation in Small-Cohort Deep Learning and Machine Learning: A Leakage-Controlled Codebase Applied to Meningioma Response Prediction After Gamma Knife Radiosurgery
Background/Objectives: Deep learning may support the prediction of meningioma response after Gamma Knife radiosurgery (GKRS), but small clinical cohorts increase the risk of overfitting and unstable performance. We developed leakage-controlled code adjusted for small clinical datasets. Methods: Pretreatment contrast-enhanced T1 MRI from 190 patients was analysed using one tumour-centred axial slice per patient. Response was defined as a tumour volume decrease ≥ 30% per log10 (months). ResNet-10t, ResNet-18, ResNet-50 and DenseNet-121 with ImageNet or RadImageNet pretraining were evaluated in three workflows: (1) frozen backbone CNN, head-only training; (2) logistic regression or naive Bayes classification of features extracted from frozen CNNs; and (3) restricted CNN fine-tuning. Five partially overlapping chronological development/test splits were assessed, with all data-dependent fitting, feature selection and model optimisation restricted to development data. Results: The smallest RadImageNet ResNet-10t CNN performed best when fine-tuned (peak AUCs 0.953 and 1.000) but showed the widest split-to-split variation (range 0.738–1.000). When RadImageNet features extracted from ResNet-10t were classified by naive Bayes and logistic regression, they reached lower median and peak AUCs but within narrower ranges. Predictive performance did not consistently depend on CNN size. Conclusions: To our knowledge, this is the first deep learning study predicting volumetric meningioma response after GKRS. We provide openly available, ready-to-use code for three major workflows in small clinical datasets, with safeguards against information leakage. The benefit of fine-tuning and of RadImageNet pretraining depended on architecture. Robust small-cohort modelling requires attention to performance consistency across chronological splits, not only to the highest AUC in a single split.
Authors
- Herwin Speckter (ORCID: https://orcid.org/0000-0002-9904-2395)
- Luis Suazo (ORCID: https://orcid.org/0000-0003-0227-6775)
- Peter Stoeter (ORCID: https://orcid.org/0000-0003-4045-4990)
- Giancarlo Hernandez
- Marko Radulović (ORCID: https://orcid.org/0000-0002-2314-7457)
- Ismael Peralta (ORCID: https://orcid.org/0000-0002-1162-1318)
- José Bido
- Issael Ramírez (ORCID: https://orcid.org/0000-0002-1266-9460)
- Velicko Vranes (ORCID: https://orcid.org/0000-0001-8907-7781)
- Santiago Valenzuela (ORCID: https://orcid.org/0009-0003-4136-2745)
- Diones Rivera (ORCID: https://orcid.org/0000-0001-5935-8005)
- Teuddis Bernard
- Jeffrey Paulino
- Ivan Gonzalez (ORCID: https://orcid.org/0009-0008-6043-6718)
Institutions
- Instituto Tecnológico de Santo Domingo (DO)
- Institut za onkologiju i radiologiju Srbije (RS)
Publication Details
- Journal
- Cancers
- Published
- 2026-11-09
- DOI
- https://doi.org/10.3390/cancers18213590
- Primary Topic
- Radiomics and Machine Learning in Medical Imaging
- Type
- article
- Field-Weighted Citation Impact
- 0.00