A Mobile Deep Learning Classification Model for Diabetic Retinopathy

Abstract

The pupil, iris, vitreous, and retina are parts of the eye, where any defect due to physical damage or chronic diseases to these parts of the eye can lead to partial vision loss or complete blindness. Changes in retinal structure due to diabetes or high blood pressure lead to diabetic retinopathy (DR). The early diagnosis of DR using computer-aided automated tools is possible due to tremendous advancements in machine and deep learning models in the last decade. Devising and implementing innovative deep learning models for retinal structural analysis is crucial to the early diagnosis of DR and other eye diseases. In this work, we have developed a new approach, which involves the development of a lightweight convolutional neural network (CNN)-based model for segmentation of retinal vessels and a mobile application for DR grading. This paper covers the development process of an Android application that leverages the power of CNN-based deep learning model to detect DR regardless of its stage. To achieve this, two models have been created and compared, the best one having an accuracy of 96.72 %. An Android application has then been developed, that makes calls to this model and then displays the results on screen with a simple-to-understand interface developed using the Kivy framework.

Keywords

• diabetic retinopathy
• deep neural networks
• machine learning
• retinal vessel segmentation