| Literature DB >> 36249693 |
Aaron S Coyner1, Jimmy S Chen1,2, Ken Chang3,4, Praveer Singh3,4, Susan Ostmo1, R V Paul Chan5, Michael F Chiang6, Jayashree Kalpathy-Cramer3,4, J Peter Campbell1.
Abstract
Purpose: Developing robust artificial intelligence (AI) models for medical image analysis requires large quantities of diverse, well-chosen data that can prove challenging to collect because of privacy concerns, disease rarity, or diagnostic label quality. Collecting image-based datasets for retinopathy of prematurity (ROP), a potentially blinding disease, suffers from these challenges. Progressively growing generative adversarial networks (PGANs) may help, because they can synthesize highly realistic images that may increase both the size and diversity of medical datasets. Design: Diagnostic validation study of convolutional neural networks (CNNs) for plus disease detection, a component of severe ROP, using synthetic data. Participants: Five thousand eight hundred forty-two retinal fundus images (RFIs) collected from 963 preterm infants.Entities:
Keywords: AI, artificial intelligence; Artificial intelligence; CNN, convolutional neural network; DL, deep learning; Deep learning; FID, Fréchet inception distance; GAN, generative adversarial network; Generative adversarial network; PGAN, progressively growing generative adversarial network; RFI, retinal fundus image; ROP, retinopathy of prematurity; RVM, retinal vessel map; Retinopathy of prematurity; UMAP, uniform manifold approximation and projection
Year: 2022 PMID: 36249693 PMCID: PMC9560638 DOI: 10.1016/j.xops.2022.100126
Source DB: PubMed Journal: Ophthalmol Sci ISSN: 2666-9145
Partitions of the Informatics in Retinopathy of Prematurity Dataset
| Dataset | No. of Patients | No. of Images | Normal (%) | Preplus (%) | Plus (%) |
|---|---|---|---|---|---|
| Training | 534 | 3477 | 2908 (83.6) | 464 (13.4) | 105 (3.0) |
| Validation | 178 | 1167 | 903 (77.4) | 222 (19.0) | 42 (3.6) |
| Test | 179 | 1098 | 927 (84.4) | 136 (12.4) | 35 (3.2) |
| Expert test | 72 | 100 | 54 (54.0) | 31 (31.0) | 15 (15.0) |
| Total | 963 | 5842 | 4792 (82.0) | 853 (14.6) | 197 (3.4) |
Figure 1The main arteries and veins present in color retinal fundus images (RFIs; left column) can be automatically segmented into grayscale retinal vessel maps (RVMs; right column) using a previously trained u-net.
Figure 2Synthetic retinal vessel maps (RVMs) of eyes with (A) normal retinal vasculature, (B) preplus disease, or (C) plus disease can be generated by progressively growing generative adversarial networks trained on a limited number of real RVMs.
Figure 3Two-dimensional manifold generated by uniform manifold approximation and projection (UMAP) using the extracted image features of real retinal vessel maps from a convolutional neural network trained to diagnose plus disease using said retinal vessel maps (opaque triangles). Features of synthetic retinal vessel maps (transparent circles) were extracted from the same model and their locations on the UMAP manifold were predicted. Real and synthetic normal (green), preplus (orange), and plus disease (red) retinal vessel maps, respectively, overlapped with one another on the UMAP manifold.
Figure 4Pairs of real retinal vessel maps (RVMs) are closer in feature space than pairs of real and synthetic RVMs. Using the same features output by InceptionV3 to compute Fréchet inception distance, the Euclidean distances between RVMs from the same participants, different participants, and synthetic RVMs and the real RVMs they were trained on were calculated. A, The closest Euclidean distance was 5.54 and occurred between RVMs from the same participant’s eye. B, The closest RVMs from different participants showed a distance of 5.71. C, The closest synthetic RVM to a real RVM from the training dataset showed a Euclidean distance of 6.15.
Figure 5Models trained on synthetic retinal vessel maps (RVMs) detect plus disease better than those trained on real RVMs. Receiver operating characteristic curves (ROCs) of convolutional neural networks (CNNs) trained on (A) real or (B) synthetic RVMs for detection of plus disease versus normal or preplus disease from real RVMs in the test dataset are depicted. Areas under the ROC curves (AUCs) were significantly different as determined by 2 different tests (P = 0.004, P = 0.006).
A Model Trained on Synthetic Retinal Vessel Maps Predicts Plus Disease Diagnoses More Similarly to International Experts Than a Model Trained on Real Retinal Vessel Maps
| Ground Truth | ||
|---|---|---|
| Not Plus | Plus | |
| Real RVMs | ||
| Not Plus | 83 | 5 |
| Plus | 2 | 10 |
| Synthetic RVMs | ||
| Not Plus | 84 | 1 |
| Plus | 1 | 14 |
RVM = retinal vessel map.