General-purpose versus retina-specific foundation models for multilabel retinal disease classification: effects of adaptation and class weighting

Whether a retina-specific or a general-purpose foundation model performs better for retinal disease classification can depend on how the model is adapted and how class imbalance is handled during training. These factors have not been examined jointly for this model pair. In this retrospective comparative study, RETFound-MAE and DINOv2 were compared on patient-disjoint BRSET splits as validation-selected linear probes and as fully fine-tuned models, with supervised ResNet-50 as a reference. External testing used ODIR-5K, mBRSET, and APTOS-2019, and acquisition shift was evaluated between overlap-free Canon and Nikon subsets. DINOv2 had higher macro AUROC than RETFound-MAE under linear probing (0.897 vs 0.822) and after fine-tuning (0.937 vs 0.911), and fine-tuning narrowed the difference. Class-weighted training produced substantial calibration error. Correcting the weighting-induced probability offset reduced macro ECE from 0.152 to 0.007 for RETFound-MAE and from 0.066 to 0.012 for DINOv2, and matched unweighted training improved both discrimination and calibration. Fine-tuned DINOv2 had the highest observed AUROC on all three external cohorts, and its frozen probe showed a smaller Canon-to-Nikon decrease than the RETFound-MAE probe. Under the evaluated recipes, backbone choice, adaptation regime, and loss formulation each influenced performance.

Authors

Institutions

Publication Details

Journal
Zenodo (CERN European Organization for Nuclear Research)
Published
2026-10-27
DOI
https://doi.org/10.5281/zenodo.23702744
Primary Topic
Retinal Imaging and Analysis
Type
preprint
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
preprint

General-purpose versus retina-specific foundation models for multilabel retinal disease classification: effects of adaptation and class weighting

Samarth Mishra
2 links
Zenodo (CERN European Organization for Nuclear Research)
Retinal Imaging and Analysis
preprint

General-purpose versus retina-specific foundation models for multilabel retinal disease classification: effects of adaptation and class weighting

Samarth Mishra
preprint en

Abstract

Whether a retina-specific or a general-purpose foundation model performs better for retinal disease classification can depend on how the model is adapted and how class imbalance is handled during training. These factors have not been examined jointly for this model pair. In this retrospective comparative study, RETFound-MAE and DINOv2 were compared on patient-disjoint BRSET splits as validation-selected linear probes and as fully fine-tuned models, with supervised ResNet-50 as a reference. External testing used ODIR-5K, mBRSET, and APTOS-2019, and acquisition shift was evaluated between overlap-free Canon and Nikon subsets. DINOv2 had higher macro AUROC than RETFound-MAE under linear probing (0.897 vs 0.822) and after fine-tuning (0.937 vs 0.911), and fine-tuning narrowed the difference. Class-weighted training produced substantial calibration error. Correcting the weighting-induced probability offset reduced macro ECE from 0.152 to 0.007 for RETFound-MAE and from 0.066 to 0.012 for DINOv2, and matched unweighted training improved both discrimination and calibration. Fine-tuned DINOv2 had the highest observed AUROC on all three external cohorts, and its frozen probe showed a smaller Canon-to-Nikon decrease than the RETFound-MAE probe. Under the evaluated recipes, backbone choice, adaptation regime, and loss formulation each influenced performance.

Zenodo (CERN European Organization for Nuclear Research)
Stanford University (US)
Good health and well-being
Retinal Imaging and Analysis
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.

Laika access coming soon