MobileNVC: Real-time 1080p Neural Video Compression on a Mobile Device
Ties van Rozendaal,Tushar Singhal,Hoang Le,Guillaume Sautiere,Amir Said,Krishna Buska,Anjuman Raha,Dimitris Kalatzis,Hitarth Mehta,Frank Mayer,Liang Zhang,Markus Nagel,Auke Wiggers
2023-11-15
Abstract:Neural video codecs have recently become competitive with standard codecs such as HEVC in the low-delay setting. However, most neural codecs are large floating-point networks that use pixel-dense warping operations for temporal modeling, making them too computationally expensive for deployment on mobile devices. Recent work has demonstrated that running a neural decoder in real time on mobile is feasible, but shows this only for 720p RGB video. This work presents the first neural video codec that decodes 1080p YUV420 video in real time on a mobile device. Our codec relies on two major contributions. First, we design an efficient codec that uses a block-based motion compensation algorithm available on the warping core of the mobile accelerator, and we show how to quantize this model to integer precision. Second, we implement a fast decoder pipeline that concurrently runs neural network components on the neural signal processor, parallel entropy coding on the mobile GPU, and warping on the warping core. Our codec outperforms the previous on-device codec by a large margin with up to 48% BD-rate savings, while reducing the MAC count on the receiver side by $10 \times$. We perform a careful ablation to demonstrate the effect of the introduced motion compensation scheme, and ablate the effect of model quantization.
Image and Video Processing,Computer Vision and Pattern Recognition,Machine Learning