A latent space exploration for microscopic skin lesion augmentations with VQ-VAE-2 and PixelSNAIL

Alessio Gallucci; Nicola Pezzotti; Dmitry Znamenskiy; Milan Petkovic

doi:10.1117/12.2580664

15 February 2021 A latent space exploration for microscopic skin lesion augmentations with VQ-VAE-2 and PixelSNAIL

Alessio Gallucci, Nicola Pezzotti, Dmitry Znamenskiy, Milan Petkovic

Proceedings Volume 11596, Medical Imaging 2021: Image Processing; 115962X (2021) https://doi.org/10.1117/12.2580664
Event: SPIE Medical Imaging, 2021, Online Only

Conference Poster

Abstract

Skin cancer affects more than 3 million people only in the US. Comprehensive microscopic databases include around 30 thousand samples, limiting the richness of patterns that can be presented to machine learning. To this end, generative models such as GANs have been proposed for creating realistic synthetic images but, despite their popularity, they are often difficult to train and control. Recently an autoregressive approach based on a quantized autoencoder showed state of the art performances while being simple to train and provide synthetic data generation opportunities. In the first part of this paper we evaluate the training of VQ-VAE-2 with different latent space configuration. In the second part, we show how to use a learned prior over the latent space with PixelSNAIL to generate and modify skin lesions. We show how this process can be used for powerful data augmentation and visualization for skin health, evaluating it on a downstream application that classifies malignant lesions

Conference Presentation

Citation Download Citation

Alessio Gallucci, Nicola Pezzotti, Dmitry Znamenskiy, and Milan Petkovic "A latent space exploration for microscopic skin lesion augmentations with VQ-VAE-2 and PixelSNAIL", Proc. SPIE 11596, Medical Imaging 2021: Image Processing, 115962X (15 February 2021); https://doi.org/10.1117/12.2580664

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available