轻量级分类器动态调整帧率,解决视频编码中运动估计不准问题
Neural B-Frame Coding: Tackling Domain Shift Issues with Lightweight Online Motion Resolution Adaptation
- 用简单状态信号预测最优下采样因子,平衡性能与计算开销
- 三种分类器在不重训练编码器前提下,接近穷举搜索的压缩效果
- 适合追求高效视频编码且需应对场景变化的应用开发者
基于学习的B帧编码器在分层时间预测中常因训练与测试时的图像组(GOP)尺寸不匹配,导致大运动估计不准。现有方法通过下采样将大运动转为小运动,但最优下采样因子需耗时的码率-失真优化。本文提出轻量级分类器,利用当前帧与参考帧的状态信号,预测最佳下采样因子,实现性能与计算成本的平衡。提出三种变体:(1) 采用焦点损失训练的二分类器(Bi-Class),在高低分辨率间选择;(2) 基于码率-失真代价设计软标签的多分类器(Mu-Class);(3) 结合多分类器预测能力与二分类器搜索优势的联合分类器(Co-Class)。所有方法可无缝集成至现有B帧编码器,无需重新训练。实验表明,其性能接近穷举搜索,同时显著降低计算复杂度。代码已开源:https://github.com/NYCU-MAPL/Fast-OMRA.git。
原文摘要 · Abstract (English)
Learned B-frame codecs with hierarchical temporal prediction often encounter the domain-shift issue due to mismatches between the Group-of-Pictures (GOP) sizes for training and testing, leading to inaccurate motion estimates, particularly for large motion. A common solution is to turn large motion into small motion by downsampling video frames during motion estimation. However, determining the optimal downsampling factor typically requires costly rate-distortion optimization. This work introduces lightweight classifiers to predict downsampling factors. These classifiers leverage simple state signals from current and reference frames to balance rate-distortion performance with computational cost. Three variants are proposed: (1) a binary classifier (Bi-Class) trained with Focal Loss to choose between high and low resolutions, (2) a multi-class classifier (Mu-Class) trained with novel soft labels based on rate-distortion costs, and (3) a co-class approach (Co-Class) that combines the predictive capability of the multi-class classifier with the selective search of the binary classifier. All classifier methods can work seamlessly with existing B-frame codecs without requiring codec retraining. Experimental results show that they achieve coding performance comparable to exhaustive search methods while significantly reducing computational complexity. The code is available at: https://github.com/NYCU-MAPL/Fast-OMRA.git.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。