khamirdesai commited on
Commit
c4edd3f
路
verified 路
1 Parent(s): 0b14f5f

Add Kermany dataset, reported result and validated training instructions

Browse files
Files changed (1) hide show
  1. README.md +10 -2
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 路 Sree Saicharan Vadapalli 路 Mayuri A. Mehta
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