Calibrated Hybrid CNN-Transformer for Retinal OCT Classification
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