Explainable Diabetic Retinopathy Detection Using a Distributed CNN and LightGBM Framework
Source abstract
Abstract
Diabetic Retinopathy (DR) is a critical disorder that affects the retina due to the constant rise in diabetics and remains the major cause of blindness across the world. Early detection and timely treatment are essential to mitigate the effects of DR, such as retinal damage and vision impairment. Several conventional approaches have been proposed to detect DR early and accurately, but they are limited by data imbalance, interpretability, overfitting, convergence time, and other issues. To address these drawbacks and improve DR detection accurately, a distributed Explainable Convolutional Neural network-enabled Light Gradient Boosting Machine (DE-ExLNN) is proposed in this research. The model combines an explainable Convolutional Neural Network (CNN) and Light Gradient Boosting Machine (LightGBM), achieving highly accurate outcomes in DR detection. LightGBM serves as the detection model, and the inclusion of an explainable CNN addresses issues that conventional CNN classifiers could not resolve. A custom dataset was created for this research, containing both fundus and OCTA images collected from a real-time environment, providing more accurate results compared to standard conventional DR datasets. The custom dataset demonstrates notable accuracy, sensitivity, specificity, and Matthews Correlation Coefficient (MCC) scores, underscoring the effectiveness of this approach. Evaluations against other standard datasets achieved an accuracy of 93.94%, sensitivity of 93.90%, specificity of 93.99%, and MCC of 93.88% for fundus images. For OCTA images, the results obtained an accuracy of 95.30%, sensitivity of 95.50%, specificity of 95.09%, and MCC of 95%. Results prove that the combination of explainable CNN and LightGBM outperforms other methods. The inclusion of distributed learning enhances the model’s efficiency by reducing time consumption and complexity while facilitating feature extraction.
Source: openalex_inverted_index · Quality: baseline
Research context
OpenAlex topics: Retinal Imaging and Analysis · Brain Tumor Detection and Classification · Artificial Intelligence in Healthcare
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
References (43)
Show source-provided references
1. Automated screening system for retinal health using bi-dimensional empirical mode decomposition and integrated index10.1016/j.compbiomed.2016.04.015 ↗
2. Multi-stream deep neural network for diabetic retinopathy severity classification under a boosting framework10.1109/ACCESS.2022.3217216 ↗
3. Vision transformer model for predicting the severity of diabetic retinopathy in fundus photography-based retina images10.1109/ACCESS.2023.3326528 ↗
4. A faster RCNN-based diabetic retinopathy detection method using fused features from retina images10.1109/ACCESS.2023.3330104 ↗
5. A regression-based approach to diabetic retinopathy diagnosis using efficientnet10.3390/diagnostics13040774 ↗
6. Diabetic retinopathy: looking forward to 203010.3389/fendo.2022.1077669 ↗
7. Artificial intelligence for diabetic retinopathy screening using color retinal photographs: from development to deployment10.1007/s40123-023-00691-3 ↗
8. Internet of Things and deep learning enabled diabetic retinopathy diagnosis using retinal fundus images10.1109/ACCESS.2023.3257988 ↗
9. Goodfellow I. Nips 2016 tutorial: generative adversarial networks. arXiv:1701.00160. 2016.
10. Deep learning based computer-aided automatic prediction and grading system for diabetic retinopathy10.1007/s11042-023-14970-5 ↗
11. Deep learning in computer vision: a critical review of emerging techniques and application scenarios
12. Diabetic retinopathy detection using red lesion localization and convolutional neural networks10.1016/j.compbiomed.2019.103537 ↗
13. Diabetic macular edema grading based on improved Faster R-CNN and MD-ResNet10.1007/s11760-020-01792-3 ↗
14. Segmentation of Hard exudates for the detection of Diabetic Retinopathy with RNN based sematic features using fundus images
15. Deep learning for diabetic retinopathy analysis: a review, research challenges, and future directions10.3390/s22186780 ↗
16. Diabetic retinopathy screening using deep learning for multi-class imbalanced datasets10.1016/j.compbiomed.2022.105989 ↗
17. Ensembled deep convolutional generative adversarial network for grading imbalanced diabetic retinopathy recognition10.1109/ACCESS.2023.3327900 ↗
18. Segmentation using the IC2T model and classification of diabetic retinopathy using the rock hyrax swarm-based coordination attention mechanism10.1109/ACCESS.2023.3330436 ↗
19. Automatic severity classification of diabetic retinopathy based on denseNet and convolutional block attention module10.1109/ACCESS.2022.3165193 ↗
20. A hybrid convolutional neural network model for automatic diabetic retinopathy classification from fundus images10.1109/JTEHM.2023.3282104 ↗
21. Optimizing diabetic retinopathy detection with inception-V4 and dynamic version of snow leopard optimization algorithm10.1016/j.bspc.2024.106501 ↗
22. Melin · Fuzzy logic and neural networks for hybrid intelligent system design · 2023 · 23710.1007/978-3-031-22042-5_14 ↗
23. IoT-Opthom-CAD: IoT-enabled classification system of multiclass retinal eye diseases using dynamic swin transformers and explainable artificial intelligence
24. Image denoising based on gaussian/bilateral filter and its method noise thresholding10.1007/s11760-012-0372-7 ↗
25. A survey on image data augmentation for deep learning10.1186/s40537-019-0197-0 ↗
26. Investigation on the effect of a Gaussian Blur in image filtering and segmentation
27. OpenCV Flip Image (cv2.flip)
28. Automatic diagnosis of fungal keratitis using data augmentation and image fusion with deep convolutional neural network10.1016/j.cmpb.2019.105019 ↗
29. Enhancing performance of deep learning models with different data augmentation techniques: a survey
30. Detection of diabetic retinopathy using convolutional neural networks for feature extraction and classification (DRFEC)10.1007/s11042-022-14165-4 ↗
31. Mean and standard deviation features of color histogram using laplacian filter for content-based image retrieval
32. Hybrid fusion of high-resolution and ultra-widefield OCTA acquisitions for the automatic diagnosis of diabetic retinopathy10.3390/diagnostics13172770 ↗
33. An advanced deep neural network for fundus image analysis and enhancing diabetic retinopathy detection10.1016/j.health.2024.100303 ↗
34. Grad-CAM: visual explanations from deep networks via gradient-based localization
35. Grad-cam++: Generalized gradient-based visual explanations for deep convolutional networks
36. Full-gradient representation for neural network visualization
37. Diagnosis of diabetes mellitus using gradient boosting machine (LightGBM)10.3390/diagnostics11091714 ↗
38. DiaretDb1 dataset. [cited 2025 Apr 10]. Available from: https://paperswithcode.com/dataset/diaretdb1.
39. Messidor dataset. [cited 2025 Apr 10]. Available from: https://paperswithcode.com/dataset/messidor-1.
40. STARE dataset. [cited 2025 Apr 10]. Available from: https://paperswithcode.com/dataset/stare.
41. OCTA 500 dataset. [cited 2025 Apr 10]. Available from: https://paperswithcode.com/dataset/octagon.
42. Optical coherence tomography angiography-OCTA dataset for detection of diabetic retinopathy [Dataset]
43. Fundus dataset for detection of diabetic retinopathy [Dataset]