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

基于时空Transformer的视频压缩框架

Spatial-Temporal Transformer based Video Compression Framework

Shandong University(山东大学);University of Electronic Science and Technology of China(电子科技大学);Peking University(北京大学)

Yanbo Gao, Wenjia Huang, Shuai Li, Hui Yuan, Mao Ye, Siwei Ma

AI总结:

本文提出STT-VC框架,通过松弛可变形Transformer、多粒度预测模块和空间特征分布先验Transformer,解决运动信息不稳定及模块独立导致的时空冗余问题,相比VTM实现13.5%的BD-Rate节省。

摘要:

近年来,学习型视频压缩(LVC)取得了显著进展。与传统视频编码类似,LVC继承了运动估计/补偿、残差编码等模块,这些模块均由神经网络(NNs)实现。然而,在神经网络框架及其基于梯度反向传播的训练机制下,现有大多数方法往往难以从输入的颜色特征中持续稳定地生成以几何特征形式表示的运动信息。此外,帧间预测与残差编码等模块彼此独立,导致难以高效地充分降低时空冗余。为解决上述问题,本文提出了一种新颖的基于时空Transformer的视频压缩(STT-VC)框架。该框架包含:一个用于运动估计与补偿的松弛可变形Transformer(RDT),其采用基于Uformer的偏移量估计;一个基于多参考帧的多粒度预测(MGP)模块,用于预测细化;以及一个基于空间特征分布先验的Transformer(SFD-T),用于高效的时空联合残差压缩。具体而言,RDT通过深入探究基于相似性的几何运动特征提取与自注意力之间的关系,稳定地估计帧间运动信息。MGP通过有效利用由编码运动信息生成的粗粒度预测特征,融合多参考帧信息。SFD-T通过联合利用残差和时序预测中的空间特征分布来压缩残差信息,以进一步降低时空冗余。实验结果表明,与VTM相比,我们的方法实现了13.5%的BD-Rate节省,取得了最佳结果。

英文摘要:

Learned video compression (LVC) has witnessed remarkable advancements in recent years. Similar as the traditional video coding, LVC inherits motion estimation/compensation, residual coding and other modules, all of which are implemented with neural networks (NNs). However, within the framework of NNs and its training mechanism using gradient backpropagation, most existing works often struggle to consistently generate stable motion information, which is in the form of geometric features, from the input color features. Moreover, the modules such as the inter-prediction and residual coding are independent from each other, making it inefficient to fully reduce the spatial-temporal redundancy. To address the above problems, in this paper, we propose a novel Spatial-Temporal Transformer based Video Compression (STT-VC) framework. It contains a Relaxed Deformable Transformer (RDT) with Uformer based offsets estimation for motion estimation and compensation, a Multi-Granularity Prediction (MGP) module based on multi-reference frames for prediction refinement, and a Spatial Feature Distribution prior based Transformer (SFD-T) for efficient temporal-spatial joint residual compression. Specifically, RDT is developed to stably estimate the motion information between frames by thoroughly investigating the relationship between the similarity based geometric motion feature extraction and self-attention. MGP is designed to fuse the multi-reference frame information by effectively exploring the coarse-grained prediction feature generated with the coded motion information. SFD-T is to compress the residual information by jointly exploring the spatial feature distributions in both residual and temporal prediction to further reduce the spatial-temporal redundancy. Experimental results demonstrate that our method achieves the best result with 13.5% BD-Rate saving over VTM.

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

↑