Entropy Modeling Via Gaussian Process Regression for Learned Image Compression

Maida Cao,Wenrui Dai,Shaohui Li,Chenglin Li,Junni Zou,Ying Chen,Hongkai Xiong
DOI: https://doi.org/10.1109/dcc52660.2022.00024
2022-01-01
Abstract:Existing entropy models in learned image compression are cumbersome to generate fixed mean and variance for estimating Gaussian distributions for latent representation. In this paper, we propose a novel entropy model based on Gaussian process regression (GPR) that flexibly predicts the mean of Gaussians with posterior distributions characterized by covari-ance functions spanned in the high-dimensional feature space. Furthermore, we develop the rate-distortion optimization based on the proposed entropy model by approximating the bitrates with the evidence lower bound (ELBO) derived via variational inference for GPR. The proposed model can be seamlessly integrated into existing end-to-end optimized frame-works by substituting the masked convolution based autoregressive models. Experimental results demonstrate that the proposed model outperforms conventional image compression methods such as JPEG2000 and BPG, as well as recent learning based methods on the Kodak dataset in terms of rate-distortion performance.
What problem does this paper attempt to address?