Diabetic retinopathy stage detection using convolutional fine-tuned transfer Learning model
Abstract
Diabetic Retinopathy (DR) is a prevalent eye condition that occurs as a frequent complication among individuals with diabetes, particularly those who have been living with the disease for an extended period of time. This study uses fundus images to diagnose DR at five stages from early to late with No DR, Mild, Moderate, Severe, and Proliferative DR, commonly known as Stage 0 to Stage 4, respectively. This will aid in the timely treatment of diabetic patients preventing them from developing DR as early as possible. We used two most popular open-source datasets, the DR Detection database, namely APTOS 2019 and EyePACS, and combined them to create a larger dataset to trade off the data-centric obstacle and shortfall for any Deep Learning-based prediction models. Data augmentation and preprocessing techniques are applied to the images before feeding them to the proposed model to get a more accurate and efficient one. In the modern age oriented to Artificial Intelligence (AI), it is necessary to thoroughly analyze the identification of DR based on the existing Deep Learning (DL) models. After learning about the limitations of existing models, we have fine-tuned the ResNet50, DenseNet201 and InceptionV3 to enhance the model performance of the detection and categorization of DR. We have since proposed three Deep Convolutional Neural Networks (DCNN) models with better outcome based on accuracy than the existing state-of-the-art (SOTA) models. The fine-tuned DenseNet201 model, among the other two, performed significantly better with a validation accuracy of 90.04% and a negligible amount of loss, irrespective of each class, under the best configurable test conditions.
References
Carin, L., & Pencina, M.J. (2018). On deep learning for medical image analysis. Jama, 320(11), 1192-1193. https://doi.org/10.1001/jama.2018.13316
Gabriel, G., Jhair, G., Antoni, M., Jorge, L., & Christian, D.C. (2017). Detection of diabetic retinopathy based on a convolutional neural network using retinal fundus images. Artificial Neural Networks and Machine Learning–ICANN 2017: 26th International Conference on Artificial Neural Networks, Alghero, Italy, September 11-14, 2017, Proceedings, Part II 26, Springer International Publishing, pp. 635-642. https://doi.org/10.1007/978-3-319-68612-7_72
Glorindal, G., Mozhiselvi, S.A., Kumar, T.A., Kumaran, K., Katema, P.C., & Kandimba, T. (2021). A Simplified Approach for Melanoma Skin Disease Identification. IEEE, 2021 International Conference on System, Computation, Automation and Networking (ICSCAN), pp. 1-5. https://doi.org/10.1109/ICSCAN53069.2021.9526511
Islam, M.R., Abdulrazak, L.F., Nahiduzzaman, M., Goni, M.O.F., Anower, M.S., Ahsan, M., Haider, J., & Kowalski, M. (2022). Applying supervised contrastive learning for the detection of diabetic retinopathy and its severity levels from fundus images. Comput. Biol. Med., 146, 105602. https://doi.org/10.1016/j.compbiomed.2022.105602.
Jamil, S., Abbas, M.S., Umair, M., Habib, F., Fawad, & Hussain, Y. (2020). A Novel Deep Neural Network CanNet for Malignant Detection. IEEE, 2020 International Conference on Information Science and Communication Technology (ICISCT), pp. 1-5.
https://doi.org/10.1109/ICISCT49550.2020.9079937
Lahmar, C., & Ali, I. (2021). On the value of deep learning for diagnosing diabetic retinopathy. Health and Technology, 12(1), 89-105. https://doi.org/10.1007/s12553-021-00606-x
Lands, A., Kottarathil, A. J., Biju, A., Jacob, E. M., & Thomas, S. (2020). Implementation of deep learning based algorithms for diabetic retinopathy classification from fundus images. IEEE, 2020 4th International Conference on Trends in Electronics and Informatics (ICOEI)(48184), pp. 1028-1032. https://doi.org/10.1109/ICOEI48184.2020.9142878
Li, Y.H., Yeh, N.N., Chen, S.J., & Chung, Y.C. (2019). Computer-assisted diagnosis for diabetic retinopathy based on fundus images using deep convolutional neural network. Mobile Information Systems, 2019, 6142839:1-6142839:14.org/10.1155/2019/6142839
Munadi, K., Muchtar, K., Maulina, N., & Pradhan, B. (2020). Image Enhancement for Tuberculosis Detection Using Deep Learning. IEEE Access, 8, 217897-217907. https://doi.org/10.1109/ACCESS.2020.3041867
National Eye Institute. (2022). (National Eye Institute) Retrieved from https://www.nei.nih.gov/learn-about-eye-health/eye-conditions-and-diseases/diabetic-retinopathy
Nguyen, Q.H., Muthuraman, R., Singh, L., Sen, G., Tran, A. C., Nguyen, B.P., & Chua, M. (2020). Diabetic retinopathy detection using deep learning. Proceedings of the 4th International Conference on Machine Learning and Soft Computing, pp. 103-107. https://doi.org/10.1145/3380688.3380709
Olowononi, F. O., Rawat, D. B., & Liu, C. (2020). Resilient machine learning for networked cyber physical systems: A survey for machine learning security to securing machine learning for cps. IEEE Communications Surveys & Tutorials, 23(1), 524-552. https://doi.org/10.1109/COMST.2020.3036778
Pratt, H., Coenen, F., Broadbent, D. M., Harding, S. P., & Zheng, Y. (2016). Convolutional neural networks for diabetic retinopathy. Procedia Computer Science. 90, 200-205. https://doi.org/10.1016/j.procs.2016.07.014
Qummar, S., Khan, F., Shah, S., Khan, A., Shamshirband, S., Rehman, Z., Khan, I.A., & Jadoon, W. (2019). A deep learning ensemble approach for diabetic retinopathy detection. IEEE Access, 7, 150530-150539. https://doi.org/10.1109/ACCESS.2019.2947484
Retinopathy_train_2015. (2019). (Kaggle) Retrieved from Kaggle: https://www.kaggle.com/datasets/donkeys/retinopathy-train-2015
Sarki, R., Michalska, S., Ahmed, K., Wang, H., & Zhang, Y. (2019). Convolutional neural networks for mild diabetic retinopathy detection: an experimental study. BioRxiv, 763136 (pp.1-12). https://doi.org/10.1101/763136
Sayres, R., Taly, A., Rahimy, E., BS, K. B., MD, N. H., Krause, J., Narayanaswamy, A., Rastegar, Z., Wu, D., Xu, S., Barb, S., Joseph, A., Shumski, M., Smith, J., Sood, A.B., Corrado, G.S., Peng, L., & Webster, D.R. (2019). Using a deep learning algorithm and integrated gradients explanation to assist grading for diabetic retinopathy. Ophthalmology, 126(4), 552-564. https://doi.org/10.1016/j.ophtha.2018.11.016
Shin, H.C., Roth, H., Gao, M., Lu, L., Xu, Z., Nogues, I., Yao, J., Mollura, D., & Summers, R. (2016). Deep convolutional neural networks for computer-aided detection: CNN architectures, dataset characteristics and transfer learning. IEEE Transactions on Medical Imaging, 35(5), 1285-1298. https://doi.org/10.1109/TMI.2016.2528162
Teo, Z.L., Tham, Y.C., Yu, M., Chee, M.L., Rim , T.H., Cheung , N., Bikbov, M.M., Wang, Y.X., Tang, Y., Lu, Y., Wong, I.Y., Ting, D.S.W., Tan, G.S.W., Jonas, J.B., Sabanayagam, C., Wong, T.Y., & Cheng, C.Y. (2021). Global prevalence of diabetic retinopathy and projection of burden through 2045: systematic review and meta-analysis. Ophthalmology, 128(11), 1580-1591. https://doi.org/10.1016/j.ophtha.2021.04.027
Wang, X., Lu, Y., Wang, Y., & Chen, W.B. (2018). Diabetic retinopathy stage classification using convolutional neural networks. IEEE, 2018 IEEE International Conference on Information Reuse and Integration (IRI), pp. 465-471. https://doi.org/10.1109/IRI.2018.00074
Copyright (c) 2023 International Academic Publishing House (IAPH)
This work is licensed under a Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 International License.
