Diabetic Retinopathy Grading with Deep Visual Attention Network

Abstract

Diabetic Retinopathy is a serious complication arising in diabetes afflicted patients. Its effective treatment depends on early detection, and the course of action varies decisively with the intensity of the affliction. Computeraided diagnosis helps to detect not only the presence or absence of the disease, but also the severity, making it easier for ophthalmologists to construct a treatment plan. Diabetic retinopathy grading is the task of classifying images of the eye’s fundus of diabetic patients into 5 different grades ranging from 0–4 based on the severity of the disease. In this work, we propose a deep neural network architecture to address the grading problem. The method utilizes additional attention layer in the neural network model to capture the spatial relationship between the region of interests in the images during the training process to better discriminate between the different severity stage of the disease. Also, we analyze the impact of different image processing techniques on the classification results. We assessed the performance of our proposed method using a dataset of eye fundus images and obtained classification accuracy of 89.20% on average. This performance surpass that reported for other state-of-the-art methods on the same dataset. The effectiveness of the proposed method will facilitate the procedural workflow of identifying severe cases of diabetic retinopathy.