Towards Interpretable Foundation Models for Retinal Fundus Images
About
Foundation models are used to extract transferable representations from large amounts of unlabeled data, typically via self-supervised learning (SSL). However, many of these models rely on architectures that offer limited interpretability, a critical issue in high-stakes domains such as medical imaging. We propose \model, a foundation model that is interpretable-by-design via a BagNet backbone whose small receptive fields generate class evidence maps that are faithful to the model's decision-making process. Additionally, \model{} incorporates a $2D$ projection layer during pretraining that enables direct visualization of the representation space, providing a dataset-level view of the learned structure including meaningful clinical clusters as well as potential spurious correlations. We trained \model{} on over 800,000 color fundus photographs from various sources to learn generalizable representations for different downstream tasks. Our model achieves performance comparable to RETFound, which has $16\times$ more parameters, while providing interpretable predictions on out-of-distribution data. These results suggest that large-scale SSL pretraining paired with inherent interpretability can lead to robust representations for retinal imaging. Code and pretrained models are available at \href{https://anonymous.4open.science/r/dual-ifm-3D5A/README.md}{www.anonymous.4open.science/dual-IFM}.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Diabetic Retinopathy Classification | Messidor (test) | AUC86.8 | 17 | |
| Glaucoma Classification | PAPILA | AUROC64.8 | 12 | |
| Diabetic Retinopathy Grading | APTOS (test) | -- | 10 | |
| Retinal Disease Classification | FIVES (test) | AUROC92.8 | 4 | |
| Diabetic Retinopathy Detection | IDRiD (test) | AUROC71.8 | 4 | |
| Diabetic Retinopathy Grading | APTOS | AUROC94.5 | 4 | |
| Diabetic Retinopathy Grading | DeepDRiD | AUROC0.897 | 4 | |
| Diabetic Retinopathy Grading | IDRID | AUROC0.799 | 4 | |
| Diabetic Retinopathy Grading | Messidor | AUROC0.862 | 4 | |
| Glaucoma Classification | PAPILA (test) | AUROC72.9 | 4 |