Laplacian-guided Entropy Model in Neural Codec with Blur-dissipated Synthesis

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Khoshkhahtinat, Atefeh, Zafari, Ali, Mehta, Piyush M., Nasrabadi, Nasser M.
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866917621534818304
author Khoshkhahtinat, Atefeh
Zafari, Ali
Mehta, Piyush M.
Nasrabadi, Nasser M.
author_facet Khoshkhahtinat, Atefeh
Zafari, Ali
Mehta, Piyush M.
Nasrabadi, Nasser M.
contents While replacing Gaussian decoders with a conditional diffusion model enhances the perceptual quality of reconstructions in neural image compression, their lack of inductive bias for image data restricts their ability to achieve state-of-the-art perceptual levels. To address this limitation, we adopt a non-isotropic diffusion model at the decoder side. This model imposes an inductive bias aimed at distinguishing between frequency contents, thereby facilitating the generation of high-quality images. Moreover, our framework is equipped with a novel entropy model that accurately models the probability distribution of latent representation by exploiting spatio-channel correlations in latent space, while accelerating the entropy decoding step. This channel-wise entropy model leverages both local and global spatial contexts within each channel chunk. The global spatial context is built upon the Transformer, which is specifically designed for image compression tasks. The designed Transformer employs a Laplacian-shaped positional encoding, the learnable parameters of which are adaptively adjusted for each channel cluster. Our experiments demonstrate that our proposed framework yields better perceptual quality compared to cutting-edge generative-based codecs, and the proposed entropy model contributes to notable bitrate savings.
format Preprint
id arxiv_https___arxiv_org_abs_2403_16258
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Laplacian-guided Entropy Model in Neural Codec with Blur-dissipated Synthesis
Khoshkhahtinat, Atefeh
Zafari, Ali
Mehta, Piyush M.
Nasrabadi, Nasser M.
Image and Video Processing
Computer Vision and Pattern Recognition
Information Theory
Machine Learning
While replacing Gaussian decoders with a conditional diffusion model enhances the perceptual quality of reconstructions in neural image compression, their lack of inductive bias for image data restricts their ability to achieve state-of-the-art perceptual levels. To address this limitation, we adopt a non-isotropic diffusion model at the decoder side. This model imposes an inductive bias aimed at distinguishing between frequency contents, thereby facilitating the generation of high-quality images. Moreover, our framework is equipped with a novel entropy model that accurately models the probability distribution of latent representation by exploiting spatio-channel correlations in latent space, while accelerating the entropy decoding step. This channel-wise entropy model leverages both local and global spatial contexts within each channel chunk. The global spatial context is built upon the Transformer, which is specifically designed for image compression tasks. The designed Transformer employs a Laplacian-shaped positional encoding, the learnable parameters of which are adaptively adjusted for each channel cluster. Our experiments demonstrate that our proposed framework yields better perceptual quality compared to cutting-edge generative-based codecs, and the proposed entropy model contributes to notable bitrate savings.
title Laplacian-guided Entropy Model in Neural Codec with Blur-dissipated Synthesis
topic Image and Video Processing
Computer Vision and Pattern Recognition
Information Theory
Machine Learning
url https://arxiv.org/abs/2403.16258