Academia.eduAcademia.edu

Rapid classification of glaucomatous fundus images

2021, Journal of the Optical Society of America

Abstract

We propose a new method for training convolutional neural networks which integrates reinforcement learning along with supervised learning and use it for transfer learning for classification of glaucoma from colored fundus images. The training method uses hill climbing techniques via two different climber types viz. "random movement" and "random detection" integrated with supervised learning model through stochastic gradient descent with momentum (SGDM) model. The model was trained and tested using the Drishti-GS, and RIM-ONE-r2 datasets having glaucomatous and normal fundus images. The performance for prediction was tested by transfer learning on five CNN architectures namely, GoogLeNet, NASNet,. A 5-fold classification was used for evaluating the performance and high sensitivities while maintaining high accuracies were achieved. Of the models tested, DenseNet-201 architecture performed the best in terms of sensitivity and area under the curve (AUC). This method of training allows transfer learning on small datasets and can be applied for tele-ophthalmology applications including training with local datasets.