Create README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,91 @@
|
|
| 1 |
---
|
|
|
|
|
|
|
| 2 |
license: mit
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
+
language:
|
| 3 |
+
- en
|
| 4 |
license: mit
|
| 5 |
+
tags:
|
| 6 |
+
- medical-imaging
|
| 7 |
+
- ophthalmology
|
| 8 |
+
- retinal-fundus
|
| 9 |
+
- vision-language-model
|
| 10 |
+
- multimodal
|
| 11 |
+
- foundation-model
|
| 12 |
+
- vl-bert
|
| 13 |
+
- siglip2
|
| 14 |
+
pipeline_tag: image-classification
|
| 15 |
+
library_name: pytorch
|
| 16 |
---
|
| 17 |
+
|
| 18 |
+
# Luna — Retinal Foundation Model with Hybrid Clinical & Anatomical Supervision
|
| 19 |
+
|
| 20 |
+
**Luna** is a multimodal, multi-task retinal foundation model: a single-stream VL-BERT (SigLIP‑2 ViT‑Base vision encoder + PubMedBERT-initialized BERT‑Base text encoder) that reconstructs **masked clinical tokens** (disease label, age, sex, eye laterality) from 512×512 fundus images with a UNETR decoder jointly supervising vessel/optic-disc segmentation.
|
| 21 |
+
|
| 22 |
+
> Paper: *Multimodal Foundation Model of High-Resolution Fundus Photos with Clinical Metadata via Predicting Demographics and Anatomic Structure* — Lim, H. et al.
|
| 23 |
+
> Code: https://github.com/loopback-kr/Luna
|
| 24 |
+
|
| 25 |
+
## Model Details
|
| 26 |
+
|
| 27 |
+
| Attribute | Value |
|
| 28 |
+
|---|---|
|
| 29 |
+
| **Architecture** | SigLIP‑2 ViT‑Base/16 vision encoder + BERT‑Base language encoder (single-stream VL-BERT fusion) + UNETR segmentation decoder |
|
| 30 |
+
| **Parameters** | 93.5M (ViT‑Base backbone) |
|
| 31 |
+
| **Input resolution** | 512 × 512 RGB (CLAHE-enhanced) |
|
| 32 |
+
| **Vision encoder init.** | `google/siglip2-base-patch16-512` (ImageNet-pretrained) |
|
| 33 |
+
| **Text encoder init.** | `NeuML/pubmedbert-base-embeddings`, extended with `[LABEL]`, `[AGE]`, `[SEX]`, `[DIR]` special tokens |
|
| 34 |
+
| **Pre-training objective** | Masked clinical-token prediction (label/age/sex/direction) + vessel/optic-disc segmentation (Dice loss), *not* pixel-level MAE reconstruction |
|
| 35 |
+
| **Pre-training data** | 361,519 CFPs (52K public + 309K private institutional images) |
|
| 36 |
+
| **License** | MIT |
|
| 37 |
+
|
| 38 |
+
## Training Data
|
| 39 |
+
|
| 40 |
+
**\*** = private institutional data, not available externally.
|
| 41 |
+
|
| 42 |
+
**Pre-training datasets**
|
| 43 |
+
|
| 44 |
+
| Dataset | Reference |
|
| 45 |
+
|---|---|
|
| 46 |
+
| LAG | Li, L. et al. Attention Based Glaucoma Detection: A Large-Scale Database and CNN Model. *CVPR* (2019). |
|
| 47 |
+
| ODIR | Ocular Disease Intelligent Recognition (ODIR-2019). Peking University Grand Challenge (2019). |
|
| 48 |
+
| PARAGUAY | Castillo Benítez, V.E. et al. Dataset from fundus images for the study of diabetic retinopathy. *Data in Brief* 36, 107068 (2021). |
|
| 49 |
+
| G1020 | Bajwa, M.N. et al. G1020: A Benchmark Retinal Fundus Image Dataset for Computer-Aided Glaucoma Detection. *IJCNN* (2020). |
|
| 50 |
+
| FUND-OCT | Hassan, T., Akram, M.U., Werghi, N. & Nazir, N. A hybrid convolutional framework for the automated extraction of retinal lesions and lesion-influenced grading of human retinal pathology. *IEEE J. Biomed. Health Inform.* 25(1) (2020). |
|
| 51 |
+
| Drishti-GS1 | Sivaswamy, J. et al. Drishti-GS: Retinal Image Dataset for Optic Nerve Head Segmentation. *ISBI* 53–56 (2014). |
|
| 52 |
+
| HRF | Budai, A. et al. Robust Vessel Segmentation in Fundus Images. *Int. J. Biomed. Imaging* 2013, 154860 (2013). |
|
| 53 |
+
| ORIGA | Zhang, Z. et al. ORIGA(-light): An Online Retinal Fundus Image Database for Glaucoma Analysis and Research. *EMBC* 3065–3068 (2010). |
|
| 54 |
+
| OIA-DDR | Li, T. et al. Diagnostic Assessment of Deep Learning Algorithms for Diabetic Retinopathy Screening. *Information Sciences* 501, 511–522 (2019). |
|
| 55 |
+
| SUSTech-SYSU | Lin, L. et al. The SUSTech-SYSU dataset for automated exudate detection and diabetic retinopathy grading. *Sci. Data* 7, 409 (2020). |
|
| 56 |
+
| JICHI | Takahashi, H. et al. Applying artificial intelligence to disease staging: Deep learning for improved staging of diabetic retinopathy. *PLoS ONE* 12, e0179790 (2017). |
|
| 57 |
+
| CHAKSU | Kumar J H, R. et al. Chákṣu: A glaucoma specific fundus image database. *Sci. Data* 10, 70 (2023). |
|
| 58 |
+
| DR1 & DR2 | Pires, R. et al. Advancing Bag-of-Visual-Words Representations for Lesion Classification in Retinal Images. *PLoS ONE* 9, e96814 (2014). |
|
| 59 |
+
| DeepDRiD | Liu, R. et al. DeepDRiD: Diabetic Retinopathy—Grading and Image Quality Estimation Challenge. *Patterns* 3, 100512 (2022). |
|
| 60 |
+
| AIROGS-light-V2 | Kiefer, R. Glaucoma Dataset: EyePACS-AIROGS-light-V2. Kaggle (2024). doi:10.34740/KAGGLE/DSV/7802508 |
|
| 61 |
+
| FIDVS | Jin, K. et al. Fundus Image Dataset for Vessel Segmentation. Kaggle (2025). doi:10.34740/KAGGLE/DS/7319687 |
|
| 62 |
+
| JustRAIGS | Madadi, Y. et al. JustRAIGS: Justified Referral in AI Glaucoma Screening Challenge. *IEEE Trans. Med. Imaging* (2025). |
|
| 63 |
+
| SMDG | Kiefer, R. SMDG, A Standardized Fundus Glaucoma Dataset. Kaggle (2023). doi:10.34740/KAGGLE/DS/2329670 |
|
| 64 |
+
| AMC health-screen\* | Institutional cohort, Asan Medical Center — not externally published. |
|
| 65 |
+
| AMC clinic\* | Institutional cohort, Asan Medical Center — not externally published. |
|
| 66 |
+
|
| 67 |
+
**Downstream evaluation datasets**
|
| 68 |
+
|
| 69 |
+
| Dataset | Reference |
|
| 70 |
+
|---|---|
|
| 71 |
+
| APTOS 2019 | Karthik, Maggie & Dane, S. APTOS 2019 Blindness Detection. Kaggle (2019). |
|
| 72 |
+
| IDRiD | Porwal, P. et al. Indian Diabetic Retinopathy Image Dataset (IDRiD): A Database for Diabetic Retinopathy Screening Research. *Data* 3, 25 (2018). |
|
| 73 |
+
| MESSIDOR | Decencière, E. et al. Feedback on a Publicly Distributed Image Database: The MESSIDOR Database. *Image Anal. Stereol.* 231–234 (2014). |
|
| 74 |
+
| PAPILA | Kovalyk, O. et al. PAPILA: Dataset with fundus images and clinical data of both eyes of the same patient for glaucoma assessment. *Sci. Data* 9, 291 (2022). |
|
| 75 |
+
| Retina | jr2ngb. Retina Cataract Dataset. Kaggle (2016). |
|
| 76 |
+
| JSIEC | Cen, L.-P. et al. Automatic detection of 39 fundus diseases and conditions in retinal photographs using deep neural networks. *Nat. Commun.* 12, 4828 (2021). |
|
| 77 |
+
| BRSET | Nakayama, L.F. et al. A Brazilian Multilabel Ophthalmological Dataset (BRSET). PhysioNet (2024). |
|
| 78 |
+
| ARIA | Farnell, D.J.J. et al. Enhancement of blood vessels in digital fundus photographs via the application of multiscale line operators. *J. Franklin Inst.* 345, 748–765 (2008). |
|
| 79 |
+
| MAPLES-DR | Lepetit-Aimon, G. et al. MAPLES-DR: MESSIDOR Anatomical and Pathological Labels for Explainable Screening of Diabetic Retinopathy. *Sci. Data* 11, 914 (2024). |
|
| 80 |
+
| AMC health-screen subset\* | Institutional cohort, Asan Medical Center — systemic/anthropometric & lifestyle biomarker fine-tuning. |
|
| 81 |
+
| AMC CAC cohort\* | Institutional cohort, Asan Medical Center — coronary artery calcium classification. |
|
| 82 |
+
|
| 83 |
+
## Citation
|
| 84 |
+
|
| 85 |
+
```bibtex
|
| 86 |
+
@article{lim_luna_2026,
|
| 87 |
+
title = {Multimodal Foundation Model of High-Resolution Fundus Photos with Clinical Metadata via Predicting Demographics and Anatomic Structure},
|
| 88 |
+
author = {Lim, Hyunseok and Kim, Junseok and Oh, Joonseo and Kim, Kanghyun and Lim, Jongsoo and Jeong, Jinhoon and Jeong, Hy and Kim, Yoonjeon and Kim, Namkug},
|
| 89 |
+
journal = {(manuscript in preparation)},
|
| 90 |
+
}
|
| 91 |
+
```
|