Multistage Spatial Context Models for Learned Image Compression
Fangzheng Lin, Heming Sun, Jinming Liu, Jiro Katto
Abstract
Recent state-of-the-art Learned Image Compression methods feature spatial context models, achieving great rate-distortion improvements over hyperprior methods. However, the autoregressive context model requires serial decoding, limiting run-time performance. The Checkerboard context model allows parallel decoding at a cost of reduced RD performance. We present a series of multistage spatial context models allowing both fast decoding and better RD performance. We split the latent space into square patches and decode serially within each patch while different patches are decoded in parallel. The proposed method features a comparable decoding speed to Checkerboard while reaching the RD performance of Autoregressive and even also outperforming Autoregressive. Inside each patch, the decoding order must be carefully decided as a bad order negatively impacts performance; therefore, we also propose a decoding order optimization algorithm.
BibTeX
@inproceedings{icassp2023_multistagespatia,
title = {Multistage Spatial Context Models for Learned Image Compression},
author = {Fangzheng Lin and Heming Sun and Jinming Liu and Jiro Katto},
booktitle = {ICASSP 2023},
year = {2023}
}