How to Design and Train Your Implicit Neural Representation for Video Compression
This work addresses practical adoption barriers for INR-based video compression by improving encoding speed and quality, though it builds incrementally on existing NeRV methods.
The paper tackles slow encoding speeds in implicit neural representation (INR) methods for video compression by developing a library to analyze NeRV-family components and proposing Rabbit NeRV (RNeRV), which achieves +1.27% average PSNR improvement over alternatives with equal training time, and introduces hyper-networks with masking to improve PSNR and MS-SSIM by 1.7% at 0.037 bpp while addressing real-time encoding.
Implicit neural representation (INR) methods for video compression have recently achieved visual quality and compression ratios that are competitive with traditional pipelines. However, due to the need for per-sample network training, the encoding speeds of these methods are too slow for practical adoption. We develop a library to allow us to disentangle and review the components of methods from the NeRV family, reframing their performance in terms of not only size-quality trade-offs, but also impacts on training time. We uncover principles for effective video INR design and propose a state-of-the-art configuration of these components, Rabbit NeRV (RNeRV). When all methods are given equal training time (equivalent to 300 NeRV epochs) for 7 different UVG videos at 1080p, RNeRV achieves +1.27% PSNR on average compared to the best-performing alternative for each video in our NeRV library. We then tackle the encoding speed issue head-on by investigating the viability of hyper-networks, which predict INR weights from video inputs, to disentangle training from encoding to allow for real-time encoding. We propose masking the weights of the predicted INR during training to allow for variable, higher quality compression, resulting in 1.7% improvements to both PSNR and MS-SSIM at 0.037 bpp on the UCF-101 dataset, and we increase hyper-network parameters by 0.4% for 2.5%/2.7% improvements to PSNR/MS-SSIM with equal bpp and similar speeds. Our project website is available at https://mgwillia.github.io/vinrb/ and our code is available at https://github.com/mgwillia/vinrb.