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

Arunava Chakravarty,Jayanthi Sivswamy
DOI: https://doi.org/10.48550/arXiv.1808.01355
2018-07-29
Abstract: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.
Computer Vision and Pattern Recognition,Machine Learning
What problem does this paper attempt to address?
This paper aims to solve the problem of early detection of glaucoma. Specifically, the authors propose a multi - task Convolutional Neural Network (CNN), which can simultaneously segment the Optic Disc (OD) and Optic Cup (OC) and predict the presence of glaucoma in color fundus images. By combining image appearance features and structural features obtained from OD - OC segmentation, this method can be trained on a limited data set and performs well on an independent validation set, showing its potential as a large - scale screening tool. ### Main problems and solutions 1. **Problem**: Glaucoma is a chronic retinal disease that causes irreversible vision loss. It is asymptomatic in the early stage, so large - scale screening is required to prevent its progression. 2. **Solutions**: - **Multi - task CNN architecture**: This network can simultaneously complete three tasks - the segmentation of OD and OC and the image - level prediction of glaucoma. - **Feature sharing**: By sharing CNN features among multiple related tasks, the generalization ability of the network is improved and the risk of overfitting is reduced. - **Small - scale network**: Compared with existing methods, this network has fewer parameters and is suitable for training with a small amount of data. - **Robustness**: This method performs well on images of different cameras and resolutions, indicating that it has strong robustness. ### Experimental results - **Segmentation performance**: In cross - validation, the Dice coefficient of OD is 0.96 and that of OC is 0.86; in the independent validation set, the Dice coefficient of OD is 0.92 and that of OC is 0.84. - **Classification performance**: In cross - validation, the AUC is 0.96, the sensitivity is 0.88, and the specificity is 0.91; in the independent validation set, the AUC is 0.95. ### Conclusion The multi - task CNN architecture proposed in this study performs excellently in the early detection of glaucoma, has high segmentation accuracy and classification performance, and is suitable for large - scale screening.