Conference CORE A31% acc.

What can we learn about a generated image corrupting its latent representation?

Agnieszka Tomczak, Aarushi Gupta, Slobodan Ilic, Nassir Navab, Shadi Albarqouni

International Conference on Medical Image Computing and Computer-Assisted Intervention, 2022

Teaser figure

Abstract

Generative adversarial networks (GANs) offer an effective solution to the image-to-image translation problem, thereby allowing for new possibilities in medical imaging. They can translate images from one imaging modality to another at a low cost. For unpaired datasets, they rely mostly on cycle loss. Despite its effectiveness in learning the underlying data distribution, it can lead to a discrepancy between input and output data. The purpose of this work is to investigate the hypothesis that we can predict image quality based on its latent representation in the GANs bottleneck. We achieve this by corrupting the latent representation with noise and generating multiple outputs. The degree of differences between them is interpreted as the strength of the representation: the more robust the latent representation, the fewer changes in the output image the corruption causes. Our results demonstrate that our proposed method has the ability to i) predict uncertain parts of synthesized images, and ii) identify samples that may not be reliable for downstream tasks, e.g., liver segmentation task.

Interactive explainer

Coming soon. An interactive Slidev walkthrough of this paper — with live, explorable figures — will appear here. Authors: set slidev: in this page's front matter to the deployed deck URL.

Research themes

Keywords

Deep LearningMedical ImagingGenerative ModelsUncertainty

Lab authors

Cite (BibTeX)
@inproceedings{tomczak2022can,
 author = {Tomczak, Agnieszka and Gupta, Aarushi and Ilic, Slobodan and Navab, Nassir and Albarqouni, Shadi},
 booktitle = {International Conference on Medical Image Computing and Computer-Assisted Intervention},
 organization = {Springer, Cham},
 pages = {505--515},
 title = {What can we learn about a generated image corrupting its latent representation?},
 year = {2022}
}

Related publications

Flamby: Datasets and benchmarks for cross-silo federated learning in realistic healthcare settings

Jean Ogier Du Terrail, Samy-Safwan Ayed, Edwige Cyffers, Felix Grimberg, Chaoyang He, Regis Loeb, Paul Mangold, Tanguy Marchand, Othmane Marfoq, Erum Mushtaq, Boris Muzellec, Constantin Philippenko, Santiago Silva, Maria Telenczuk, Shadi Albarqouni, Salman Avestimehr, Aurélien Bellet, Aymeric Dieuleveut, Martin Jaggi, Sai Praneeth Karimireddy, Marco Lorenzi, Giovanni Neglia, Marc Tommasi, Mathieu Andreux

Advances in Neural Information Processing Systems, 2022