Sample what you cant compress

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Birodkar, Vighnesh, Barcik, Gabriel, Lyon, James, Ioffe, Sergey, Minnen, David, Dillon, Joshua V.
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914053078646784
author Birodkar, Vighnesh
Barcik, Gabriel
Lyon, James
Ioffe, Sergey
Minnen, David
Dillon, Joshua V.
author_facet Birodkar, Vighnesh
Barcik, Gabriel
Lyon, James
Ioffe, Sergey
Minnen, David
Dillon, Joshua V.
contents For learned image representations, basic autoencoders often produce blurry results. Reconstruction quality can be improved by incorporating additional penalties such as adversarial (GAN) and perceptual losses. Arguably, these approaches lack a principled interpretation. Concurrently, in generative settings diffusion has demonstrated a remarkable ability to create crisp, high quality results and has solid theoretical underpinnings (from variational inference to direct study as the Fisher Divergence). Our work combines autoencoder representation learning with diffusion and is, to our knowledge, the first to demonstrate jointly learning a continuous encoder and decoder under a diffusion-based loss and showing that it can lead to higher compression and better generation. We demonstrate that this approach yields better reconstruction quality as compared to GAN-based autoencoders while being easier to tune. We also show that the resulting representation is easier to model with a latent diffusion model as compared to the representation obtained from a state-of-the-art GAN-based loss. Since our decoder is stochastic, it can generate details not encoded in the otherwise deterministic latent representation; we therefore name our approach "Sample what you can't compress", or SWYCC for short.
format Preprint
id arxiv_https___arxiv_org_abs_2409_02529
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Sample what you cant compress
Birodkar, Vighnesh
Barcik, Gabriel
Lyon, James
Ioffe, Sergey
Minnen, David
Dillon, Joshua V.
Machine Learning
Computer Vision and Pattern Recognition
For learned image representations, basic autoencoders often produce blurry results. Reconstruction quality can be improved by incorporating additional penalties such as adversarial (GAN) and perceptual losses. Arguably, these approaches lack a principled interpretation. Concurrently, in generative settings diffusion has demonstrated a remarkable ability to create crisp, high quality results and has solid theoretical underpinnings (from variational inference to direct study as the Fisher Divergence). Our work combines autoencoder representation learning with diffusion and is, to our knowledge, the first to demonstrate jointly learning a continuous encoder and decoder under a diffusion-based loss and showing that it can lead to higher compression and better generation. We demonstrate that this approach yields better reconstruction quality as compared to GAN-based autoencoders while being easier to tune. We also show that the resulting representation is easier to model with a latent diffusion model as compared to the representation obtained from a state-of-the-art GAN-based loss. Since our decoder is stochastic, it can generate details not encoded in the otherwise deterministic latent representation; we therefore name our approach "Sample what you can't compress", or SWYCC for short.
title Sample what you cant compress
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2409.02529