arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2203.02452eess.IVcs.CVcs.LG

Contextformer:一种用于学习图像压缩中上下文建模的具有空间-通道注意力的Transformer

Contextformer: A Transformer with Spatio-Channel Attention for Context Modeling in Learned Image Compression

Technical University of Munich(慕尼黑工业大学);Huawei Munich Research Center(华为慕尼黑研究中心);Tencent America(腾讯美国公司);Huawei Moscow Research Center(华为莫斯科研究中心)

A. Burakhan Koyuncu, Han Gao, Atanas Boev, Georgii Gaikov, Elena Alshina, Eckehard Steinbach

AI总结:

本文提出Contextformer,将Transformer注意力推广到空间-通道维度以改进图像压缩的熵建模,在多个数据集上相比VTM 16.2节省最高11%码率并提升PSNR和MS-SSIM。

摘要:

熵建模是高性能图像压缩算法的关键组成部分。自回归上下文建模的最新进展帮助基于学习的方法超越了其经典对应方法。然而,由于潜在空间中空间-通道依赖关系未被充分利用,以及上下文自适应性的次优实现,这些模型的性能仍可进一步提升。受Transformer自适应特性的启发,我们提出了一种基于Transformer的上下文模型,命名为Contextformer,它将事实上的标准注意力机制推广到空间-通道注意力。我们将现代压缩框架的上下文模型替换为Contextformer,并在广泛使用的Kodak、CLIC2020和Tecnick图像数据集上进行了测试。实验结果表明,与标准的多功能视频编码(VVC)测试模型(VTM)16.2相比,所提模型可节省高达11%的码率,并且在PSNR和MS-SSIM方面优于各种基于学习的模型。

英文摘要:

Entropy modeling is a key component for high-performance image compression algorithms. Recent developments in autoregressive context modeling helped learning-based methods to surpass their classical counterparts. However, the performance of those models can be further improved due to the underexploited spatio-channel dependencies in latent space, and the suboptimal implementation of context adaptivity. Inspired by the adaptive characteristics of the transformers, we propose a transformer-based context model, named Contextformer, which generalizes the de facto standard attention mechanism to spatio-channel attention. We replace the context model of a modern compression framework with the Contextformer and test it on the widely used Kodak, CLIC2020, and Tecnick image datasets. Our experimental results show that the proposed model provides up to 11% rate savings compared to the standard Versatile Video Coding (VVC) Test Model (VTM) 16.2, and outperforms various learning-based models in terms of PSNR and MS-SSIM.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑