Weakly-supervised localization of diabetic retinopathy lesions in retinal fundus images

Convolutional neural networks (CNNs) show impressive performance for image classification and detection, extending heavily to the medical image domain. Nevertheless, medical experts are skeptical in these predictions as the nonlinear multilayer structure resulting in a classification outcome is not directly graspable. Recently, approaches have been shown which help the user to understand the discriminative regions within an image which are decisive for the CNN to conclude to a certain class. Although these approaches could help to build trust in the CNNs predictions, they are only slightly shown to work with medical image data which often poses a challenge as the decision for a class relies on different lesion areas scattered around the entire image. Using the DiaretDB1 dataset, we show that on retina images different lesion areas fundamental for diabetic retinopathy are detected on an image level with high accuracy, comparable or exceeding supervised methods. On lesion level, we achieve few false positives with high sensitivity, though, the network is solely trained on image-level labels which do not include information about existing lesions. Classifying between diseased and healthy images, we achieve an AUC of 0.954 on the DiaretDB1.

[1]  Sergey Ioffe,et al.  Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift , 2015, ICML.

[2]  Alireza Osareh,et al.  A Computational-Intelligence-Based Approach for Detection of Exudates in Diabetic Retinopathy Images , 2009, IEEE Transactions on Information Technology in Biomedicine.

[3]  Gretchen A. Stevens,et al.  Causes of vision loss worldwide, 1990-2010: a systematic analysis. , 2013, The Lancet. Global health.

[4]  Jie Chen,et al.  A location-to-segmentation strategy for automatic exudate segmentation in colour retinal fundus images , 2017, Comput. Medical Imaging Graph..

[5]  Samarendra Dandapat,et al.  A Gaussian Scale Space Approach For Exudates Detection, Classification And Severity Prediction , 2015, ArXiv.

[6]  Jorge A Cuadros,et al.  EyePACS: An Adaptable Telemedicine System for Diabetic Retinopathy Screening , 2009, Journal of diabetes science and technology.

[7]  Ivan Laptev,et al.  Is object localization for free? - Weakly-supervised learning with convolutional neural networks , 2015, 2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[8]  J. Boyce,et al.  Automated detection of diabetic retinopathy in digital retinal images: a tool for diabetic retinopathy screening , 2004, Diabetic medicine : a journal of the British Diabetic Association.

[9]  Hiroshi Fujita,et al.  CAD scheme to detect hemorrhages and exudates in ocular fundus images , 2007, SPIE Medical Imaging.

[10]  T. Williamson,et al.  Automatic detection of diabetic retinopathy using an artificial neural network: a screening tool. , 1996, The British journal of ophthalmology.

[11]  B. Schmauch,et al.  Deep learning approach for diabetic retinopathy screening , 2016 .

[12]  Sven Loncaric,et al.  Detection of exudates in fundus photographs using convolutional neural networks , 2015, 2015 9th International Symposium on Image and Signal Processing and Analysis (ISPA).

[13]  Philip J. Morrow,et al.  Algorithms for digital image processing in diabetic retinopathy , 2009, Comput. Medical Imaging Graph..

[14]  Bolei Zhou,et al.  Object Detectors Emerge in Deep Scene CNNs , 2014, ICLR.

[15]  Ştefan Ţălu,et al.  Characterisation of human non-proliferative diabetic retinopathy using the fractal analysis. , 2015, International journal of ophthalmology.

[16]  Rob Fergus,et al.  Visualizing and Understanding Convolutional Networks , 2013, ECCV.

[17]  Arunkumar Rajendran,et al.  Multi-retinal disease classification by reduced deep learning features , 2017, Neural Computing and Applications.

[18]  Gwénolé Quellec,et al.  Deep image mining for diabetic retinopathy screening , 2016, Medical Image Anal..

[19]  Xiangqian Wu,et al.  Retinal Microaneurysms Detection Using Gradient Vector Analysis and Class Imbalance Classification , 2016, PloS one.

[20]  Subhashini Venugopalan,et al.  Development and Validation of a Deep Learning Algorithm for Detection of Diabetic Retinopathy in Retinal Fundus Photographs. , 2016, JAMA.

[21]  Bolei Zhou,et al.  Learning Deep Features for Discriminative Localization , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[22]  Vijay M. Mane,et al.  Detection of Red lesions in diabetic retinopathy affected fundus images , 2015, 2015 IEEE International Advance Computing Conference (IACC).

[23]  Alexander Binder,et al.  Evaluating the Visualization of What a Deep Neural Network Has Learned , 2015, IEEE Transactions on Neural Networks and Learning Systems.

[24]  S. Resnikoff,et al.  The number of ophthalmologists in practice and training worldwide: a growing gap despite more than 200 000 practitioners , 2012, British Journal of Ophthalmology.

[25]  Andrew Zisserman,et al.  Deep Inside Convolutional Networks: Visualising Image Classification Models and Saliency Maps , 2013, ICLR.

[26]  Joni-Kristian Kämäräinen,et al.  The DIARETDB1 Diabetic Retinopathy Database and Evaluation Protocol , 2007, BMVC.

[27]  Anurag Mittal,et al.  Automated feature extraction for early detection of diabetic retinopathy in fundus images , 2009, 2009 IEEE Conference on Computer Vision and Pattern Recognition.

[28]  Marios S. Pattichis,et al.  Multiscale AM-FM Methods for Diabetic Retinopathy Lesion Detection , 2010, IEEE Transactions on Medical Imaging.

[29]  Jie Yang,et al.  Automatic hemorrhage detection in color fundus images based on gradual removal of vascular branches , 2016, 2016 IEEE International Conference on Image Processing (ICIP).

[30]  Mrinal Haloi,et al.  Improved Microaneurysm Detection using Deep Neural Networks , 2015, ArXiv.

[31]  Dragomir Anguelov,et al.  Self-taught object localization with deep networks , 2014, 2016 IEEE Winter Conference on Applications of Computer Vision (WACV).

[32]  B. Klein,et al.  Global Prevalence and Major Risk Factors of Diabetic Retinopathy , 2012, Diabetes Care.

[33]  D P Chakraborty,et al.  Maximum likelihood analysis of free-response receiver operating characteristic (FROC) data. , 1989, Medical physics.