Transfer learning and hybrid models for the classification of eye diseases on retinal images

Dhriti Kumar; Chaya Ravindra

doi:10.1117/12.3027368

3 October 2024 Transfer learning and hybrid models for the classification of eye diseases on retinal images

Dhriti Kumar, Chaya Ravindra

Author Affiliations +

Proceedings Volume 13138, Applications of Machine Learning 2024; 131380Z (2024) https://doi.org/10.1117/12.3027368
Event: Optical Engineering + Applications, 2024, San Diego, California, United States

Conference Poster

Abstract

Glaucoma (GC), diabetic retinopathy (DR), and cataracts are the leading causes of vision loss globally. However, the diseases can be prevented from further progression when detected in the early stages. Conventional manual diagnosis is prolonged and requires the experience of a trained ophthalmologist. Therefore, automated techniques using artificial intelligence algorithms are evaluated for their ability to identify these eye diseases. In this study, transfer learning and featurization techniques are employed using pre-trained deep-learning (DL) neural networks (ResNet50, MobileNetV2, and VGG16) and statistical machine learning (ML) algorithms (MLP Classifier, KNN, and Random Forest Classifier). These architectures were trained, validated, and tested using a public dataset that included retinal images for diseased (GC, DR, and cataracts) and normal eyes. The ResNet50 neural network architecture had the highest testing accuracy of 91.51% among the deep learning methods. Due to its high performance, features were extracted from this model (featurization) and used for the statistical ML classifiers, creating hybrid models. The MLP Classifier hybrid model had the highest accuracy value at 92.04%. The knowledge from this study has the potential to aid, hasten, and improve accuracy in the process of eye disease diagnosis.

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Dhriti Kumar and Chaya Ravindra "Transfer learning and hybrid models for the classification of eye diseases on retinal images", Proc. SPIE 13138, Applications of Machine Learning 2024, 131380Z (3 October 2024); https://doi.org/10.1117/12.3027368

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

;

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE