Abstract
Variable-rate mechanism has improved the flexibility and efficiency of learning-based image compression that trains multiple models for different rate-distortion tradeoffs. One of the most common approaches for variable-rate is to channel-wisely or spatial-uniformly scale the internal features. However, the diversity of spatial importance is instructive for bit allocation of image compression. In this paper, we introduce a Spatial Importance Guided Variable-rate Image Compression (SigVIC), in which a spatial gating unit (SGU) is designed for adaptively learning a spatial importance mask. Then, a spatial scaling network (SSN) takes the spatial importance mask to guide the feature scaling and bit allocation for variable-rate. Moreover, to improve the quality of decoded image, Top-K shallow features are selected to refine the decoded features through a shallow feature fusion module (SFFM). Experiments show that our method outperforms other learning-based methods (whether variable-rate or not) and traditional codecs, with storage saving and high flexibility.
Abstract (translated)
可变速率机制已经改进了基于学习的图像压缩的灵活性和效率,该方法训练了多个模型以不同的速率失真权衡。最常见的可变速率方法之一是按通道或空间均匀尺度调整内部特征。然而,空间重要性的多样性对于图像压缩中的比特分配具有重要的启示作用。在本文中,我们介绍了一种基于空间重要性的可变速率图像压缩方法(SigVIC),其中空间限制单元(SGU)旨在自适应学习空间重要性掩码。然后,空间缩放网络(SSN)使用空间重要性掩码指导可变速率的特征缩放和比特分配。此外,为了提高解码图像的质量,我们选择K浅特征进行优化,通过浅特征融合模块(SFFM)进行解码特征的精化。实验结果表明,我们的方法比其他任何基于学习的方法(无论是可变速率还是非可变速率)以及传统编码器,具有存储节省和高灵活性。
URL
https://arxiv.org/abs/2303.09112