Abstract:High Efficiency Video Coding (HEVC) offers superior compression rates, but its adoption introduces increased coding complexity due to its reliance on a recursive quad-tree for partitioning frames into varying block sizes. This quad-tree process is a central feature in upcoming video coding standards. Our paper presents a novel framework, SVG-CNN, which integrates three shallow Convolutional Neural Networks (CNNs) inspired by VGGNet. Each CNN is specifically designed for individual quad-tree levels to predict the Code Unit (CU) partition in HEVC, leading to reduced intra-frame coding time. SVG-CNN has an inherent capability for early terminations, leveraging sequential CNN feeding based on quad-tree level probabilities. This provides a mechanism to halt processes when further refinement is seemed unlikely. Enhancing the model's efficacy, we have crafted three specialized datasets, each focusing on distinct quad-tree levels and quantization parameter (QP) contexts. This allows each CNN within our framework to undergo targeted training, establishing a cutting-edge training methodology. Our study shows that performance, in terms of accuracy and F1 metrics, is highly dependent on QP settings, with lower QPs yielding better results, and higher QPs diminishing performance due to potential loss of critical features. To enhance our model, we tackled hyperparameter selection and CU split threshold determination for HEVC prediction. We utilized Grid Search Cross-Validation for the former and assessed multiple thresholds across selected videos for the latter. The model has a moderate complexity with over 328,000 parameters across 18 layers, which ensures memory efficiency. It boasts a swift prediction time of 0.05 ms and reduces HEVC encoding time by 61.64%, while slightly improving the bitrate-distortion performance by -0.24% BDBR, indicating better compression without notable PSNR loss. Significantly, our approach outperforms other CNN-based quad-tree partitioning methods that reduce HEVC coding complexity but sacrifice compression performance.

Adaptive CU Split Decision with Pooling-variable CNN for VVC Intra Encoding.

Multi-scale and Bi-path Method Based on Image Entropy and CNN for Fast CU Partition in VVC

ResNet-Based Fast CU Partition Decision Algorithm for VVC

Effective CU size decision for HEVC intracoding.

Width-Adaptive CNN: Fast CU Partition Prediction for VVC Screen Content Coding

Fast CU Size Decision and Mode Decision Algorithm for HEVC Intra Coding

Faster Intra-Prediction of Versatile Video Coding Using a Concatenate-Designed CNN via DCT Coefficients

SVG-CNN: A shallow CNN based on VGGNet applied to intra prediction partition block in HEVC

Light-weight CNN-based VVC Inter Partitioning Acceleration

Efficient Intra Coding Through Hierarchical CU Partition Prediction for VVC.

Early Intra CU Size Decision for Versatile Video Coding Based on a Tunable Decision Model

Fast Intra CU Size Decision for HEVC Based on Machine Learning

HG-FCN: Hierarchical Grid Fully Convolutional Network for Fast VVC Intra Coding

A Fast Decision Algorithm for VVC Intra-Coding Based on Texture Feature and Machine Learning

Fast CU Partition Decision Method Based on Bayes and Improved De-Blocking Filter for H.266/VVC

An Effective CU Size Decision Method for HEVC Encoders

Cnn Based Cu Partition Mode Decision Algorithm For Hevc Inter Coding

Fast QTBT Partitioning Decision for Interframe Coding with Convolution Neural Network.

Adaptive Transform Size Decision Algorithm for High-Efficiency Video Coding Inter Coding

Fast QTMT Partition for VVC Intra Coding Using U-Net Framework

Rapid CU Partitioning and Joint Intra-Frame Mode Decision Algorithm