Home / Current Issue / Paper 1705807
Early Detection of Diabetic Retinopathy with Segmentation Model U-Net
Subject area: Science,Engineering and Technology · Area of research: Artificial Intelligence and Data Science
Abstract
Diabetic Retinopathy (DR) is a serious diabetes complication leading to irreversible vision loss, thus emphasizing the need for early detection and intervention. Leveraging advancements in deep learning, this study investigates the application of a segmentation model based on the U-Net architecture for the early detection of DR. The U-Net model, renowned for its efficacy in semantic segmentation tasks, offers a promising approach to accurately delineating retinal structures indicative of DR-related lesions. Through comprehensive experimentation and evaluation, including the comparison of a standard U-Net segmentation model and U-Net with a VGG16 pre-trained encoder, termed U-NetVGG16, the models' proficiency in achieving high accuracy, precision, recall, Intersection over Union (IoU), and F1-score metrics was demonstrated. U-NetVGG16 excelled, earning a notable IoU of 98.62%, an accuracy of 99.10%, a precision of 99.30%, a recall of 99.40%, and an f1-score of 99.30%. The results highlight the potential of deep learning-based segmentation models in revolutionizing diabetic eye care by facilitating automated and precise identification of DR-related abnormalities. This study contributes to advancing the field of early DR detection, aiming to mitigate the global burden of preventable vision impairment associated with this debilitating condition.
References
[1] Dutta, A., Agarwal, P., Mittal, A. and Khandelwal, S. (2021). Detecting grades of diabetic retinopathy by extraction of retinal lesions using digital fundus images. Research on Biomedical Engineering, 37(4), pp.641–656. doi:https://doi.org/10.1007/s42600-021-00177-w.
[2] Who.int. (2020). World Diabetes Day 2020: Introducing the Global Diabetes Compact. [online] Available at: https://www.who.int/news-room/events/detail/2020/11/14/default-calendar/world-diabetes-day-2020-introducing-the-global-diabetes-compact.
[3] Goh, J.K.H., Cheung, C.Y., Sim, S.S., Tan, P.C., Tan, G.S.W. and Wong, T.Y. (2016). Retinal Imaging Techniques for Diabetic Retinopathy Screening. Journal of Diabetes Science and Technology, [online] 10(2), pp.282–294. doi:https://doi.org/10.1177/1932296816629491.
[4] Athanasios Valavanidis(2023) 'Artificial Intelligence in Medical Diagnostics and Imaging. Applications that will revolutionize the fields in biomedical research and healthcare’. Available at: https://www.researchgate.net/publication/375714812_Artificial_Intelligence_in_Medical_Diagnostics_and_Imaging_Applications_that_will_revolutionize_the_fields_in_biomedical_research_and_healthcare (Accessed: [January, 2024]).
[5] Xiao-Xia Yin, Le Sun, Yuhan Fu, Ruiliang Lu & Yanchun Zhang. (2022) 'U-Net-Based Medical Image Segmentation'. Available at: https://www.researchgate.net/publication/359998197_U-Net-Based_Medical_Image_Segmentation (Accesed:[January,2024]).
[6] Anas Bilal, Liucun Zhu, Anan Deng, Huihui Lu, Ning Wu. (2022). AI-Based Automatic Detection and Classification of Diabetic Retinopathy Using U-Net and Deep Learning. Journal of Computational and Theoretical Nanoscience, 14(7), 1427-1437.
[7] Revathy, R., Nithya, B. S., Reshma, J. J., Ragendhu, S. S., & Sumithra, M. D. (2020) 'Diabetic Retinopathy Detection using Machine Learning'. Available at: https://www.researchgate.net/publication/342120641_Diabetic_Retinopathy_Detection_using_Machine Learning (Accessed: [August,2023]).
[8] Parthasharathi, G. U., Kumar, K. V., Nivas, R. P., & Kj, J. (2022) 'Diabetic Retinopathy Detection Using Machine Learning'. Available at: https://www.researchgate.net/publication/360649393_Diabetic_Retinopathy_Detection_Using_Machine_Learning (Accessed: [August 2023]).
[9] Gunasekaran, K., Pitchai, R., Chaitanya, G.K., Selvaraj, D., Annie Sheryl, S., Almoallim, H.S., Alharbi, S.A., Raghavan, S.S. and Tesemma, B.G. (2022). A Deep Learning Framework for Earlier Prediction of Diabetic Retinopathy from Fundus Photographs. BioMed Research International, 2022, pp.1–15. doi:https://doi.org/10.1155/2022/3163496.
[10] Rakhlin, A. (2018). Diabetic Retinopathy detection through integration of Deep Learning classification framework. doi:https://doi.org/10.1101/225508.
[11] Mujeeb Rahman K K, Mohamed Nasor, and Ahmed Imran (2022). Automatic Screening of Diabetic Retinopathy Using Fundus Images and Machine Learning Algorithms, doi:https://doi.org/10.3390/diagnostics12092262.
[12] Nathan Zhang (2022). Predicting Diabetic Retinopathy Using Machine Learning. Journal of Student Research, 11(4). doi:https://doi.org/10.47611/jsrhs.v11i4.3179.
[13] Thippa Reddy Gadekallu, Neelu Khare, Sweta Bhattacharya and Saurabh Singh (2020). Early Detection of Diabetic Retinopathy Using PCA-Firefly Based Deep Learning Model. Electronics, 9(2), p.274. doi:https://doi.org/10.3390/electronics9020274.
[14] Mohamed Mahmoud, Salman Alamery, Hassan Fouad, Amir Altinawi, and Ahmed Youssef, (2021). An automatic detection system of diabetic retinopathy using a hybrid inductive machine learning algorithm. Personal and Ubiquitous Computing. doi:https://doi.org/10.1007/s00779-020-01519-8.
[15] Penikalapati Pragathi and Agastyaraju Nagaraja Rao (2022) An effective integrated machine learning approach for detecting diabetic retinopathy. Available at: https://www.researchgate.net/publication/359162008_An_effective_integrated_machine_learning_approach_for_detecting_diabetic_retinopathy
[16] Diabetes Retinopathy Dataset, 2021. Available at: Diabetic_Retinopathy_Balanced | Kaggle (Accessed: [August 2023]).
[17] Baccouch, W., Oueslati, S., Solaiman, B. and Labidi, S. (2023). A comparative study of CNN and U-Net performance for automatic segmentation of medical images: application to cardiac MRI. Procedia Computer Science, 219, pp.1089–1096. doi:https://doi.org/10.1016/j.procs.2023.01.388.
[18] Awf Abd & Muhammet Baykara (2022) 'A Novel Approach to Detect COVID-19: Enhanced Deep Learning Models with Convolutional Neural Networks'. Available at: https://www.researchgate.net/publication/363656696_A_Novel_Approach_to_Detect_COVID-19_Enhanced_Deep_Learning_Models_with_Convolutional_Neural_Networks (Accessed at [Febuary 2024]).
[19] TensorFlow. (n.d.). Module: tf.keras | TensorFlow Core v2.4.1. [online] Available at: https://www.tensorflow.org/api_docs/python/tf/keras [Accessed 7 Dec. 2023].
[20] Soulami, K.B., Kaabouch, N., Saidi, M.N. and Tamtaoui, A. (2021). Breast cancer: One-stage automated detection, segmentation, and classification of digital mammograms using UNet model based-semantic segmentation. Biomedical Signal Processing and Control, 66, p.102481. doi:https://doi.org/10.1016/j.bspc.2021.102481.
[21] Tejpal Virdl, John T. Guibas, Peter S. Li (2017), Synthetic Medical Images from Dual Generative Adversarial Networks. (PDF) Synthetic Medical Images from Dual Generative Adversarial Networks (researchgate.net). Available at https://www.researchgate.net/publication/319524751_Synthetic_Medical_Images_from_Dual_Generative_Adversarial_Networks
[22] Shah, A. D., Jain, S. M., & Patel, M. P. Automatic Screening of Diabetic Retinopathy Using Fundus Images and Machine Learning Algorithms. Available at: https://www.researchgate.net/publication/363674594_Automatic_Screening_of_Diabetic_Retinopathy_Using_Fundus_Images_and_Machine_Learning_Algorithms
How to cite this paper
@article{1705807,
author = {Adeyinka Mayowa-Majaro},
title = {Early Detection of Diabetic Retinopathy with Segmentation Model U-Net},
journal = {Iconic Research And Engineering Journals},
year = {2024},
volume = {7},
number = {11},
pages = {337-343},
issn = {2456-8880},
url = {https://www.irejournals.com/formatedpaper/1705807.pdf},
abstract = {Diabetic Retinopathy (DR) is a serious diabetes complication leading to irreversible vision loss, thus emphasizing the need for early detection and intervention. Leveraging advancements in deep learning, this study investigates the application of a segmentation model based on the U-Net architecture for the early detection of DR. The U-Net model, renowned for its efficacy in semantic segmentation tasks, offers a promising approach to accurately delineating retinal structures indicative of DR-related lesions. Through comprehensive experimentation and evaluation, including the comparison of a standard U-Net segmentation model and U-Net with a VGG16 pre-trained encoder, termed U-NetVGG16, the models' proficiency in achieving high accuracy, precision, recall, Intersection over Union (IoU), and F1-score metrics was demonstrated. U-NetVGG16 excelled, earning a notable IoU of 98.62%, an accuracy of 99.10%, a precision of 99.30%, a recall of 99.40%, and an f1-score of 99.30%. The results highlight the potential of deep learning-based segmentation models in revolutionizing diabetic eye care by facilitating automated and precise identification of DR-related abnormalities. This study contributes to advancing the field of early DR detection, aiming to mitigate the global burden of preventable vision impairment associated with this debilitating condition.},
month = {May},
}