Comparative Analysis of Pre-trained CNN Models on Retinal Diseases Classification

Hartanto, Theodore Alvin; Hansun, Seng

doi:10.22068/ijiepr.35.3.1935

Volume 35, Issue 3 (IJIEPR- In Progress 2024) IJIEPR 2024, 35(3): 1-12 | Back to browse issues page

‎ 10.22068/ijiepr.35.3.1935

Mendeley

Zotero

RefWorks

Hartanto T A, Hansun S. Comparative Analysis of Pre-trained CNN Models on Retinal Diseases Classification. IJIEPR 2024; 35 (3) :1-12
URL: http://ijiepr.iust.ac.ir/article-1-1935-en.html

Comparative Analysis of Pre-trained CNN Models on Retinal Diseases Classification

Theodore Alvin Hartanto¹

, Seng Hansun

1- Universitas Multimedia Nusantara
2- Universitas Multimedia Nusantara , seng.hansun@lecturer.umn.ac.id

Abstract: (189 Views)

One method to diagnose retinal diseases is by using the Optical Coherence Tomography (OCT) scans. Annually, it is estimated that around 30 million OCT scans are performed worldwide. However, the process of analyzing and diagnosing OCT scan results by an ophthalmologist requires a long time so machine learning, especially deep learning, can be utilized to shorten the diagnosis process and speed up the treatment process. In this study, several pre-trained deep learning models are compared, including EfficientNet-B0, ResNet-50V2, Inception-V3, and DenseNet-169. These models will be fine-tuned and trained with a dataset containing OCT scanned images to classify four retinal conditions, namely Choroidal Neovascularization (CNV), Diabetic Macular Edema (DME), Drusen, and Normal. The models that have been trained are then tested to classify the test set and the results are evaluated using a confusion matrix in terms of accuracy, recall, precision, and F1-score. The results show that the model with the best classification results in the batch size of 32 scenario is the ResNet-50V2 model with an accuracy value of 98.24%, precision of 98.25%, recall of 98.24%, and F1-score of 98.24%. While for the batch size of 64, the EfficientNet-B0 model is the model with the best classification results with an accuracy value of 96.59%, precision of 96.84%, recall of 96.59%, and F1-score of 96.59%.

Keywords: Classification, Comparative analysis, Deep learning, Optical coherence tomography, Pre-trained model, Retinal.Introduction.

Full-Text [PDF 690 kb] (63 Downloads)

Type of Study: Research | Subject: Other Related Subject
Received: 2023/12/7 | Accepted: 2024/06/9 | Published: 2024/09/9