English

Calibrated Hybrid CNN-Transformer for Retinal OCT Classification

Image and Video Processing 2026-07-09 v1 Computer Vision and Pattern Recognition

Abstract

Deep models for retinal optical coherence tomography (OCT) classification report high accuracy but rarely report whether their confidence can be trusted -- a gap that matters when a wrong-but-confident reading delays sight-saving treatment. We pair a hybrid convolutional-Transformer encoder with a gradient-boosting (XGBoost) classification head and a three-part clinical safety layer: confidence calibration, out-of-distribution (OOD) rejection, and per-prediction uncertainty flagging. On four-class OCT (84,495 scans) the model reaches 95.4% accuracy while cutting calibration error twelve-fold (expected calibration error, ECE = 0.0024), so the confidence it reports tracks its true accuracy. To our knowledge this is the first OCT classifier to validate all three safety mechanisms jointly, with public weights and reproducible multi-seed evaluation.

Cite

@article{arxiv.2607.09809,
  title  = {Calibrated Hybrid CNN-Transformer for Retinal OCT Classification},
  author = {Animesh Kumar},
  journal= {arXiv preprint arXiv:2607.09809},
  year   = {2026}
}

Comments

4 pages, 2 figures, 4 tables. Code, model weights, and REST inference API are available on GitHub and HuggingFace

R2 v1 2026-07-22T20:37:15.801Z