Add Kermany dataset, reported result and validated training instructions
Browse files
README.md
CHANGED
|
@@ -17,9 +17,9 @@ base_model: timm/vit_base_patch16_224.augreg_in21k_ft_in1k
|
|
| 17 |
|
| 18 |
**Sequential Hybrid Vision Transformer with MedMamba for Retinal Disease Classification**
|
| 19 |
|
| 20 |
-
Accepted for presentation at **IEEE GSCon 2027**
|
| 21 |
|
| 22 |
-
Khamir Desai 路
|
| 23 |
|
| 24 |
[Code](https://github.com/khamirdesai/RetViM) 路 [Checkpoint details](https://github.com/khamirdesai/RetViM/blob/main/docs/checkpoint.md) 路 [Citation](https://github.com/khamirdesai/RetViM/blob/main/CITATION.cff)
|
| 25 |
|
|
@@ -36,6 +36,14 @@ Khamir Desai 路 Sree Saicharan Vadapalli 路 Mayuri A. Mehta
|
|
| 36 |
|
| 37 |
The supplied checkpoint uses **native 1D Mamba**, with pooling fusion and classifier dimensions that differ from the manuscript's summarized SS2D configuration. The linked implementation preserves the trained tensors and loads them strictly. It is a custom PyTorch model, not a Transformers `AutoModel` checkpoint.
|
| 38 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
## Inference
|
| 40 |
|
| 41 |
```bash
|
|
|
|
| 17 |
|
| 18 |
**Sequential Hybrid Vision Transformer with MedMamba for Retinal Disease Classification**
|
| 19 |
|
| 20 |
+
Accepted for presentation at **IEEE GSCon 2027**.
|
| 21 |
|
| 22 |
+
Khamir Desai 路 Mayuri A. Mehta 路 Sree Saicharan Vadapalli
|
| 23 |
|
| 24 |
[Code](https://github.com/khamirdesai/RetViM) 路 [Checkpoint details](https://github.com/khamirdesai/RetViM/blob/main/docs/checkpoint.md) 路 [Citation](https://github.com/khamirdesai/RetViM/blob/main/CITATION.cff)
|
| 25 |
|
|
|
|
| 36 |
|
| 37 |
The supplied checkpoint uses **native 1D Mamba**, with pooling fusion and classifier dimensions that differ from the manuscript's summarized SS2D configuration. The linked implementation preserves the trained tensors and loads them strictly. It is a custom PyTorch model, not a Transformers `AutoModel` checkpoint.
|
| 38 |
|
| 39 |
+
## Dataset and reported result
|
| 40 |
+
|
| 41 |
+
[Kermany retinal OCT dataset](https://www.kaggle.com/datasets/paultimothymooney/kermany2018), with CNV, DME, DRUSEN and NORMAL labels. [Original dataset record](https://data.mendeley.com/datasets/rscbjbr9sj/2).
|
| 42 |
+
|
| 43 |
+
The manuscript reports **99.90% accuracy (967/968 test images)**, the highest among the methods compared there. This is paper-reported, not an independently verified overall leaderboard ranking or a newly reproduced benchmark.
|
| 44 |
+
|
| 45 |
+
Standalone [training and evaluation instructions](https://github.com/khamirdesai/RetViM/blob/main/docs/training.md) are available with the source. New training runs use explicit, hashed patient-disjoint manifests and are not claimed to recreate the original split.
|
| 46 |
+
|
| 47 |
## Inference
|
| 48 |
|
| 49 |
```bash
|