Eye Diseases Classification Using Transfer Learning Technique of Deep Neural Network

Authors

  • Sheetal J. Nagar Department of Computer / IT Engineering, Gujarat Technological University, Ahmedabad, Gujarat, India Author
  • Nikhil Gondaliya Department of Information Technology Engineering, G H Patel College of Engineering & Technology, Vallabh Vidyanagar, Gujarat, India Author

DOI:

https://doi.org/10.32628/CSEIT251112124

Keywords:

Eye Diseases Classification, Diabetic Retinopathy, Medical Image Classification, CNN, Transfer Learning

Abstract

The human eye is a complex organ susceptible to various diseases and abnormalities that can impair vision. Identifying retinal diseases early and precisely categorizing them is crucial to ensure effective treatment is administered promptly. In this proposed study, we implemented Convolutional Neural Networks (CNNs) along with pretrained neural network architectures. We use existing methods as the foundation and develop models tailored for detecting eye diseases, such as Diabetic Retinopathy (DR), Cataracts, and Glaucoma, specifically among diabetic patients. These models are applied for screening of retinal fundus images for accurate diagnosis. Prompt identification and suitable treatment can aid in preventing the commencement and advancement of diabetic retinopathy, cataracts, and glaucoma among patients. Accurate identification of these conditions is paramount in the healthcare sector. Our research aims to categorize eye diseases through the utilization of retinal fundus images for multiclass classification. The multiclass classification outcomes, implemented on the Kaggle platform, indicated that CNN, MobileNet, DenseNet121, ResNet and EfficientNetB3 yielded training accuracies of 98.50%, 99.94%, 99.37%, 99.90% and 99.97 as well as the validation accuracies of 88.22%, 95.67%, 93.99%, 93.30% and 95.97% respectively.

Downloads

Download data is not yet available.

References

World Health Organization, “Strengthening diagnosis and treatment of diabetic retinopathy in the South-East Asia Region”, New Delhi. World Health Organization, Regional Office for South-East Asia, 2020.

M. Puttagunta, S. Ravi, “Medical image analysis based on deep learning approach, Multimedia Tools and Applications”, Springer, vol. 80, pp. 24365– 24398, 2021.

M. Li, Y. Jiang, Y. Zhang, H. Zhu, “Medical image analysis using deep learning algorithms”, Frontiers in Public Health, vol. 11, pp. 01–28, 2023.

M. Rana, M. Bhushan, “Machine learning and deep learning approach for medical image analysis: diagnosis to detection”, vol. 82, pp 26731–26769 Multimedia Tools and Applications, Springer Nature 2023.

G. Kumar, S. Chatterjee, C. Chattopadhyay, “DRDNET: DIAGNOSIS OF DIABETIC RETINOPATHY USING CAPSULE NETWORK”, 2020 IEEE Sixth International Conference on Multimedia Big Data (BigMM), 379–385, 2020.

Revathy R, Nithya B S, Reshma J J, Ragendhu S S, Sumithra M D, “Diabetic Retinopathy Detection using Machine Learning”, International Journal of Engineering Research & Technology (IJERT), vol. 09, Issue 06, pp 122-126, 2020.

V. Rao Naramala, B. Kumar, V. Rao, A. Mishra, S Abdul Hannan, Y. Baker El-Ebiary, R. Manikandan, “Enhancing Diabetic Retinopathy Detection Through Machine Learning with Restricted Boltzmann Machines”, International Journal of Advanced Computer Science and Applications, vol. 14 Issue 9, pp. 573-585, 2023.

D. Das, S. Kumar Biswas, S. Bandyopadhyay, “Detection of Diabetic Retinopathy using Convolutional Neural Networks for Feature Extraction and Classification (DRFEC), Multimedia Tools and Applications”, Springer Nature, vol. 82, pp. 29943–30001, 2022.

A. M Pamadi, A. Ravishankar, A. Nithya P¸ G Jahnavi, S. Kathavate, “Diabetic Retinopathy Detection using MobileNetV2 Architecture”, International Conference on Smart Technologies and Systems for Next Generation Computing, pp. 1-5, 2022.

eye-diseases-classification Dataset, Available: https://www.kaggle.com/datasets/gunavenkatdoddi/eye-diseases-classification

tf.keras.preprocessing.image.ImageDataGenerator, Available: https://www.tensorflow.org/api_docs/python/tf/keras/preprocessing/image/ImageDataGenerator

Models and layers, Available: https://www.tensorflow.org/js/guide/models_and_layers

A. G. Howard, M. Zhu, B. Chen, D. Kalenichenko, W. Wang, T. Weyand, M. Andreetto, H. Adam, “MobileNets: Efficient Convolutional Neural Networks for Mobile Vision”, Research Gate (2017)

W. Nazih, A. O. Aseeri, O. Youssef Atallah, S. El-Sappagh, “Vision Transformer Model for Predicting the Severity of Diabetic Retinopathy in Fundus Photography-Based Retina Images”, pp. 117546–117561, IEEE Access, vol. 11, pp. 117546-117561,2023.

I. A1-Kamachy, R. Hassanpour, R. Choupani, “Classification of Diabetic Retinopathy using Pre-Trained Deep Learning Models”, arXiv preprint arXiv:2403.19905, 2024.

G. Huang, Z. Liu, L. van der Maaten, “Densely Connected Convolutional Networks”, Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 2261-2269, (2017).

X. Zhao, L. Wang, Y. Zhang, X. Han, M. Deveci, M. Parmar, “A Review Of Convolutional Neural Networks In Computer Vision”, Artificial Intelligence Review, Springer, vol. 57, pp. 1-43 (2024).

M. Tan, Q. V. Le, “EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks”, Proceedings of the 36th International Conference on Machine Learning, Long Beach, California, PMLR 97 (2020).

F. Ramzan1, M. Khan, A. Rehmat, S. Iqbal, T. Saba, A. Rehman, Z. Mehmood, “A Deep Learning Approach for Automated Diagnosis and Multi-Class Classification of Alzheimer’s Disease Stages Using Resting-State fMRI and Residual Neural Networks”, Journal of Medical Systems, vol. 44, 2020

Downloads

Published

30-01-2025

Issue

Section

Research Articles

How to Cite

Eye Diseases Classification Using Transfer Learning Technique of Deep Neural Network. (2025). International Journal of Scientific Research in Computer Science, Engineering and Information Technology, 11(1), 1247-1257. https://doi.org/10.32628/CSEIT251112124