Optimizing Glaucoma Diagnosis with Deep Learning-Based Segmentation and Classification of Retinal Images
Source abstract
Abstract
Glaucoma, a leading cause of permanent blindness worldwide, necessitates early detection to prevent vision loss, a task that is challenging and time-consuming when performed manually. This study proposes an automatic glaucoma detection method on enhanced retinal images using deep learning. The system analyzes retinal images, generating masks for the optic disc and optic cup, and providing a classification for glaucoma diagnosis. We employ a U-Net architecture with a pretrained residual neural network (ResNet34) for segmentation and an EfficientNetB0 for classification. The proposed framework is tested on publicly available datasets, including ORIGA, REFUGE, RIM-ONE DL, and HRF. Our work evaluated the U-Net model with five pretrained backbones (ResNet34, ResNet50, VGG19, DenseNet121, and EfficientNetB0) and examined preprocessing effects. We optimized model training with limited data using transfer learning and data augmentation techniques. The segmentation model achieves a mean intersection over union (mIoU) value of 0.98. The classification model shows remarkable performance with 99.9% training and 100% testing accuracy on ORIGA, 99.9% training and 99% testing accuracy on RIM-ONE DL, and 98% training and 100% testing accuracy on HRF. The proposed model outperforms related works and demonstrates potential for accurate glaucoma classification and detection tasks.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Retinal Imaging and Analysis · Digital Imaging for Blood Diseases · Glaucoma and retinal disorders
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Computer Science Department, College of Computer Science and Information Technology, King Faisal University, Al Ahsa 36291, Saudi Arabia
Funding
Deputyship for Research & Innovation, Ministry of Education in Saudi Arabia
References (29)
Show source-provided references
1. Segmentation and Classification of Glaucoma Using U-Net with Deep Learning Model10.1155/2022/1601354 ↗
2. Optic Disc and Optic Cup Segmentation for Glaucoma Detection from Blur Retinal Images Using Improved Mask-RCNN10.1155/2021/6641980 ↗
3. ODGNet: A Deep Learning Model for Automated Optic Disc Localization and Glaucoma Classification Using Fundus Images10.1007/s42452-022-04984-3 ↗
4. A Novel Optic Disc and Optic Cup Segmentation Technique to Diagnose Glaucoma Using Deep Learning Convolutional Neural Network over Retinal Fundus Images
5. Camara, J., Neto, A., Pires, I.M., Villasana, M.V., Zdravevski, E., and Cunha, A. (2022). Literature Review on Artificial Intelligence Methods for Glaucoma Screening, Segmentation, and Classification. J. Imaging, 8.10.3390/jimaging8020019 ↗
6. Deep Learning Techniques for Medical Image Segmentation: Achievements and Challenges10.1007/s10278-019-00227-x ↗
7. Kim, M., Han, J.C., Hyun, S.H., Janssens, O., Van Hoecke, S., Kee, C., and De Neve, W. (2019). Medinoid: Computer-Aided Diagnosis and Localization of Glaucoma Using Deep Learning †. Appl. Sci., 9.10.3390/app9153064 ↗
8. Norouzifard, M., Nemati, A., GholamHosseini, H., Klette, R., Nouri-Mahdavi, K., and Yousefi, S. (2018, January 19–21). Automated Glaucoma Diagnosis Using Deep and Transfer Learning: Proposal of a System for Clinical Testing. Proceedings of the 2018 International Conference on Image and Vision Computing (IVCNZ), Auckland, New Zealand.10.1109/IVCNZ.2018.8634671 ↗
9. Nawaz, M., Nazir, T., Javed, A., Tariq, U., Yong, H.-S., Khan, M.A., and Cha, J. (2022). An Efficient Deep Learning Approach to Automatic Glaucoma Detection Using Optic Disc and Optic Cup Localization. Sensors, 22.10.3390/s22020434 ↗
10. Maheshwari, S., Kanhangad, V., and Pachori, R.B. (2020). CNN-Based Approach for Glaucoma Diagnosis Using Transfer Learning and LBP-Based Data Augmentation. arXiv.
11. Mallick, S., Saha, N., Paul, J., Ganguli, I., Debnath, S., and Sil, J. (November, January 31). An Efficient Deep Learning Framework for Glaucoma Diagnosis Using Convolution Mixed Transformer Network. Proceedings of the TENCON 2023—2023 IEEE Region 10 Conference (TENCON), Chiang Mai, Thailand.10.1109/TENCON58879.2023.10322467 ↗
12. GlauNet: Glaucoma Diagnosis for OCTA Imaging Using a New CNN Architecture10.1109/ACCESS.2022.3204029 ↗
13. MTRA-CNN: A Multi-Scale Transfer Learning Framework for Glaucoma Classification in Retinal Fundus Images10.1109/ACCESS.2023.3342910 ↗
14. Glaucoma Identification Using Convolutional Neural Networks Ensemble for Optic Disc and Cup Segmentation10.1109/ACCESS.2024.3412185 ↗
15. Automated Tool Support for Glaucoma Identification with Explainability Using Fundus Images10.1109/ACCESS.2024.3359698 ↗
16. Zhang, Z., Yin, F.S., Liu, J., Wong, W.K., Tan, N.M., Lee, B.H., Cheng, J., and Wong, T.Y. (September, January 31). ORIGA-light: An Online Retinal Fundus Image Database for Glaucoma Analysis and Research. Proceedings of the 2010 Annual International Conference of the IEEE Engineering in Medicine and Biology, Buenos Aires, Argentina.
17. Budai, A., Bock, R., Maier, A., Hornegger, J., and Michelson, G. (2013). Robust Vessel Segmentation in Fundus Images. Int. J. Biomed. Imaging, 2013.10.1155/2013/154860 ↗
18. RIM-ONE DL: A Unified Retinal Image Database for Assessing Glaucoma Using Deep Learning10.5566/ias.2346 ↗
19. REFUGE Challenge: A Unified Framework for Evaluating Automated Methods for Glaucoma Assessment from Fundus Photographs10.1016/j.media.2019.101570 ↗
20. GC-NET for Classification of Glaucoma in the Retinal Fundus Image10.1007/s00138-020-01091-4 ↗
21. AI Applications to Medical Images: From Machine Learning to Deep Learning10.1016/j.ejmp.2021.02.006 ↗
22. Buslaev, A., Iglovikov, V.I., Khvedchenya, E., Parinov, A., Druzhinin, M., and Kalinin, A.A. (2020). Albumentations: Fast and Flexible Image Augmentations. Information, 11.10.3390/info11020125 ↗
23. Segmentation Models with Pretrained Backbones (2024, July 04). Keras and TensorFlow Keras. Available online: https://github.com/qubvel/segmentation_models.
24. Ansari, S. (2020). Building Computer Vision Applications Using Artificial Neural Networks, Apress.10.1007/978-1-4842-5887-3 ↗
25. Tan, M., and Le, Q.V. (2019). EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks. arXiv.
26. Goutte, C., and Gaussier, E. (2005, January 21–23). A Probabilistic Interpretation of Precision, Recall and F-Score, with Implication for Evaluation. Proceedings of the European Conference on Information Retrieval, Santiago de Compostela, Spain.10.1007/978-3-540-31865-1_25 ↗
27. Bharati, S., Khan, T.Z., Podder, P., and Hung, N.Q. (2021). A Comparative Analysis of Image Denoising Problem: Noise Models, Denoising Filters and Applications. Cognitive Internet of Medical Things for Smart Healthcare: Services and Applications, Springer.10.1007/978-3-030-55833-8_3 ↗
28. (2024, July 04). Colab Pro. Available online: https://colab.research.google.com/signup.
29. (2024, May 06). Gradio. Available online: https://www.gradio.app/.