A Deep Learning Based Automatic Severity Detector for Diabetic Retinopathy

• Published in: Machine Learning and Data Mining in Pattern Recognition Vol. 10934; pp. 64 - 76
• Main Authors: AlSaad, Rawan, Al-maadeed, Somaya, Al Mamun, Md. Abdullah, Boughorbel, Sabri
• Format: Book Chapter
• Language: English
• Published: Switzerland Springer International Publishing AG 2018; Springer International Publishing
• Series: Lecture Notes in Computer Science
• Subjects: Convolutional Neural Networks; Deep learning; Diabetic retinopathy; Medical imaging
• ISBN: 3319961357; 9783319961354
• ISSN: 0302-9743; 1611-3349
• DOI: 10.1007/978-3-319-96136-1_6

Automated Diabetic Retinopathy (DR) screening methods with high accuracy have the strong potential to assist doctors in evaluating more patients and quickly routing those who need help to a specialist. In this work, we used Deep Convolutional Neural Network architecture to diagnosing DR from digital fundus images and accurately classifying its severity. We train this network using a graphics processor unit (GPU) on the publicly available Kaggle dataset. We used Theano, Lasagne, and cuDNN libraries on two Amazon EC2 p2.xlarge instances and demonstrated impressive results, particularly for a high-level classification task. On the dataset of 30,262 training images and 4864 testing images, our model achieves an accuracy of 72%. Our experimental results showed that increasing the batch size does not necessarily speed up the convergence of the gradient computations. Also, it demonstrated that the number and size of fully connected layers do not have a significant impact on the performance of the model.