A Deep Learning Based Pipeline for Image Grading of Diabetic Retinopathy

Files

TR Number

Date

2018-06-21

Authors

Journal Title

Journal ISSN

Volume Title

Publisher

Virginia Tech

Abstract

Diabetic Retinopathy (DR) is one of the principal sources of blindness due to diabetes mellitus. It can be identified by lesions of the retina, namely microaneurysms, hemorrhages, and exudates. DR can be effectively prevented or delayed if discovered early enough and well-managed. Prior studies on diabetic retinopathy typically extract features manually but are time-consuming and not accurate. In this research, we propose a research framework using advanced retina image processing, deep learning, and a boosting algorithm for high-performance DR grading. First, we preprocess the retina image datasets to highlight signs of DR, then follow by a convolutional neural network to extract features of retina images, and finally apply a boosting tree algorithm to make a prediction based on extracted features. Experimental results show that our pipeline has excellent performance when grading diabetic retinopathy images, as evidenced by scores for both the Kaggle dataset and the IDRiD dataset.

Description

Keywords

Retina Image, Image Grading, Diabetic Retinopathy, Early Detection, Feature Extraction, ConvNN, Deep learning (Machine learning), Boosting Decision Tree

Citation

Collections