Automated segmentation of optic disc and optic cup for glaucoma assessment using improved UNET++ architecture

Akshat Tulsani, Preetham Kumar, Sumaiya Pathan

Research output: Contribution to journalArticlepeer-review

Abstract

Glaucoma is one of the leading cause of blindness for over 60 million people around the world. Since a cure for glaucoma doesn't yet exist, early screening and diagnosis become critical for the prevention of the disease. Optic disc and optic cup evaluation are one of the preeminent steps for glaucoma diagnosis. A novel approach is developed in this paper for the identification of glaucoma using a segmentation based approach on the optic disc and optic cup. The Dhristi dataset was used to help improve performance on a small dataset. A custom UNET++ model is built for the segmentation task by tuning the hyper-parameters in addition to a custom loss function. The developed loss function helps tackle the class imbalance occurring due to small size of the optic nerve head. The proposed approach achieves 96% accuracy in classifying glaucomatous and non-glaucomatous images based on clinical feature identification. The improvised model is able to achieve state-of-art results for Intersection over Union (IOU) scores, 0.9477 for optic disc and 0.9321 for optic cup, along with providing an enhancement in reducing the training time. The model was tested on publicly available datasets RIM-ONE, DRIONS-DB and ORIGA and is able to achieve an accuracy of 91%, 92% and 90% respectively. The developed approach is validated by training it over RIM-ONE dataset independently, without changing any model parameters. The model provides significant improvement in segmentation of the optic disc and optic cup along with improvement in training time.

Original languageEnglish
Pages (from-to)819-832
Number of pages14
JournalBiocybernetics and Biomedical Engineering
Volume41
Issue number2
DOIs
Publication statusPublished - 01-04-2021

All Science Journal Classification (ASJC) codes

  • Biomedical Engineering

Fingerprint

Dive into the research topics of 'Automated segmentation of optic disc and optic cup for glaucoma assessment using improved UNET++ architecture'. Together they form a unique fingerprint.

Cite this