Diabetic retinopathy disease detection using shapley additive ensembled densenet-121 resnet-50 model

被引:0
|
作者
Mary, A. Rosline [1 ]
Kavitha, P. [2 ]
机构
[1] Visvesvaraya Technol Univ, Vemana Inst Technol, Dept Comp Sci & Engn, Belagavi 590018, Karnataka, India
[2] Visvesvaraya Technol Univ, CMR Inst Technol, Dept CSE, Belagavi 590018, Karnataka, India
关键词
Diabetic Retinopathy; DenseNet-121; ResNet-50; Shapley additive Explainable AI; Ensemble; U-Net; ARTIFICIAL-INTELLIGENCE;
D O I
10.1007/s11042-024-18309-6
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Diabetic retinopathy (DR) is a common eye disease that results in vision loss by damaging the blood vessels. Diabetic patients are at high risk of developing DR owing to the damage of retinal lesions, thereby causing clots, injuries and bleeding. The disease leads to abnormal changes in the structure of the retina. Therefore, the timely detection and early treatment of eye diseases are essential for preventing people from vision loss. Ophthalmologists distinguish DR based on features such as exudes, microaneurysms, blood vessel area, hemorrhages, etc. An artificial intelligence (AI) method is proposed by ensembling a deep learning (DL) model with the explainable AI based Shapley additive (SHAP) method for DR image segmentation and classification. The proposed model uses fundus images to detect abnormalities in the eye. First, the DR images are collected from the Asia Pacific Tele-Ophthalmology Society 2019 blindness detection (APTOS 2019) dataset. Data augmentation is performed to artificially increase the size of a training dataset by generating new data samples from existing ones by rescaling, flipping, rotating, zooming, etc. In order to improve the quality and enhance specific features, pre-processing is performed. The pre-processed images are segmented using the improved U-Net model, where the severity of the disease gets predicted, and the fundus images are segmented using the trained model to attain precise abstraction of retinal blood vessels. Finally, the proposed study used the Shapley Additive Ensembled DenseNet-121 ResNet-50 (SAE-DR) model to detect DR disease based on the features. To improve the readability of the deep learning model, an explainable AI based Shapley additive method is proposed. Then, compare the results of the proposed model with existing state-of-the-art methods. The simulation results prove that the proposed model achieves superior detection performance with an accuracy of 98.69%, sensistivity of 86.23%, specificity of 97.54%, F-score of 90.26%, precision of 94.26% and processing time of 0.153 s.
引用
收藏
页码:69797 / 69824
页数:28
相关论文
共 5 条
  • [1] Development of revised ResNet-50 for diabetic retinopathy detection
    Chun-Ling Lin
    Kun-Chi Wu
    [J]. BMC Bioinformatics, 24
  • [2] Development of revised ResNet-50 for diabetic retinopathy detection
    Lin, Chun-Ling
    Wu, Kun-Chi
    [J]. BMC BIOINFORMATICS, 2023, 24 (01)
  • [3] An enhanced ResNet-50 deep learning model for arrhythmia detection using electrocardiogram biomedical indicators
    Anand, R.
    Lakshmi, S. Vijaya
    Pandey, Digvijay
    Pandey, Binay Kumar
    [J]. EVOLVING SYSTEMS, 2024, 15 (01) : 83 - 97
  • [4] An enhanced ResNet-50 deep learning model for arrhythmia detection using electrocardiogram biomedical indicators
    R. Anand
    S. Vijaya Lakshmi
    Digvijay Pandey
    Binay Kumar Pandey
    [J]. Evolving Systems, 2024, 15 : 83 - 97
  • [5] Diabetic Retinopathy: Detection and Classification Using AlexNet, GoogleNet and ResNet50 Convolutional Neural Networks
    Caicho, Jhonny
    Chuya-Sumba, Cristina
    Jara, Nicole
    Salum, Graciela M.
    Tirado-Espin, Andres
    Villalba-Meneses, Gandhi
    Alvarado-Cando, Omar
    Cadena-Morejon, Carolina
    Almeida-Galarraga, Diego A.
    [J]. SMART TECHNOLOGIES, SYSTEMS AND APPLICATIONS, SMARTTECH-IC 2021, 2022, 1532 : 259 - 271