Checkerboard Context Model for Efficient Learned Image Compression
arXiv:2103.15306
Abstract
For learned image compression, the autoregressive context model is proved effective in improving the rate-distortion (RD) performance. Because it helps remove spatial redundancies among latent representations. However, the decoding process must be done in a strict scan order, which breaks the parallelization. We propose a parallelizable checkerboard context model (CCM) to solve the problem. Our two-pass checkerboard context calculation eliminates such limitations on spatial locations by re-organizing the decoding order. Speeding up the decoding process more than 40 times in our experiments, it achieves significantly improved computational efficiency with almost the same rate-distortion performance. To the best of our knowledge, this is the first exploration on parallelization-friendly spatial context model for learned image compression.
CVPR 2021
References in corpus (6)
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
- Joint Autoregressive and Hierarchical Priors for Learned Image Compression
- Context-adaptive Entropy Model for End-to-end Optimized Image Compression
- An End-to-End Joint Learning Scheme of Image Compression and Quality Enhancement with Improved Entropy Minimization
- Computationally Efficient Neural Image Compression
- Multi-scale and Context-adaptive Entropy Model for Image Compression