arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2309.10799eess.IVcs.CVcs.LG

多上下文双超先验神经图像压缩

Multi-Context Dual Hyper-Prior Neural Image Compression

West Virginia University(西弗吉尼亚大学)

Atefeh Khoshkhahtinat, Ali Zafari, Piyush M. Mehta, Mohammad Akyash, Hossein Kashiani, Nasser M. Nasrabadi

AI总结:

本文提出基于Transformer的变换和结合双超先验与全局因果注意力的熵模型,用于神经图像压缩,在率失真性能上超越现有方法。

摘要:

变换模型和熵模型是深度图像压缩神经网络中的两个核心组成部分。现有的大多数基于学习的图像压缩方法采用基于卷积的变换,由于卷积操作的感受野有限,缺乏建模长距离依赖的能力。为了解决这一局限性,我们提出了一种基于Transformer的非线性变换。该变换具有高效捕获输入图像局部和全局信息的显著能力,从而得到更去相关的潜在表示。此外,我们引入了一种新颖的熵模型,该模型结合两种不同的超先验来建模潜在表示的跨通道依赖和空间依赖。为了进一步改进熵模型,我们添加了一个全局上下文,利用远距离关系来更准确地预测当前潜在表示。该全局上下文采用因果注意力机制,以内容相关的方式提取长距离信息。我们的实验表明,所提出的框架在率失真性能方面优于最先进的方法。

英文摘要:

Transform and entropy models are the two core components in deep image compression neural networks. Most existing learning-based image compression methods utilize convolutional-based transform, which lacks the ability to model long-range dependencies, primarily due to the limited receptive field of the convolution operation. To address this limitation, we propose a Transformer-based nonlinear transform. This transform has the remarkable ability to efficiently capture both local and global information from the input image, leading to a more decorrelated latent representation. In addition, we introduce a novel entropy model that incorporates two different hyperpriors to model cross-channel and spatial dependencies of the latent representation. To further improve the entropy model, we add a global context that leverages distant relationships to predict the current latent more accurately. This global context employs a causal attention mechanism to extract long-range information in a content-dependent manner. Our experiments show that our proposed framework performs better than the state-of-the-art methods in terms of rate-distortion performance.

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

↑