arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2309.11661cs.CVeess.IV

使用掩码稀疏视觉表示的神经图像压缩

Neural Image Compression Using Masked Sparse Visual Representation

Futurewei Technologies Inc.(未来华技术公司)

Wei Jiang, Wei Wang, Yue Chen

AI总结:

针对基于SVR的图像压缩难以实现率失真权衡的问题,本文提出M-AdaCode,通过语义类别基码本自适应加权及掩码与恢复机制控制比特率和失真,并在JPEG-AI上验证有效性。

摘要:

我们研究基于稀疏视觉表示(Sparse Visual Representation,SVR)的神经图像压缩,其中图像被嵌入到由学习到的视觉码本张成的离散潜在空间中。通过与解码器共享码本,编码器传输高效且跨平台稳健的整数字索引,解码器则使用这些索引检索嵌入的潜在特征以进行重建。以往基于SVR的压缩缺乏实现率失真权衡的有效机制,人们只能追求高重建质量或低传输比特率。我们提出一种掩码自适应码本学习(Masked Adaptive Codebook learning,M-AdaCode)方法,该方法对潜在特征子空间施加掩码,以平衡比特率与重建质量。该方法学习一组依赖语义类别的基码本,并对其进行加权组合,以生成用于高质量重建的丰富潜在特征。组合权重由每张输入图像自适应推导,在产生额外传输成本的情况下提供保真度信息。通过在编码器中掩码掉不重要的权重并在解码器中恢复它们,我们可以用重建质量交换传输比特,而掩码率控制着比特率与失真之间的平衡。在标准JPEG-AI数据集上的实验证明了我们的M-AdaCode方法的有效性。

英文摘要:

We study neural image compression based on the Sparse Visual Representation (SVR), where images are embedded into a discrete latent space spanned by learned visual codebooks. By sharing codebooks with the decoder, the encoder transfers integer codeword indices that are efficient and cross-platform robust, and the decoder retrieves the embedded latent feature using the indices for reconstruction. Previous SVR-based compression lacks effective mechanism for rate-distortion tradeoffs, where one can only pursue either high reconstruction quality or low transmission bitrate. We propose a Masked Adaptive Codebook learning (M-AdaCode) method that applies masks to the latent feature subspace to balance bitrate and reconstruction quality. A set of semantic-class-dependent basis codebooks are learned, which are weighted combined to generate a rich latent feature for high-quality reconstruction. The combining weights are adaptively derived from each input image, providing fidelity information with additional transmission costs. By masking out unimportant weights in the encoder and recovering them in the decoder, we can trade off reconstruction quality for transmission bits, and the masking rate controls the balance between bitrate and distortion. Experiments over the standard JPEG-AI dataset demonstrate the effectiveness of our M-AdaCode approach.

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

↑