Automatic detection and classification of diabetic retinopathy stages using CNN

A Convolutional Neural Networks (CNNs) approach is proposed to automate the method of Diabetic Retinopathy(DR) screening using color fundus retinal photography as input. Our network uses CNN along with denoising to identify features like micro-aneurysms and haemorrhages on the retina. Our models were developed leveraging Theano, an open source numerical computation library for Python. We trained this network using a high-end GPU on the publicly available Kaggle dataset. On the data set of over 30,000 images our proposed model achieves around 95% accuracy for the two class classification and around 85% accuracy for the five class classification on around 3,000 validation images.

[1]  Shantala Giraddi,et al.  Identifying Abnormalities in the Retinal Images using SVM Classifiers , 2015 .

[2]  Max A. Viergever,et al.  Ridge-based vessel segmentation in color images of the retina , 2004, IEEE Transactions on Medical Imaging.

[3]  Pierre Baldi,et al.  Understanding Dropout , 2013, NIPS.

[4]  Nitish Srivastava,et al.  Dropout: a simple way to prevent neural networks from overfitting , 2014, J. Mach. Learn. Res..

[5]  Jian Sun,et al.  Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classification , 2015, 2015 IEEE International Conference on Computer Vision (ICCV).

[6]  Gongping Yang,et al.  Hierarchical retinal blood vessel segmentation based on feature and ensemble learning , 2015, Neurocomputing.

[7]  U. Rajendra Acharya,et al.  Computer-aided diagnosis of diabetic retinopathy: A review , 2013, Comput. Biol. Medicine.

[8]  Yoshua Bengio,et al.  Understanding the difficulty of training deep feedforward neural networks , 2010, AISTATS.

[9]  A. Baudes,et al.  A Nonlocal Algorithm for Image Denoising , 2005, CVPR 2005.

[10]  Jean-Michel Morel,et al.  A non-local algorithm for image denoising , 2005, 2005 IEEE Computer Society Conference on Computer Vision and Pattern Recognition (CVPR'05).

[11]  Jon Cornwall,et al.  The current state of diabetes mellitus in India. , 2014, The Australasian medical journal.

[12]  J. Forrester,et al.  Epidemiology of diabetic retinopathy and macular oedema: a systematic review , 2004, Eye.

[13]  D. Jeyakumari,et al.  Multiclass SVM-based automated diagnosis of diabetic retinopathy , 2013, 2013 International Conference on Communication and Signal Processing.

[14]  M. Engelgau,et al.  The Evolving Diabetes Burden in the United States , 2004, Annals of Internal Medicine.

[15]  Geoffrey E. Hinton,et al.  On the importance of initialization and momentum in deep learning , 2013, ICML.

[16]  Sven Loncaric,et al.  Retinal Vessel Segmentation using Deep Neural Networks , 2015, VISAPP.