Using Transfer Learning for Diabetic Retinopathy Stages Classification
Abstract Purpose – Diabetes is a chronic disease, that leads to damage of many systems of the body. One of the dangerous complications of diabetes is diabetic retinopathy. Frequent inspection for diabetic retinopathy is essential to recognize patients at risk of visual impairment. The disease grade level must be tracked to manage its progress and to start the appropriate decision for treatment in time. Effective automated methods for the detection of diabetic retinopathy and the classification of its severity stage are necessary. It also helps to decrease the burden on ophthalmologists and reduce diagnostic contradictions among manual readers. Methods– In this research, a convolutional neural network (CNN) is used based on color retinal fundus images for the detection of diabetic retinopathy (DR) and classification of its stages. CNN can recognize sophisticated features on the retina and so provide an automatic diagnosis. The pre-trained CNN model Visual Geometry Group (VGG) is applied on DR data using a transfer learning approach to utilize the already learnt parameters based on 1,000,000 images of ImageNet with 1000 classes.Results – By conducting different experiments with different classes setting the built models achieved promising results. The best achieved accuracies for 2-ary, 3-ary, 4-ary, and 5-ary classification are 85.99, 80.5, 61.28, and 71, respectively.