loopback-kr commited on
Commit
983ded9
·
verified ·
1 Parent(s): 3cfb85c

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +88 -0
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
+ ```