Multi-Cell Multi-Task Convolutional Neural Networks for Diabetic Retinopathy Grading

Kang Zhou, Zaiwang Gu, Wen Liu, Weixin Luo, Jun Cheng, Shenghua Gao, Jiang Liu

Research output: Chapter in Book/Conference proceedingConference contributionpeer-review

75 Citations (Scopus)

Abstract

Diabetic Retinopathy (DR) is a non-negligible eye disease among patients with Diabetes Mellitus, and automatic retinal image analysis algorithm for the DR screening is in high demand. Considering the resolution of retinal image is very high, where small pathological tissues can be detected only with large resolution image and large local receptive field are required to identify those late stage disease, but directly training a neural network with very deep architecture and high resolution image is both time computational expensive and difficult because of gradient vanishing/exploding problem, we propose a Multi-Cell architecture which gradually increases the depth of deep neural network and the resolution of input image, which both boosts the training time but also improves the classification accuracy. Further, considering the different stages of DR actually progress gradually, which means the labels of different stages are related. To considering the relationships of images with different stages, we propose a Multi-Task learning strategy which predicts the label with both classification and regression. Experimental results on the Kaggle dataset show that our method achieves a Kappa of 0.841 on test set which is the 4th rank of all state-of-the-arts methods. Further, our Multi-Cell Multi-Task Convolutional Neural Networks (M2CNN) solution is a general framework, which can be readily integrated with many other deep neural network architectures.

Original languageEnglish
Title of host publication40th Annual International Conference of the IEEE Engineering in Medicine and Biology Society, EMBC 2018
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages2724-2727
Number of pages4
ISBN (Electronic)9781538636466
DOIs
Publication statusPublished - 26 Oct 2018
Externally publishedYes
Event40th Annual International Conference of the IEEE Engineering in Medicine and Biology Society, EMBC 2018 - Honolulu, United States
Duration: 18 Jul 201821 Jul 2018

Publication series

NameProceedings of the Annual International Conference of the IEEE Engineering in Medicine and Biology Society, EMBS
Volume2018-July
ISSN (Print)1557-170X

Conference

Conference40th Annual International Conference of the IEEE Engineering in Medicine and Biology Society, EMBC 2018
Country/TerritoryUnited States
CityHonolulu
Period18/07/1821/07/18

Keywords

  • Deep Learning
  • Diabetic Retinopathy
  • Medical Image
  • Multi-Cell Architecture
  • MultiTask Learning

ASJC Scopus subject areas

  • Signal Processing
  • Biomedical Engineering
  • Computer Vision and Pattern Recognition
  • Health Informatics

Fingerprint

Dive into the research topics of 'Multi-Cell Multi-Task Convolutional Neural Networks for Diabetic Retinopathy Grading'. Together they form a unique fingerprint.

Cite this