Journal article Open Access

Learning Rate Optimization in CNN for Accurate Ophthalmic Classification

Mahmoud Smaida*; Serhii Yaroshchak; Ahmed Y. Ben Sasi

Sponsor(s)
Blue Eyes Intelligence Engineering and Sciences Publication

One of the most important hyper-parameters for model training and generalization is the learning rate. Recently, many research studies have shown that optimizing the learning rate schedule is very useful for training deep neural networks to get accurate and efficient results. In this paper, different learning rate schedules using some comprehensive optimization techniques have been compared in order to measure the accuracy of a convolutional neural network CNN model to classify four ophthalmic conditions. In this work, a deep learning CNN based on Keras and TensorFlow has been deployed using Python on a database that contains 1692 images, which consists of four types of ophthalmic cases: Glaucoma, Myopia, Diabetic retinopathy, and Normal eyes. The CNN model has been trained on Google Colab. GPU with different learning rate schedules and adaptive learning algorithms. Constant learning rate, time-based decay, step-based decay, exponential decay, and adaptive learning rate optimization techniques for deep learning have been addressed. Adam adaptive learning rate method. has outperformed the other optimization techniques and achieved the best model accuracy of 92.58% for training set and 80.49% for validation datasets, respectively.

Files (920.4 kB)
Name Size
B82591210220.pdf
md5:4a956870e73ed04a5ae1c92df52dfb98
920.4 kB Download
31
13
views
downloads
Views 31
Downloads 13
Data volume 12.0 MB
Unique views 24
Unique downloads 12

Share

Cite as