A novel deep transfer learning based computerized diagnostic Systems for Multi-class imbalanced diabetic retinopathy severity classification

被引:19
|
作者
Vij, Richa [1 ]
Arora, Sakshi [1 ]
机构
[1] Shri Mata Vaishno Devi Univ, Sch Comp Sci & Engn, Katra 182320, Jammu And Kashm, India
关键词
Diabetic retinopathy; Multiclass DR severity classification; Deep inductive transfer learning model; Global average pooling layer; IDRiD dataset; Severity stage classification; CONVOLUTIONAL NEURAL-NETWORKS; FEATURES; ARCHITECTURE; IMAGE; RECOGNITION;
D O I
10.1007/s11042-023-14963-4
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Diabetic Retinopathy (DR) is a retinal condition that leads to gradual degeneration of the retina and eventual blindness, so early detection and evaluation of disease development are essential for effective treatment. Retinography is the most prevalent method for diagnosing DR; nevertheless, manual diagnosis is time-consuming and unpleasant. Deep Learning (DL) based algorithms have shown potential as a diagnostic tool for DR, achieving performance comparable to human image evaluation. The goal of this study is to develop Deep Transfer Learning-based Computerized Diagnostic Systems (DTL-CDS) for Multiclass DR Severity Classification (MCDR) by modifying and comparing deep inductive transfer learning (DITL) models (Inception V3, ResNet34, EfficientNet B0, VGG16, Xception). The four main objectives are i) pre-processing and balancing the imbalanced data labels, ii) proposing modified DITL model to extract features using global average pooling layer to avoid overfitting and reduce losses using leaky ReLU. The final classification uses a softmax layer to automatically classify Diabetic Retinopathy severity stages using IDRiD dataset, (iii) comparing different base and modified DITL models to existing work, and (iv) evaluating the robustness of proposed model using various performance metrics. Comprehensive analysis of MCDR between the proposed and the existing work shows that the proposed approach outperforms the state-of-the-art, with 99.36% accuracy, 0.986 precision, 0.986 recall, 0.986 F1-score, 0.997 AUC-ROC, 0.9902 AUC. The experimental results show the enhancement in diagnosis performance and modified DITL results in robust and reliable computer-aided diagnosis systems to aid specialists in proper detection of DR severity stages by reducing human errors and reducing costs.
引用
收藏
页码:34847 / 34884
页数:38
相关论文
共 50 条
  • [41] OTLAMC: An Online Transfer Learning Algorithm for Multi-class Classification
    Kang, Zhongfeng
    Yang, Bo
    Li, Zesong
    Wang, Peng
    KNOWLEDGE-BASED SYSTEMS, 2019, 176 : 133 - 146
  • [42] Deep Learning-Based Classification of Diabetic Retinopathy
    Huang, Zhenjia
    PROCEEDINGS OF 2023 4TH INTERNATIONAL SYMPOSIUM ON ARTIFICIAL INTELLIGENCE FOR MEDICINE SCIENCE, ISAIMS 2023, 2023, : 371 - 375
  • [43] Semi-supervised Learning Algorithm Based on Linear Lie Group for Imbalanced Multi-class Classification
    Chengjun Xu
    Guobin Zhu
    Neural Processing Letters, 2020, 52 : 869 - 889
  • [44] Semi-supervised Learning Algorithm Based on Linear Lie Group for Imbalanced Multi-class Classification
    Xu, Chengjun
    Zhu, Guobin
    NEURAL PROCESSING LETTERS, 2020, 52 (01) : 869 - 889
  • [45] Deep learning based binary classification of diabetic retinopathy images using transfer learning approach
    Saproo, Dimple
    Mahajan, Aparna N.
    Narwal, Seema
    JOURNAL OF DIABETES AND METABOLIC DISORDERS, 2024, 23 (02) : 2289 - 2314
  • [47] Transfer Learning based Classification of Diabetic Retinopathy Stages
    Pradhan, Adarsh
    Dey, Bhiman Kr
    Nath, Rahul Kumar
    Sarma, Bhaskarjyoti
    2020 INTERNATIONAL CONFERENCE ON COMPUTATIONAL PERFORMANCE EVALUATION (COMPE-2020), 2020, : 813 - 819
  • [48] Multi-Class Brain Lesion Classification Using Deep Transfer Learning With MobileNetV3
    Majeed, Ahmed Firas
    Salehpour, Pedram
    Farzinvash, Leili
    Pashazadeh, Saeid
    IEEE ACCESS, 2024, 12 : 155295 - 155308
  • [49] Application of data-efficient generative techniques for Multi-Class Diabetic Retinopathy Classification
    Du, Melissa
    Elangovan, Kabilan
    Lim, Gilbert
    Ting, Daniel S. W.
    INVESTIGATIVE OPHTHALMOLOGY & VISUAL SCIENCE, 2023, 64 (08)
  • [50] Multi-Class Document Classification Using Lexical Ontology-Based Deep Learning †
    Yelmen, Ilkay
    Gunes, Ali
    Zontul, Metin
    APPLIED SCIENCES-BASEL, 2023, 13 (10):