The use of U-Net lite and Extreme Gradient Boost (XGB) for glaucoma detection

Afolabi, OJMabuza-Hocquet, Gugulethu PNelwamondo, Fulufhelo VPaul, BS2021-07-132021-07-132021-03Afolabi, O., Mabuza-Hocquet, G.P., Nelwamondo, F.V. & Paul, B. 2021. The use of U-Net lite and Extreme Gradient Boost (XGB) for glaucoma detection. IEEE Access, 9. http://hdl.handle.net/10204/120482169-3536DOI: 10.1109/ACCESS.2021.3068204http://hdl.handle.net/10204/12048Glaucoma has been credited to be the foremost cause of preventable loss of sight in the world second only to cataract. Its effect on the eye is usually irreversible and can only be prevented by early detection. In this paper, we developed a glaucoma detection technique. This technique includes a modified U-Net model called ‘U-Net lite’ and an extreme gradient boost (XGB) algorithm. The novel U-Net lite model is designed to have fewer parameters than the original U-Net model. The U-Net lite’s parameters are 40 times fewer than the original U-Net model which makes the proposed model faster and cheaper to train. The proposed model is utilized to segment both the optic cup and the optic disc from the fundus images. The extreme gradient boost algorithm is utilized to analyze extracted features from segmented optic cups and discs and hence detect glaucoma. The proposed U-Net lite model was both trained and tested on the DRIONS, DRISHTI-GS, RIM-ONE V2 and the RIM-ONE V3 databases. When tested for optic disc segmentation on the four databases, the model achieved the following average dice-scores: 0.96 on RIM-ONE V3, 0.97 on RIM-ONE V2, 0.96 on DRIONS, and 0.97 on DRISHTI-GS. The XGB algorithm achieved an accuracy of 88.6% and an AUC-ROC of 93.6 % in detecting glaucoma from the RIM-ONE V3 and DRISHTI-GS database. The proposed glaucoma detection technique achieves a state-of-the-art accuracy and is useful for observing structural changes in an optic cup and optic disc.FulltextenFundus imageGlaucomaSegmentationU-NetExtreme Gradient BoostXGBThe use of U-Net lite and Extreme Gradient Boost (XGB) for glaucoma detectionArticleAfolabi, O., Mabuza-Hocquet, G. P., Nelwamondo, F. V., & Paul, B. (2021). The use of U-Net lite and Extreme Gradient Boost (XGB) for glaucoma detection. IEEE Access, 9, http://hdl.handle.net/10204/12048Afolabi, OJ, Gugulethu P Mabuza-Hocquet, Fulufhelo V Nelwamondo, and BS Paul "The use of U-Net lite and Extreme Gradient Boost (XGB) for glaucoma detection." IEEE Access, 9 (2021) http://hdl.handle.net/10204/12048Afolabi O, Mabuza-Hocquet GP, Nelwamondo FV, Paul B. The use of U-Net lite and Extreme Gradient Boost (XGB) for glaucoma detection. IEEE Access, 9. 2021; http://hdl.handle.net/10204/12048.TY - Article AU - Afolabi, OJ AU - Mabuza-Hocquet, Gugulethu P AU - Nelwamondo, Fulufhelo V AU - Paul, BS AB - Glaucoma has been credited to be the foremost cause of preventable loss of sight in the world second only to cataract. Its effect on the eye is usually irreversible and can only be prevented by early detection. In this paper, we developed a glaucoma detection technique. This technique includes a modified U-Net model called ‘U-Net lite’ and an extreme gradient boost (XGB) algorithm. The novel U-Net lite model is designed to have fewer parameters than the original U-Net model. The U-Net lite’s parameters are 40 times fewer than the original U-Net model which makes the proposed model faster and cheaper to train. The proposed model is utilized to segment both the optic cup and the optic disc from the fundus images. The extreme gradient boost algorithm is utilized to analyze extracted features from segmented optic cups and discs and hence detect glaucoma. The proposed U-Net lite model was both trained and tested on the DRIONS, DRISHTI-GS, RIM-ONE V2 and the RIM-ONE V3 databases. When tested for optic disc segmentation on the four databases, the model achieved the following average dice-scores: 0.96 on RIM-ONE V3, 0.97 on RIM-ONE V2, 0.96 on DRIONS, and 0.97 on DRISHTI-GS. The XGB algorithm achieved an accuracy of 88.6% and an AUC-ROC of 93.6 % in detecting glaucoma from the RIM-ONE V3 and DRISHTI-GS database. The proposed glaucoma detection technique achieves a state-of-the-art accuracy and is useful for observing structural changes in an optic cup and optic disc. DA - 2021-03 DB - ResearchSpace DP - CSIR J1 - IEEE Access, 9 KW - Fundus image KW - Glaucoma KW - Segmentation KW - U-Net KW - Extreme Gradient Boost KW - XGB LK - https://researchspace.csir.co.za PY - 2021 SM - 2169-3536 T1 - The use of U-Net lite and Extreme Gradient Boost (XGB) for glaucoma detection TI - The use of U-Net lite and Extreme Gradient Boost (XGB) for glaucoma detection UR - http://hdl.handle.net/10204/12048 ER -24664