Classification of Diabetic Retinopathy using Pre-Trained Deep Learning Models

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Al-Kamachy, Inas, Hassanpour, Reza, Choupani, Roya
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866912006922043392
author Al-Kamachy, Inas
Hassanpour, Reza
Choupani, Roya
author_facet Al-Kamachy, Inas
Hassanpour, Reza
Choupani, Roya
contents Diabetic Retinopathy (DR) stands as the leading cause of blindness globally, particularly affecting individuals between the ages of 20 and 70. This paper presents a Computer-Aided Diagnosis (CAD) system designed for the automatic classification of retinal images into five distinct classes: Normal, Mild, Moderate, Severe, and Proliferative Diabetic Retinopathy (PDR). The proposed system leverages Convolutional Neural Networks (CNNs) employing pre-trained deep learning models. Through the application of fine-tuning techniques, our model is trained on fundus images of diabetic retinopathy with resolutions of 350x350x3 and 224x224x3. Experimental results obtained on the Kaggle platform, utilizing resources comprising 4 CPUs, 17 GB RAM, and 1 GB Disk, demonstrate the efficacy of our approach. The achieved Area Under the Curve (AUC) values for CNN, MobileNet, VGG-16, InceptionV3, and InceptionResNetV2 models are 0.50, 0.70, 0.53, 0.63, and 0.69, respectively.
format Preprint
id arxiv_https___arxiv_org_abs_2403_19905
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Classification of Diabetic Retinopathy using Pre-Trained Deep Learning Models
Al-Kamachy, Inas
Hassanpour, Reza
Choupani, Roya
Computer Vision and Pattern Recognition
Artificial Intelligence
68T07
Diabetic Retinopathy (DR) stands as the leading cause of blindness globally, particularly affecting individuals between the ages of 20 and 70. This paper presents a Computer-Aided Diagnosis (CAD) system designed for the automatic classification of retinal images into five distinct classes: Normal, Mild, Moderate, Severe, and Proliferative Diabetic Retinopathy (PDR). The proposed system leverages Convolutional Neural Networks (CNNs) employing pre-trained deep learning models. Through the application of fine-tuning techniques, our model is trained on fundus images of diabetic retinopathy with resolutions of 350x350x3 and 224x224x3. Experimental results obtained on the Kaggle platform, utilizing resources comprising 4 CPUs, 17 GB RAM, and 1 GB Disk, demonstrate the efficacy of our approach. The achieved Area Under the Curve (AUC) values for CNN, MobileNet, VGG-16, InceptionV3, and InceptionResNetV2 models are 0.50, 0.70, 0.53, 0.63, and 0.69, respectively.
title Classification of Diabetic Retinopathy using Pre-Trained Deep Learning Models
topic Computer Vision and Pattern Recognition
Artificial Intelligence
68T07
url https://arxiv.org/abs/2403.19905