URNet: System for recommending referrals for community screening of diabetic retinopathy based on deep learning.

Journal: Experimental biology and medicine (Maywood, N.J.)
Published Date:

Abstract

Diabetic retinopathy (DR) will cause blindness if the detection and treatment are not carried out in the early stages. To create an effective treatment strategy, the severity of the disease must first be divided into referral-warranted diabetic retinopathy (RWDR) and non-referral diabetic retinopathy (NRDR). However, there are usually no sufficient fundus examinations due to lack of professional service in the communities, particularly in the developing countries. In this study, we introduce UGAN_Resnet_CBAM (URNet; UGAN is a generative adversarial network that uses Unet for feature extraction), a two-stage end-to-end deep learning technique for the automatic detection of diabetic retinopathy. The characteristics of DDR fundus data set were used to design an adaptive image preprocessing module in the first stage. Gradient-weighted Class Activation Mapping (Grad-CAM) and t-distribution and stochastic neighbor embedding (t-SNE) were used as the evaluation indices to analyze the preprocessing results. In the second stage, we enhanced the performance of the Resnet50 network by integrating the convolutional block attention module (CBAM). The outcomes demonstrate that our proposed solution outperformed other current structures, achieving 94.5% and 94.4% precisions, and 96.2% and 91.9% recall for NRDR and RWDR, respectively.

Authors

  • Kun Yang
    Department of Bone and Joint Surgery, Affiliated Hospital of Southwest Medical University, Luzhou Sichuan, 646000, P.R.China.
  • Yufei Lu
    College of Quality and Technical Supervision, Hebei University, Baoding 071002, China.
  • Linyan Xue
    School of Quality and Technical Supervision, Hebei University, Baoding, Hebei 071002, P.R.China.
  • Yueting Yang
    College of Quality and Technical Supervision, Hebei University, Baoding 071002, China.
  • Shilong Chang
    College of Quality and Technical Supervision, Hebei University, Baoding 071002, Hebei, China.
  • Chuanqing Zhou
    Institute of Biomedical Engineering, Shenzhen Bay Laboratory, Shenzhen, China.