Repurposing the image generative potential: Exploiting GANs to grade diabetic retinopathy

Isabella Poles, Eleonora D’Arnese, Luca G. Cellamare, Marco D. Santambrogio, Darvin Yi

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

Diabetic Retinopathy (DR) is a common cause of irreversible vision loss in the working-age population. Automatic DR grading allows ophthalmologists to provide timely treatment to numerous patients. However, developing a robust grading model needs large, balanced, and annotated data, which poses challenges in the collection. Moreover, data augmentation often fails to generate diverse data, necessitating alternative approaches such as Generative Adversarial Networks (GANs). However, GANs often operate with low-resolution images as a result of their costly training process. Therefore, we present a novel method that repurposes the discriminator of an unconditional Progressive GAN, leveraging the generative knowledge gained for DR grading. Furthermore, a new Log-Likelihood Inception Distance (LLID) metric estimates the similarity between one synthesized and a set of real images, thereby capturing human judgment more effectively. Our method is validated through extensive experiments on three public datasets, outperforming the baseline classifiers’ performance by 12.5% and 14.33% average accuracy on small data regimes and when combined with state-of-the-art methods on large datasets, respectively. Additionally, LLID reproduces the comprehension ability of most of our Visual Turing Test participants, enabling differentiation between a synthesized image and a set of reference images with 82.88% accuracy. This confirms the quality of generated images and the metric consistency with human decision-making mechanisms.
Original languageEnglish
Title of host publication2024 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW)
PublisherInstitute of Electrical and Electronics Engineers
Pages2305-2314
Number of pages10
ISBN (Electronic)9798350365474
ISBN (Print)9798350365481
DOIs
Publication statusPublished - 27 Sept 2024
Event2024 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW) - Seattle Convention Center, Seattle, United States
Duration: 17 Jun 202421 Jun 2024

Publication series

NameProceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops
PublisherInstitute of Electrical and Electronics Engineers
ISSN (Print)2160-7508
ISSN (Electronic)2160-7516

Conference

Conference2024 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW)
Country/TerritoryUnited States
CitySeattle
Period17/06/2421/06/24

Keywords / Materials (for Non-textual outputs)

  • measurement
  • training
  • diabetic retinopathy
  • visualization
  • accuracy
  • decision making
  • generative adversarial networks

Fingerprint

Dive into the research topics of 'Repurposing the image generative potential: Exploiting GANs to grade diabetic retinopathy'. Together they form a unique fingerprint.

Cite this