Explainable Diabetic Retinopathy Detection Using a Distributed CNN and LightGBM Framework

被引:0
作者
Bidwai, Pooja [1 ,2 ]
Gite, Shilpa [1 ,3 ]
Pradhan, Biswajeet [4 ]
Almari, Abdullah [5 ]
机构
[1] Symbiosis Int Deemed Univ, Symbiosis Inst Technol, Artificial Intelligence & Machine Learning Dept, Pune 412115, India
[2] SPPU, Pimpri Chinchwad Coll Engn, Dept Comp Engn, Pune 411044, Maharashtra, India
[3] Symbiosis Int Deemed Univ, Symbiosis Inst Technol, Artificial Intelligence & Machine Learning Dept, Pune 412115, Maharashtra, India
[4] Univ Technol Sydney, Ctr Adv Modelling & Geospatial Informat Syst CAMG, Sch Civil & Environm Engn, Sydney, NSW 2007, Australia
[5] King Saud Univ, Coll Sci, Dept Geol & Geophys, Riyadh 11451, Saudi Arabia
来源
CMC-COMPUTERS MATERIALS & CONTINUA | 2025年 / 84卷 / 02期
关键词
Diabetic retinopathy; explainable convolutional neural network; light gradient boosting machine; fundus image; custom dataset; CLASSIFICATION; SYSTEM;
D O I
10.32604/cmc.2025.061018
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
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 realtime 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.
引用
收藏
页码:2645 / 2676
页数:32
相关论文
共 43 条
[1]   Automated screening system for retinal health using bi-dimensional empirical mode decomposition and integrated index [J].
Acharya, U. Rajendra ;
Mookiah, Muthu Rama Krishnan ;
Koh, Joel E. W. ;
Tan, Jen Hong ;
Bhandary, Sulatha V. ;
Rao, A. Krishna ;
Fujita, Hamido ;
Hagiwara, Yuki ;
Chua, Chua Kuang ;
Laude, Augustinus .
COMPUTERS IN BIOLOGY AND MEDICINE, 2016, 75 :54-62
[2]  
AlBalawi T, 2024, INT J ADV COMPUT SC, V15, P163
[3]   A Hybrid Convolutional Neural Network Model for Automatic Diabetic Retinopathy Classification From Fundus Images [J].
Ali, Ghulam ;
Dastgir, Aqsa ;
Iqbal, Muhammad Waseem ;
Anwar, Muhammad ;
Faheem, Muhammad .
IEEE JOURNAL OF TRANSLATIONAL ENGINEERING IN HEALTH AND MEDICINE, 2023, 11 :341-350
[4]  
Bidwai Pooja, 2024, Zenodo, DOI 10.5281/ZENODO.13734036
[5]  
Bidwai Pooja, 2023, Zenodo, DOI 10.5281/ZENODO.10400092
[6]   Deep learning in computer vision: A critical review of emerging techniques and application scenarios [J].
Chai, Junyi ;
Zeng, Hao ;
Li, Anming ;
Ngai, Eric W. T. .
MACHINE LEARNING WITH APPLICATIONS, 2021, 6
[7]   Grad-CAM plus plus : Generalized Gradient-based Visual Explanations for Deep Convolutional Networks [J].
Chattopadhay, Aditya ;
Sarkar, Anirban ;
Howlader, Prantik ;
Balasubramanian, Vineeth N. .
2018 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV 2018), 2018, :839-847
[8]   Detection of Diabetic Retinopathy using Convolutional Neural Networks for Feature Extraction and Classification (DRFEC) [J].
Das, Dolly ;
Biswas, Saroj Kumar ;
Bandyopadhyay, Sivaji .
MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 82 (19) :29943-30001
[9]   Automatic Severity Classification of Diabetic Retinopathy Based on DenseNet and Convolutional Block Attention Module [J].
Farag, Mohamed M. ;
Fouad, Mariam ;
Abdel-Hamid, Amr T. .
IEEE ACCESS, 2022, 10 :38299-38308
[10]  
Gedraite ES, 2011, ELMAR PROC, P393