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.

Machine learning approaches with HEVC intra prediction on CU partition for complexity reduction

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

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

High Speed Coding Unit Depth Identification Based on Texture Image Information Using SVM

Effective CU size decision for HEVC intracoding.

Fast Intra CU Size Decision for HEVC Based on Machine Learning

SVM based approach for complexity control of HEVC intra coding

Fast CU partitioning algorithm for HEVC intra coding using data mining

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

Machine Learning-Based H.264/AVC to HEVC Transcoding Via Motion Information Reuse and Coding Mode Similarity Analysis.

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

Fast CU Partition Decision Using Machine Learning for Screen Content Compression.

An Effective CU Size Decision Method for HEVC Encoders

Texture and Correlation Based Fast Intra Prediction Algorithm for HEVC

Low-complexity CNN-based CU partitioning for intra frames

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

Machine learning-based coding unit depth decisions for flexible complexity allocation in high efficiency video coding

Fast CU-splitting decisions based on data mining

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

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

Fast CU partition decision for H.266/VVC based on the improved DAG-SVM classifier model