2018/09/12 by Avinash Kori, Kori, Avinash, Sai Saketh Chennamsetty +5 · 1 citation
Computer Science · Medicine · #Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences #Imbalanced Data Classification Techniques #Retinal Diseases and Treatments #Retinal Imaging and Analysis
paper · pdf · doi:10.48550/arxiv.1809.04228
openalex publication_date 2018/09/12 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28
In this manuscript, we automate the procedure of grading of diabetic retinopathy and macular edema from fundus images using an ensemble of convolutional neural networks. The availability of limited amount of labeled data to perform supervised learning was circumvented by using transfer learning approach. The models in the ensemble were pre-trained on a large dataset comprising natural images and were later fine-tuned with the limited data for the task of choice. For an image, the ensemble of classifiers generate multiple predictions, and a max-voting based approach was utilized to attain the final grade of the anomaly in the image. For the task of grading DR, on the test data (n=56), the ensemble achieved an accuracy of 83.9%, while for the task for grading macular edema the network achieved an accuracy of 95.45% (n=44).