Red Lesion Segmentation of Fundus Image with Multi-task Learning
Author:
Affiliation:

Clc Number:

TP391

Fund Project:

National Natural Science Foundation of China (61872200); National Key Research and Development Program of China (2016YFC0400709, 2018YFB2100300); Natural Science Foundation of Tianjin Municipality (18YFYZCG00060, 19JCZDJC316 00); Key Cultivation Project of Education Achievement Award of Tianjin Municipality (PYGJ-018)

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    Diabetic retinopathy (DR) is the leading cause of vision loss for adult individuals, and early fundus screening can significantly reduce this visual loss. Color fundus image is often used in large-scale fundus screening due to the acquisition convenience and its human-harmless. As a kind of red lesions in fundus images, the appearance of microaneurysms is the main marker of mild non-proliferative DR, and hemorrhage, as another kind of red lesions, is related to moderate and severe non-proliferative DR. So that red lesions in fundus images are important indicators for the screening of DR. This study proposes a multi-task network, named Red-Seg, for red lesion segmentation. The network contains two individual branches, each is used for one kind of lesion segmentation. Meantime, a two-stage training algorithm is presented where different loss functions are used in different stages. In the first stage, modified Top-k balanced cross-entropy loss is used to push the network focuses on hard-to-classify samples. And, in the second stage, false positive and false negative are integrated as loss function into training to reduce misclassification further. At last, extensive experiments are employed on the IDRiD dataset, and the lesion segmentation results are compared with other methods. Experimental results show that proposed two-stage training algorithm can lead to much higher precision and recall, which means this method can reduce misclassification in some certain. Specifically for hemorrhage segmentation, both recall and precision increased by at least 2.8%. Meanwhile, compared with other image-level lesion segmentation models, such as HED, FCRN, DeepLabv3+, and L-Seg, Red-Seg achieves much higher AUC_PR on microaneurysm segmentation.

    Reference
    Related
    Cited by
Get Citation

郭松,李涛,李宁,康宏,张玉军,王恺.基于多任务学习的眼底图像红色病变点分割.软件学报,2021,32(11):3646-3658

Copy
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:July 18,2019
  • Revised:November 04,2019
  • Adopted:
  • Online: November 05,2021
  • Published: November 06,2021
You are the firstVisitors
Copyright: Institute of Software, Chinese Academy of Sciences Beijing ICP No. 05046678-4
Address:4# South Fourth Street, Zhong Guan Cun, Beijing 100190,Postal Code:100190
Phone:010-62562563 Fax:010-62562533 Email:jos@iscas.ac.cn
Technical Support:Beijing Qinyun Technology Development Co., Ltd.

Beijing Public Network Security No. 11040202500063