A Deep Learning based Joint Segmentation and Classification Framework for Glaucoma Assesment in Retinal Color Fundus Images

Automated Computer Aided diagnostic tools can be used for the early detection of glaucoma to prevent irreversible vision loss. In this work, we present a Multi-task Convolutional Neural Network (CNN) that jointly segments the Optic Disc (OD), Optic Cup (OC) and predicts the presence of glaucoma in color fundus images. The CNN utilizes a combination of image appearance features and structural features obtained from the OD-OC segmentation to obtain a robust prediction. The use of fewer network parameters and the sharing of the CNN features for multiple related tasks ensures the good generalizability of the architecture, allowing it to be trained on small training sets. The cross-testing performance of the proposed method on an independent validation set acquired using a different camera and image resolution was found to be good with an average dice score of 0.92 for OD, 0.84 for OC and AUC of 0.95 on the task of glaucoma classification illustrating its potential as a mass screening tool for the early detection of glaucoma.

[1]  Tien Yin Wong,et al.  Superpixel Classification Based Optic Disc and Optic Cup Segmentation for Glaucoma Screening , 2013, IEEE Transactions on Medical Imaging.

[2]  Jayanthi Sivaswamy,et al.  Joint optic disc and cup boundary extraction from monocular fundus images , 2017, Comput. Methods Programs Biomed..

[3]  Tien Yin Wong,et al.  Similarity regularized sparse group lasso for cup to disc ratio computation. , 2017, Biomedical optics express.

[4]  Jayanthi Sivaswamy,et al.  Automated Detection of Glaucoma From Topographic Features of the Optic Nerve Head in Color Fundus Photographs , 2016, Journal of glaucoma.

[5]  U. Rajendra Acharya,et al.  Automated Diagnosis of Glaucoma Using Empirical Wavelet Transform and Correntropy Features Extracted From Fundus Images , 2017, IEEE Journal of Biomedical and Health Informatics.

[6]  Jayanthi Sivaswamy,et al.  RACE-Net: A Recurrent Neural Network for Biomedical Image Segmentation , 2019, IEEE Journal of Biomedical and Health Informatics.

[7]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[8]  Jayanthi Sivaswamy,et al.  Glaucoma classification with a fusion of segmentation and image-based features , 2016, 2016 IEEE 13th International Symposium on Biomedical Imaging (ISBI).

[9]  Xiaochun Cao,et al.  Joint Optic Disc and Cup Segmentation Based on Multi-Label Deep Network and Polar Transformation , 2018, IEEE Transactions on Medical Imaging.

[10]  Lizhuang Ma,et al.  Color transfer in correlated color space , 2006, VRCIA '06.

[11]  Jayanthi Sivaswamy,et al.  A Comprehensive Retinal Image Dataset for the Assessment of Glaucoma from the Optic Nerve Head Analysis , 2015 .

[12]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[13]  Xiaochun Cao,et al.  Disc-Aware Ensemble Network for Glaucoma Screening From Fundus Image , 2018, IEEE Transactions on Medical Imaging.

[14]  Jayanthi Sivaswamy,et al.  Coupled Sparse Dictionary for Depth-Based Cup Segmentation from Single Color Fundus Image , 2014, MICCAI.

[15]  Thomas Brox,et al.  U-Net: Convolutional Networks for Biomedical Image Segmentation , 2015, MICCAI.