Good, Cheap, and Fast: Overfitted Image Compression with Wasserstein Distortion

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Ballé, Jona, Versari, Luca, Dupont, Emilien, Kim, Hyunjik, Bauer, Matthias
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916660136378368
author Ballé, Jona
Versari, Luca
Dupont, Emilien
Kim, Hyunjik
Bauer, Matthias
author_facet Ballé, Jona
Versari, Luca
Dupont, Emilien
Kim, Hyunjik
Bauer, Matthias
contents Inspired by the success of generative image models, recent work on learned image compression increasingly focuses on better probabilistic models of the natural image distribution, leading to excellent image quality. This, however, comes at the expense of a computational complexity that is several orders of magnitude higher than today's commercial codecs, and thus prohibitive for most practical applications. With this paper, we demonstrate that by focusing on modeling visual perception rather than the data distribution, we can achieve a very good trade-off between visual quality and bit rate similar to "generative" compression models such as HiFiC, while requiring less than 1% of the multiply-accumulate operations (MACs) for decompression. We do this by optimizing C3, an overfitted image codec, for Wasserstein Distortion (WD), and evaluating the image reconstructions with a human rater study, showing that WD clearly outperforms LPIPS as an optimization objective. The study also reveals that WD outperforms other perceptual metrics such as LPIPS, DISTS, and MS-SSIM as a predictor of human ratings, remarkably achieving over 94% Pearson correlation with Elo scores.
format Preprint
id arxiv_https___arxiv_org_abs_2412_00505
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Good, Cheap, and Fast: Overfitted Image Compression with Wasserstein Distortion
Ballé, Jona
Versari, Luca
Dupont, Emilien
Kim, Hyunjik
Bauer, Matthias
Computer Vision and Pattern Recognition
Image and Video Processing
Inspired by the success of generative image models, recent work on learned image compression increasingly focuses on better probabilistic models of the natural image distribution, leading to excellent image quality. This, however, comes at the expense of a computational complexity that is several orders of magnitude higher than today's commercial codecs, and thus prohibitive for most practical applications. With this paper, we demonstrate that by focusing on modeling visual perception rather than the data distribution, we can achieve a very good trade-off between visual quality and bit rate similar to "generative" compression models such as HiFiC, while requiring less than 1% of the multiply-accumulate operations (MACs) for decompression. We do this by optimizing C3, an overfitted image codec, for Wasserstein Distortion (WD), and evaluating the image reconstructions with a human rater study, showing that WD clearly outperforms LPIPS as an optimization objective. The study also reveals that WD outperforms other perceptual metrics such as LPIPS, DISTS, and MS-SSIM as a predictor of human ratings, remarkably achieving over 94% Pearson correlation with Elo scores.
title Good, Cheap, and Fast: Overfitted Image Compression with Wasserstein Distortion
topic Computer Vision and Pattern Recognition
Image and Video Processing
url https://arxiv.org/abs/2412.00505