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.

CNN-based intra partitioning process for spatial and SNR scalability for SHVC

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

Light-weight CNN-based VVC Inter Partitioning Acceleration

Low-complexity CNN-based CU partitioning for intra frames

Machine Learning-Based approaches to reduce HEVC intra coding unit partition decision complexity

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

CNN-based Partitioning Structure Prediction for VVC Intra Speedup: Bottom-Up-based and Top-Down-based.

Cnn-Based Depth Map Prediction for Fast Block Partitioning in HEVC Intra Coding.

Fast HEVC intra-CU decision partition algorithm with modified LeNet-5 and AlexNet

Intra Block Partition Structure Prediction Via Convolutional Neural Network.

Cnn Based Cu Partition Mode Decision Algorithm For Hevc Inter Coding

Partition-Aware Adaptive Switching Neural Networks for Post-Processing in HEVC

ResNet-Based Fast CU Partition Decision Algorithm for VVC

Effective CU size decision for HEVC intracoding.

CNN Oriented Fast CU Partition Decision and PU Mode Decision for HEVC Intra Encoding

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

Encoder-Decoder-Based Intra-Frame Block Partitioning Decision

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

A Method to Reduce the Intra-Frame Prediction Complexity of HEVC Based on D-CNN

Fcd-cnn: FPGA-based CU depth decision for HEVC intra encoder using CNN

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