Diabetic retinopathy (DR) is a leading cause of blindness, requiring accurate grading and lesion segmentation for effective diagnosis and treatment. Existing models often face challenges with large parameters and computational complexity, typically focusing on either grading or single-lesion segmentation. This paper investigates solving the above issues from the perspective of transfer learning, using a simple dual-branch model to accomplish lesion grading and multi-lesion segmentation tasks. Firstly, we employ fine-tuning to transfer the ResMLP-12 model to grade diabetic retinopathy lesions, achieving favorable results on our dataset while reducing training and computational costs. Secondly, we design an end-to-end U-shaped architecture for lesion segmentation tasks, where the pre-trained ResMLP-12 model serves as the encoder of our network, replacing the convolutional layers in the original U-shaped network, further reducing computational costs. To achieve more precise segmentation of lesions of different scales, we design a multi-scale pooling module (MSP) between the encoder and decoder. We train and validate our model on two datasets, APTOS 2019 and IDRiD. The experimental results prove the effectiveness of our model.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Integrated Grading and Segmentation for Diabetic Retinopathy: A Perspective from Transfer Learning

  • Yun Jiang,
  • Miaofeng Lu,
  • ZeKun Zhang,
  • Huanting Guo,
  • Wenping Liu,
  • Yan Yao

摘要

Diabetic retinopathy (DR) is a leading cause of blindness, requiring accurate grading and lesion segmentation for effective diagnosis and treatment. Existing models often face challenges with large parameters and computational complexity, typically focusing on either grading or single-lesion segmentation. This paper investigates solving the above issues from the perspective of transfer learning, using a simple dual-branch model to accomplish lesion grading and multi-lesion segmentation tasks. Firstly, we employ fine-tuning to transfer the ResMLP-12 model to grade diabetic retinopathy lesions, achieving favorable results on our dataset while reducing training and computational costs. Secondly, we design an end-to-end U-shaped architecture for lesion segmentation tasks, where the pre-trained ResMLP-12 model serves as the encoder of our network, replacing the convolutional layers in the original U-shaped network, further reducing computational costs. To achieve more precise segmentation of lesions of different scales, we design a multi-scale pooling module (MSP) between the encoder and decoder. We train and validate our model on two datasets, APTOS 2019 and IDRiD. The experimental results prove the effectiveness of our model.