Comparative analysis of CNN, Transformer, and Mamba-based models for automated eye disease classification from fundus images
- 1Department of Computer Science and Engineering, Faculty of Engineering, University of Rajshahi, Rajshahi, Bangladesh.
Abstract
We present a comparative analysis of six models belonging to Convolutional Neural Network (CNN), Transformer, and Mamba families for automated eye disease classification from fundus images under identical experimental conditions. To the best of our knowledge, our work is the first to explore Mamba-based models for automated multi-class eye disease classification from color fundus images. Our analysis suggests that, when evaluated fairly, modern CNN, Transformer, and Mamba-based models provide similar performance for eye disease classification on the Eye Disease Image Dataset (EDID). Along with analyzing individual model performance, we further evaluate performance using a logit-level ensemble of the top three models. The ensemble results imply that these model families capture partially complementary visual features, resulting in a slight performance gain when combined. In both cases, class-wise performance examination indicates that diseases with distinctive structural characteristics are recognized more consistently, while categories with subtle or overlapping features remain challenging. Our implementation, including the training and evaluation pipelines as well as the data splits, is publicly available at https://github.com/ahanaf019/EyeDisease2026.
Related
- A lightweight privacy-aware federated vision transformer for tomato leaf disease classification
- MangoSwin-attention: a swin transformer with attention pooling for generalizable mango leaf disease classification
- A hybrid EfficientNet-B0-CBAM-Vision Transformer framework for accurate cherry leaf disease classification
- Gradient boosting models for gallstone disease classification: A comparative study using demographic and clinical data
- Improved vision transformer for sports image classification via transfer learning and knowledge distillation
BibTeX
@article{khan2026comparative,
title = {{Comparative analysis of CNN, Transformer, and Mamba-based models for automated eye disease classification from fundus images}},
author = {Md Ahanaf Arif Khan and Sangeeta Biswas},
journal = {PLOS One},
year = {2026},
month = oct,
doi = {10.1371/journal.pone.0360171},
url = {https://doi.org/10.1371/journal.pone.0360171},
note = {CC BY},
}
RIS
TY - JOUR
TI - Comparative analysis of CNN, Transformer, and Mamba-based models for automated eye disease classification from fundus images
AU - Md Ahanaf Arif Khan
AU - Sangeeta Biswas
PY - 2026
DA - 2026/10/26
JO - PLOS One
DO - 10.1371/journal.pone.0360171
UR - https://doi.org/10.1371/journal.pone.0360171
AB - We present a comparative analysis of six models belonging to Convolutional Neural Network (CNN), Transformer, and Mamba families for automated eye disease classification from fundus images under identical experimental conditions. To the best of our knowledge, our work is the first to explore Mamba-based models for automated multi-class eye disease classification from color fundus images. Our analysis suggests that, when evaluated fairly, modern CNN, Transformer, and Mamba-based models provide similar performance for eye disease classification on the Eye Disease Image Dataset (EDID). Along with analyzing individual model performance, we further evaluate performance using a logit-level ensemble of the top three models. The ensemble results imply that these model families capture partially complementary visual features, resulting in a slight performance gain when combined. In both cases, class-wise performance examination indicates that diseases with distinctive structural characteristics are recognized more consistently, while categories with subtle or overlapping features remain challenging. Our implementation, including the training and evaluation pipelines as well as the data splits, is publicly available at https://github.com/ahanaf019/EyeDisease2026.
SN - 1932-6203
ER -