Multi-Task Learning for Diabetic Retinopathy Grading and Lesion Segmentation

Authors

  • Alex Foo National University of Singapore
  • Wynne Hsu National University of Singapore
  • Mong Li Lee National University of Singapore
  • Gilbert Lim National University of Singapore
  • Tien Yin Wong Duke-NUS Medical School

DOI:

https://doi.org/10.1609/aaai.v34i08.7035

Abstract

Although deep learning for Diabetic Retinopathy (DR) screening has shown great success in achieving clinically acceptable accuracy for referable versus non-referable DR, there remains a need to provide more fine-grained grading of the DR severity level as well as automated segmentation of lesions (if any) in the retina images. We observe that the DR severity level of an image is dependent on the presence of different types of lesions and their prevalence. In this work, we adopt a multi-task learning approach to perform the DR grading and lesion segmentation tasks. In light of the lack of lesion segmentation mask ground-truths, we further propose a semi-supervised learning process to obtain the segmentation masks for the various datasets. Experiments results on publicly available datasets and a real world dataset obtained from population screening demonstrate the effectiveness of the multi-task solution over state-of-the-art networks.

Downloads

Published

2020-04-03

How to Cite

Foo, A., Hsu, W., Lee, M. L., Lim, G., & Wong, T. Y. (2020). Multi-Task Learning for Diabetic Retinopathy Grading and Lesion Segmentation. Proceedings of the AAAI Conference on Artificial Intelligence, 34(08), 13267-13272. https://doi.org/10.1609/aaai.v34i08.7035

Issue

Section

IAAI Technical Track: Emerging Papers