Saved in:
Bibliographic Details
Main Authors: NguyenQuang, Sang, Gao, Zong-Lin, Ho, Kuan-Wei, HoangVan, Xiem, Peng, Wen-Hsiao
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2410.21763
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909370416103424
author NguyenQuang, Sang
Gao, Zong-Lin
Ho, Kuan-Wei
HoangVan, Xiem
Peng, Wen-Hsiao
author_facet NguyenQuang, Sang
Gao, Zong-Lin
Ho, Kuan-Wei
HoangVan, Xiem
Peng, Wen-Hsiao
contents Most learned B-frame codecs with hierarchical temporal prediction suffer from the domain shift issue caused by the discrepancy in the Group-of-Pictures (GOP) size used for training and test. As such, the motion estimation network may fail to predict large motion properly. One effective strategy to mitigate this domain shift issue is to downsample video frames for motion estimation. However, finding the optimal downsampling factor involves a time-consuming rate-distortion optimization process. This work introduces lightweight classifiers to determine the downsampling factor. To strike a good rate-distortion-complexity trade-off, our classifiers observe simple state signals, including only the coding and reference frames, to predict the best downsampling factor. We present two variants that adopt binary and multi-class classifiers, respectively. The binary classifier adopts the Focal Loss for training, classifying between motion estimation at high and low resolutions. Our multi-class classifier is trained with novel soft labels incorporating the knowledge of the rate-distortion costs of different downsampling factors. Both variants operate as add-on modules without the need to re-train the B-frame codec. Experimental results confirm that they achieve comparable coding performance to the brute-force search methods while greatly reducing computational complexity.
format Preprint
id arxiv_https___arxiv_org_abs_2410_21763
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Fast-OMRA: Fast Online Motion Resolution Adaptation for Neural B-Frame Coding
NguyenQuang, Sang
Gao, Zong-Lin
Ho, Kuan-Wei
HoangVan, Xiem
Peng, Wen-Hsiao
Computer Vision and Pattern Recognition
Image and Video Processing
Most learned B-frame codecs with hierarchical temporal prediction suffer from the domain shift issue caused by the discrepancy in the Group-of-Pictures (GOP) size used for training and test. As such, the motion estimation network may fail to predict large motion properly. One effective strategy to mitigate this domain shift issue is to downsample video frames for motion estimation. However, finding the optimal downsampling factor involves a time-consuming rate-distortion optimization process. This work introduces lightweight classifiers to determine the downsampling factor. To strike a good rate-distortion-complexity trade-off, our classifiers observe simple state signals, including only the coding and reference frames, to predict the best downsampling factor. We present two variants that adopt binary and multi-class classifiers, respectively. The binary classifier adopts the Focal Loss for training, classifying between motion estimation at high and low resolutions. Our multi-class classifier is trained with novel soft labels incorporating the knowledge of the rate-distortion costs of different downsampling factors. Both variants operate as add-on modules without the need to re-train the B-frame codec. Experimental results confirm that they achieve comparable coding performance to the brute-force search methods while greatly reducing computational complexity.
title Fast-OMRA: Fast Online Motion Resolution Adaptation for Neural B-Frame Coding
topic Computer Vision and Pattern Recognition
Image and Video Processing
url https://arxiv.org/abs/2410.21763