Assistive Framework for Automatic Detection of All the Zones in Retinopathy of Prematurity Using Deep Learning

J Digit Imaging. 2021 Aug;34(4):932-947. doi: 10.1007/s10278-021-00477-8. Epub 2021 Jul 8.

Abstract

Retinopathy of prematurity (ROP) is a potentially blinding disorder seen in low birth weight preterm infants. In India, the burden of ROP is high, with nearly 200,000 premature infants at risk. Early detection through screening and treatment can prevent this blindness. The automatic screening systems developed so far can detect "severe ROP" or "plus disease," but this information does not help schedule follow-up. Identifying vascularized retinal zones and detecting the ROP stage is essential for follow-up or discharge from screening. There is no automatic system to assist these crucial decisions to the best of the authors' knowledge. The low contrast of images, incompletely developed vessels, macular structure, and lack of public data sets are a few challenges in creating such a system. In this paper, a novel method using an ensemble of "U-Network" and "Circle Hough Transform" is developed to detect zones I, II, and III from retinal images in which macula is not developed. The model developed is generic and trained on mixed images of different sizes. It detects zones in images of variable sizes captured by two different imaging systems with an accuracy of 98%. All images of the test set (including the low-quality images) are considered. The time taken for training was only 14 min, and a single image was tested in 30 ms. The present study can help medical experts interpret retinal vascular status correctly and reduce subjective variation in diagnosis.

Keywords: Artificial Intelligence; Automatic zone detection; Machine learning; Retinopathy of prematurity(ROP); Segmentation; U-Net.

MeSH terms

  • Deep Learning*
  • Humans
  • Infant, Low Birth Weight
  • Infant, Newborn
  • Infant, Premature
  • Retina / diagnostic imaging
  • Retinopathy of Prematurity* / diagnostic imaging