Multi-Disease Retinal Classification Using EfficientNet-B3 and Targeted Albumentations: A Benchmark on Kaggle Retinal Fundus Images Dataset

Read full text
Authors: Kurniawan Aji Saputra, Farrikh Alzami, Defri Kurniawan, Muhammad Naufal, Muslih Muslih, Rama Aria Megantara, Ricardus Anggi Pramunendar
Cited by: 1

Abstract

Retinal diseases remain one of the leading causes of blindness worldwide. This study develops a deep learning pipeline for multiclass retinal disease classification using EfficientNet-B3 combined with Albumentations to improve generalization. We target four classes: cataract, diabetic retinopathy, glaucoma, and normal. We use the Kaggle Retinal Disease dataset (4,217 fundus images) divided into 70% training, 10% validation, and 20% testing. Images are resized to 224×224 and augmented with horizontal flip, random brightness contrast, CLAHE, shiftscale rotate, crop, gamma correction, and elastic transformation. The EfficientNet-B3 backbone is refined after head training with warm-up and learning rate regularization (batch normalization, dropout). After 50 epochs, the best validation performance reaches 0.9526, and on the hold-out test set, the model achieves 95.38% overall accuracy. The F1 scores per class were 1.0000 (diabetic retinopathy), 0.9685 (cataract), 0.9255 (normal), and 0.9184 (glaucoma). Confusion analysis showed that most errors involved glaucoma being misclassified as normal, likely due to optic disc similarities. These results demonstrate that EfficientNet-B3 with targeted augmentation provides accurate and reliable multi-disease screening of fundus images, with the potential to support faster and more consistent triage in clinical workflows. Future research should expand clinical validation and explore attention mechanisms or multimodal input to reduce glaucoma-normal ambiguity.

Fields of Study & Subjects

Retinal Imaging and Analysis Retinal Diseases and Treatments Retinopathy of Prematurity Studies

Publication details

Journal / source
SinkrOn
Publisher
Politeknik Ganesha
Volume
10
Issue
1
Pages
232–241
Publication date
3 January 2026