A Deep Learning Model for Detecting Diabetic Retinopathy Stages with Discrete Wavelet Transform
Source abstract
Abstract
Diabetic retinopathy (DR) is the primary factor leading to vision impairment and blindness in diabetics. Uncontrolled diabetes can damage the retinal blood vessels. Initial detection and prompt medical intervention are vital in preventing progressive vision impairment. Today’s growing medical field presents a more significant workload and diagnostic demands on medical professionals. In the proposed study, a convolutional neural network (CNN) is employed to detect the stages of DR. This research is crucial for studying DR because of its innovative methodology incorporating two different public datasets. This strategy enhances the model’s capacity to generalize unseen DR images, as each dataset encompasses unique demographics and clinical circumstances. The network can learn and capture complicated hierarchical image features with asymmetric weights. Each image is preprocessed using contrast-limited adaptive histogram equalization and the discrete wavelet transform. The model is trained and validated using the combined datasets of Dataset for Diabetic Retinopathy and the Asia-Pacific Tele-Ophthalmology Society. The CNN model is tuned in with different learning rates and optimizers. An accuracy of 72% and an area under curve score of 0.90 was achieved by the CNN model with the Adam optimizer. The recommended study results may reduce diabetes-related vision impairment by early identification of DR severity.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Retinal Imaging and Analysis · Artificial Intelligence in Healthcare · Digital Imaging for Blood Diseases
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Department of Computer Engineering, Kuwait University, Safat 13060, Kuwait
Kuwait Specialized Eye Center, Kuwait City 35453, Kuwait
Funding
Kuwait University
References (70)
Show source-provided references
1. IDF Diabetes Atlas: Global, regional and country-level diabetes prevalence estimates for 2021 and projections for 204510.1016/j.diabres.2021.109119 ↗
2. World Bank Group (2023, October 19). World Development Indicators: Diabetes Prevalence (% of Population Ages 20 to 79). Available online: https://databank.worldbank.org/reports.aspx?dsid=2&series=SH.STA.DIAB.ZS.
3. Diabetic retinopathy: Looking forward to 203010.3389/fendo.2022.1077669 ↗
4. Diabetic retinopathy detection using principal component analysis multi-label feature extraction and classification
5. Diabetic Retinopathy Classification with Deep Learning via Fundus Images: A Short Survey10.1109/ACCESS.2024.3361944 ↗
6. Sun, R., Pang, Y., and Li, W. (2023). Efficient Lung Cancer Image Classification and Segmentation Algorithm Based on an Improved Swin Transformer. Electronics, 12.10.3390/electronics12041024 ↗
7. Jeong, Y., Hong, Y.J., and Han, J.H. (2022). Review of Machine Learning Applications Using Retinal Fundus Images. Diagnostics, 12.10.3390/diagnostics12010134 ↗
8. A Deep Learning Approach to Diabetic Retinopathy Classification10.1007/978-3-031-09073-8_36 ↗
9. Vipparthi, V., Rao, D.R., Mullu, S., and Patlolla, V. (2022, January 17–19). Diabetic Retinopathy Classification using Deep Learning Techniques. Proceedings of the 2022 3rd International Conference on Electronics and Sustainable Communication Systems (ICESC), Coimbatore, India.10.1109/ICESC54411.2022.9885687 ↗
10. Tripathy, B.K., Lingras, P., Kar, A.K., and Chowdhary, C.L. (2022). Next Generation Healthcare Informatics, Springer Nature.10.1007/978-981-19-2416-3 ↗
11. Tsiknakis, N., Theodoropoulos, D., Manikis, G., Ktistakis, E., Boutsora, O., Berto, A., Scarpa, F., Scarpa, A., Fotiadis, D.I., and Marias, K. (2021). Deep learning for diabetic retinopathy detection and classification based on fundus images: A review. Comput. Biol. Med., 135.10.1016/j.compbiomed.2021.104599 ↗
12. Diabetic retinopathy detection and stage classification in eye fundus images using active deep learning10.1007/s11042-020-10238-4 ↗
13. A Deep Learning Ensemble Approach for Diabetic Retinopathy Detection10.1109/ACCESS.2019.2947484 ↗
14. Jain, A.K., Jalui, A., Jasani, J., Lahoti, Y., and Karani, R. (2019, January 23). Deep Learning for Detection and Severity Classification of Diabetic Retinopathy. Proceedings of the 2019 1st International Conference on Innovations in Information and Communication Technology (ICIICT), Tamil Nadu, India.10.1109/ICIICT1.2019.8741456 ↗
15. Higuera, V. (2023, December 25). The 4 Stages of Diabetic Retinopathy. Available online: https://www.healthline.com/health/diabetes/diabetic-retinopathy-stages.
16. (2022, September 05). APTOS 2019 Blindness Detection. Available online: https://www.kaggle.com/competitions/aptos2019-blindness-detection/overview.
17. Application of deep learning approaches for classification of diabetic retinopathy stages from fundus retinal images: A survey10.1007/s11042-023-17254-0 ↗
18. Xu, S., Huang, Z., and Zhang, Y. (2022, November 21). Diabetic Retinopathy Progression Recognition Using Deep Learning Method. Available online: http://cs231n.stanford.edu/reports/2022/pdfs/20.pdf.
19. Mutawa, A.M., Alnajdi, S., and Sruthi, S. (2023). Transfer Learning for Diabetic Retinopathy Detection: A Study of Dataset Combination and Model Performance. Appl. Sci., 13.10.3390/app13095685 ↗
20. Deep Learning and Medical Image Processing Techniques for Diabetic Retinopathy: A Survey of Applications, Challenges, and Future Trends10.1155/2023/2728719 ↗
21. Automated identification of diabetic retinopathy using deep learning10.1016/j.ophtha.2017.02.008 ↗
22. A Comparative Study of Learning Methods for Diabetic Retinopathy Classification10.1007/978-981-16-8403-6_21 ↗
23. Diabetic retinopathy detection using VGG-NIN a deep learning architecture10.1109/ACCESS.2021.3074422 ↗
24. Diabetic retinopathy classification using VGG16 neural network10.1007/s42600-022-00200-8 ↗
25. Automatic Severity Classification of Diabetic Retinopathy Based on DenseNet and Convolutional Block Attention Module10.1109/ACCESS.2022.3165193 ↗
26. UNIConv: An enhanced U-Net based InceptionV3 convolutional model for DR semantic segmentation in retinal fundus images10.1002/cpe.7138 ↗
27. A novel deep transfer learning based computerized diagnostic Systems for Multi-class imbalanced diabetic retinopathy severity classification10.1007/s11042-023-14963-4 ↗
28. Qian, Z., Wu, C., Chen, H., and Chen, M. (2021, January 12–14). Diabetic Retinopathy Grading Using Attention based Convolution Neural Network. Proceedings of the 2021 IEEE 5th Advanced Information Technology, Electronic and Automation Control Conference (IAEAC), Chongqing, China.10.1109/IAEAC50856.2021.9390963 ↗
29. Zhang, C., Lei, T., and Chen, P. (2022). Diabetic retinopathy grading by a source-free transfer learning approach. Biomed. Signal Process. Control, 73.10.1016/j.bspc.2021.103423 ↗
30. Diabetic Retinopathy Classification Using Swin Transformer with Multi Wavelet10.31642/JoKMC/2018/100225 ↗
31. Automated detecting and severity grading of diabetic retinopathy using transfer learning and attention mechanism10.1007/s00521-023-09001-1 ↗
32. A multilevel deep feature selection framework for diabetic retinopathy image classification
33. A Novel Attention Based Few-shot Classification Framework for Diabetic Retinopathy Detection and Grading10.1016/j.measurement.2022.111485 ↗
34. Bilal, A., Zhu, L., Deng, A., Lu, H., and Wu, N. (2022). AI-Based Automatic Detection and Classification of Diabetic Retinopathy Using U-Net and Deep Learning. Symmetry, 14.10.3390/sym14071427 ↗
35. Jena, P.K., Khuntia, B., Palai, C., Nayak, M., Mishra, T.K., and Mohanty, S.N. (2023). A Novel Approach for Diabetic Retinopathy Screening Using Asymmetric Deep Learning Features. Big Data Cogn. Comput., 7.10.3390/bdcc7010025 ↗
36. A Transfer Learning and U-Net-based automatic detection of diabetic retinopathy from fundus images10.1080/21681163.2021.2021111 ↗
37. Deep Transfer Learning Models for Medical Diabetic Retinopathy Detection10.5455/aim.2019.27.327-332 ↗
38. Kandel, I., and Castelli, M. (2020). Transfer learning with convolutional neural networks for diabetic retinopathy image classification. A review. Appl. Sci., 10.10.3390/app10062021 ↗
39. A transfer learning with deep neural network approach for diabetic retinopathy classification
40. A Benchmark for Studying Diabetic Retinopathy: Segmentation, Grading, and Transferability10.1109/TMI.2020.3037771 ↗
41. The art of data augmentation10.1198/10618600152418584 ↗
42. A survey on image data augmentation for deep learning10.1186/s40537-019-0197-0 ↗
43. Mungloo-Dilmohamud, Z., Heenaye-Mamode Khan, M., Jhumka, K., Beedassy, B.N., Mungloo, N.Z., and Peña-Reyes, C. (2022). Balancing Data through Data Augmentation Improves the Generality of Transfer Learning for Diabetic Retinopathy Classification. Appl. Sci., 12.10.3390/app12115363 ↗
44. Ashwini, K., and Dash, R. (2023). Grading diabetic retinopathy using multiresolution based CNN. Biomed. Signal Process. Control., 86.10.1016/j.bspc.2023.105210 ↗
45. Wavelet-Attention Swin for Automatic Diabetic Retinopathy Classification10.21123/bsj.2024.8565 ↗
46. Development of retinal blood vessel segmentation methodology using wavelet transforms for assessment of diabetic retinopathy
47. Yagmur, F.D., Karlik, B., and Okatan, A. (2008, January 4–6). Automatic recognition of retinopathy diseases by using wavelet based neural network. Proceedings of the 2008 First International Conference on the Applications of Digital Information and Web Technologies (ICADIWT), Ostrava, Czech Republic.10.1109/ICADIWT.2008.4664391 ↗
48. Rehman, M.U., Abbas, Z., Khan, S.H., Ghani, S.H. (2018, January 5–6). Diabetic retinopathy fundus image classification using discrete wavelet transform. Proceedings of the 2018 2nd International Conference on Engineering Innovation (ICEI), Bangkok, Thailand.
49. Developments in the detection of diabetic retinopathy: A state-of-the-art review of computer-aided diagnosis and machine learning methods10.1007/s10462-022-10185-6 ↗
50. Sebastian, A., Elharrouss, O., Al-Maadeed, S., and Almaadeed, N. (2023). A Survey on Deep-Learning-Based Diabetic Retinopathy Classification. Diagnostics, 13.10.3390/diagnostics13030345 ↗
51. Rajamani, S., and Sasikala, S. (2023). Artificial Intelligence Approach for Diabetic Retinopathy Severity Detection. Informatica, 46.10.31449/inf.v46i8.4425 ↗
52. Jiwani, N., Gupta, K., Sharif, M.H.U., Datta, R., Habib, F., and Afreen, N. (2023, January 3–5). Application of Transfer Learning Approach for Diabetic Retinopathy Classification. Proceedings of the 2023 International Conference on Power Electronics and Energy (ICPEE), Bhubaneswar, India.10.1109/ICPEE54198.2023.10060777 ↗
53. Classification of Diabetic Retinopathy Severity in Fundus Images Using the Vision Transformer and Residual Attention10.1155/2023/1305583 ↗
54. Mondal, S.S., Mandal, N., Singh, K.K., Singh, A., and Izonin, I. (2023). EDLDR: An Ensemble Deep Learning Technique for Detection and Classification of Diabetic Retinopathy. Diagnostics, 13.10.3390/diagnostics13010124 ↗
55. Diabetic retinopathy detection and classification using capsule networks10.1007/s40747-021-00318-9 ↗
56. A Hybrid Convolutional Neural Network Model for Automatic Diabetic Retinopathy Classification From Fundus Images10.1109/JTEHM.2023.3282104 ↗
57. Impact of CLAHE-based image enhancement for diabetic retinopathy classification through deep learning10.1016/j.procs.2022.12.111 ↗
58. Alwakid, G., Gouda, W., and Humayun, M. (2023). Deep Learning-Based Prediction of Diabetic Retinopathy Using CLAHE and ESRGAN for Enhancement. Healthcare, 11.10.20944/preprints202302.0097.v1 ↗
59. Automated Binary Classification of Diabetic Retinopathy by SWIN Transformer10.29304/jqcm.2023.15.1.1166 ↗
60. Diagnostic assessment of deep learning algorithms for diabetic retinopathy screening10.1016/j.ins.2019.06.011 ↗
61. Setiawan, A.W., Mengko, T.R., Santoso, O.S., and Suksmono, A.B. (2013). International Conference on ICT for Smart Society, IEEE.
62. The discrete wavelet transform: Wedding the a trous and Mallat algorithms10.1109/78.157290 ↗
63. The Applications of Discrete Wavelet Transform in Image Processing: A Review
64. The effect of choosing optimizer algorithms to improve computer vision tasks: A comparative study10.1007/s11042-022-13820-0 ↗
65. Mehmood, F., Ahmad, S., and Whangbo, T.K. (2023). An Efficient Optimization Technique for Training Deep Neural Networks. Mathematics, 11.10.3390/math11061360 ↗
66. Understanding receiver operating characteristic (ROC) curves10.1017/S1481803500013336 ↗
67. Abadi, M., Agarwal, A., Barham, P., Brevdo, E., Chen, Z., Citro, C., Corrado, G.S., Davis, A., Dean, J., and Devin, M. (2016). Tensorflow: Large-scale machine learning on heterogeneous distributed systems. arXiv.
68. PyWavelets: A Python package for wavelet analysis10.21105/joss.01237 ↗
69. Patel, S., Lohakare, M., Prajapati, S., Singh, S., and Patel, N. (2021, January 21–23). DiaRet: A browser-based application for the grading of diabetic retinopathy with integrated gradients. Proceedings of the 2021 IEEE International Conference on Robotics, Automation and Artificial Intelligence (RAAI), Hong Kong.10.1109/RAAI52226.2021.9507938 ↗
70. Rahhal, D., Alhamouri, R., Albataineh, I., and Duwairi, R. (2022, January 21–23). Detection and Classification of Diabetic Retinopathy Using Artificial Intelligence Algorithms. Proceedings of the 2022 13th International Conference on Information and Communication Systems (ICICS), Irbid, Jordan.10.1109/ICICS55353.2022.9811197 ↗