Volume 35, Issue 3 (IJIEPR 2024)                   IJIEPR 2024, 35(3): 21-32 | Back to browse issues page


XML Print


Download citation:
BibTeX | RIS | EndNote | Medlars | ProCite | Reference Manager | RefWorks
Send citation to:

Hartanto T A, Hansun S. Comparative Analysis of Pre-trained CNN Models on Retinal Diseases Classification. IJIEPR 2024; 35 (3) :21-32
URL: http://ijiepr.iust.ac.ir/article-1-1935-en.html
1- Universitas Multimedia Nusantara
2- Universitas Multimedia Nusantara , seng.hansun@lecturer.umn.ac.id
Abstract:   (922 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%.
Full-Text [PDF 690 kb]   (283 Downloads)    
Type of Study: Research | Subject: Other Related Subject
Received: 2023/12/7 | Accepted: 2024/06/9 | Published: 2024/09/9

Add your comments about this article : Your username or Email:
CAPTCHA

Send email to the article author


Rights and permissions
Creative Commons License This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.