vyshnav112233/retina-dr-deit-without-distillation

DeiT WITHOUT Distillation

This model classifies retinal images into:

0. No DR
1. Mild
2. Moderate
3. Severe
4. Proliferative DR

Model architecture: deit_small_patch16_224

Distilled model: False

Primary metric: Tuned Quadratic Weighted Kappa

Primary inference output: Threshold-tuned QWK prediction

Tuned thresholds:

[ 1.0235428891379215, 1.7234984095439454, 2.025988978797374, 2.6660228022901524 ]