Back to Search Start Over

A Comprehensive Performance Analysis of Transfer Learning Optimization in Visual Field Defect Classification.

Authors :
Abu, Masyitah
Zahri, Nik Adilah Hanin
Amir, Amiza
Ismail, Muhammad Izham
Yaakub, Azhany
Anwar, Said Amirul
Ahmad, Muhammad Imran
Source :
Diagnostics (2075-4418). May2022, Vol. 12 Issue 5, pN.PAG-N.PAG. 26p.
Publication Year :
2022

Abstract

Numerous research have demonstrated that Convolutional Neural Network (CNN) models are capable of classifying visual field (VF) defects with great accuracy. In this study, we evaluated the performance of different pre-trained models (VGG-Net, MobileNet, ResNet, and DenseNet) in classifying VF defects and produced a comprehensive comparative analysis to compare the performance of different CNN models before and after hyperparameter tuning and fine-tuning. Using 32 batch sizes, 50 epochs, and ADAM as the optimizer to optimize weight, bias, and learning rate, VGG-16 obtained the highest accuracy of 97.63 percent, according to experimental findings. Subsequently, Bayesian optimization was utilized to execute automated hyperparameter tuning and automated fine-tuning layers of the pre-trained models to determine the optimal hyperparameter and fine-tuning layer for classifying many VF defect with the highest accuracy. We found that the combination of different hyperparameters and fine-tuning of the pre-trained models significantly impact the performance of deep learning models for this classification task. In addition, we also discovered that the automated selection of optimal hyperparameters and fine-tuning by Bayesian has significantly enhanced the performance of the pre-trained models. The results observed the best performance for the DenseNet-121 model with a validation accuracy of 98.46% and a test accuracy of 99.57% for the tested datasets. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
20754418
Volume :
12
Issue :
5
Database :
Academic Search Index
Journal :
Diagnostics (2075-4418)
Publication Type :
Academic Journal
Accession number :
157191979
Full Text :
https://doi.org/10.3390/diagnostics12051258