Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

PRETI: Patient-Aware Retinal Foundation Model via Metadata-Guided Representation Learning

About

Retinal foundation models have significantly advanced retinal image analysis by leveraging self-supervised learning to reduce dependence on labeled data while achieving strong generalization. Many recent approaches enhance retinal image understanding using report supervision, but obtaining clinical reports is often costly and challenging. In contrast, metadata (e.g., age, gender) is widely available and serves as a valuable resource for analyzing disease progression. To effectively incorporate patient-specific information, we propose PRETI, a retinal foundation model that integrates metadata-aware learning with robust self-supervised representation learning. We introduce Learnable Metadata Embedding (LME), which dynamically refines metadata representations. Additionally, we construct patient-level data pairs, associating images from the same individual to improve robustness against non-clinical variations. To further optimize retinal image representation, we propose Retina-Aware Adaptive Masking (RAAM), a strategy that selectively applies masking within the retinal region and dynamically adjusts the masking ratio during training. PRETI captures both global structures and fine-grained pathological details, resulting in superior diagnostic performance. Extensive experiments demonstrate that PRETI achieves state-of-the-art results across diverse diseases and biomarker predictions using in-house and public data, indicating the importance of metadata-guided foundation models in retinal disease analysis. Our code and pretrained model are available at https://github.com/MICV-yonsei/PRETI

Yeonkyung Lee, Woojung Han, Youngjun Jun, Hyeonmin Kim, Jungkyung Cho, Seong Jae Hwang• 2025

Related benchmarks

TaskDatasetResultRank
Age-Related Macular Degeneration (AMD) Predictionin-house dataset (test)
ROC AUC0.772
7
Coronary Artery Calcium (CAC) Predictionin-house dataset (test)
ROC AUC89.9
7
Diabetic Retinopathy (DR) Predictionin-house dataset (test)
ROC AUC0.982
7
Estimated Glomerular Filtration Rate (eGFR) Predictionin-house dataset (test)
AUC ROC0.699
7
Glaucoma ClassificationPAPILA
AUROC87.9
7
Glaucoma Predictionin-house dataset (test)
ROC AUC0.777
7
Diabetic Retinopathy ClassificationMESSIDOR-2
AUROC0.866
7
Glaucoma ClassificationGF
AUROC0.955
7
Multicategory ClassificationRetina
AUROC89.3
7
Multicategory ClassificationJSIEC
AUROC99.4
7
Showing 10 of 11 rows

Other info

Code

Follow for update