Abstract:Recently, learned image compression has achieved impressive performance. The entropy model, which estimates the distribution of the latent representation, plays a crucial role in enhancing rate-distortion performance. However, existing global context modules rely on computationally intensive quadratic complexity computations to capture global correlations. This quadratic complexity imposes limitations on the potential of high-resolution image coding. Moreover, effectively capturing local, global, and channel-wise contexts with acceptable even linear complexity within a single entropy model remains a challenge. To address these limitations, we propose the Linear Complexity Multi-Reference Entropy Model (MEM++). MEM++ effectively captures the diverse range of correlations inherent in the latent representation. Specifically, the latent representation is first divided into multiple slices. When compressing a particular slice, the previously compressed slices serve as its channel-wise contexts. To capture local contexts without sacrificing performance, we introduce a novel checkerboard attention module. Additionally, to capture global contexts, we propose the linear complexity attention-based global correlations capturing by leveraging the decomposition of the softmax operation. The attention map of the previously decoded slice is implicitly computed and employed to predict global correlations in the current slice. Based on MEM++, we propose image compression model MLIC++. Extensive experimental evaluations demonstrate that our MLIC++ achieves state-of-the-art performance, reducing BD-rate by 13.39% on the Kodak dataset compared to VTM-17.0 in PSNR. Furthermore, MLIC++ exhibits linear GPU memory consumption with resolution, making it highly suitable for high-resolution image coding. Code and pre-trained models are available at <a class="link-external link-https" href="https://github.com/JiangWeibeta/MLIC" rel="external noopener nofollow">this https URL</a>.

Learned Image Compression with Discretized Gaussian Mixture Likelihoods and Attention Modules

Learned Image Compression With Gaussian-Laplacian-Logistic Mixture Model and Concatenated Residual Modules

A GAN-based Tunable Image Compression System

Learned image compression via neighborhood-based attention optimization and context modeling with multi-scale guiding

Learned Image Compression with Large Capacity and Low Redundancy of Latent Representation

Learned Lossless Image Compression With Combined Autoregressive Models And Attention Modules

MLIC++: Linear Complexity Multi-Reference Entropy Modeling for Learned Image Compression

MLIC++: Linear Complexity Attention-based Multi-Reference Entropy Modeling for Learned Image Compression

End-to-End Optimized Image Compression with Deep Gaussian Process Regression

GAN-based Image Compression with Improved RDO Process

MLIC: Multi-Reference Entropy Model for Learned Image Compression

Entropy Modeling Via Gaussian Process Regression for Learned Image Compression

Improved deep learning image compression model: performance optimization based on convolutional modules and local attention mechanism

Learned Image Compression with Separate Hyperprior Decoders

Image Compression using only Attention based Neural Networks

Channel-wise Autoregressive Entropy Models for Learned Image Compression

Fast and High-Performance Learned Image Compression With Improved Checkerboard Context Model, Deformable Residual Module, and Knowledge Distillation

Learned Image Compression with Generalized Octave Convolution and Cross-Resolution Parameter Estimation

Joint Global and Local Hierarchical Priors for Learned Image Compression

Gated Context Model with Embedded Priors for Deep Image Compression