基于神经网络的HEVC帧内预测组合快速算法

范俊宇; 宋立锋

doi:10.12052/gdutxb.230067

欢迎来到能源电力期刊网！

English Version 期刊登录

您当前的位置：

首页 >

文章列表页 >

基于神经网络的HEVC帧内预测组合快速算法

信息与通信技术 | 更新时间：2026-05-27

- 基于神经网络的HEVC帧内预测组合快速算法
- A Fast Combination Algorithm for HEVC Intra-Prediction Based on Neural Network
- 广东工业大学学报 2024年41卷第3期页码：131-140
- 作者机构：
  
  1. 广东工业大学信息工程学院,广东,广州,510006
  2. 惠州市广工大物联网协同创新研究院有限公司,广东,惠州,516025
- 作者简介：
  
  范俊宇 (1997–)，男，硕士研究生，主要研究方向为视频编解码，E-mail：telecom162@163.com
  宋立锋 (1967–)，男，副教授，博士，主要研究方向为视频编解码及传输，E-mail：songlf@gdut.edu.cn
- 基金信息：
  
  广东省科技创新战略专项资金 (省重点实验室认定) 项目 (2021B1212050003)
- DOI：10.12052/gdutxb.230067
  中图分类号： TN919.81
- 网络首发：2024-05-25，
  
  纸质出版：2024
- 稿件说明：
移动端阅览
范俊宇, 宋立锋. 基于神经网络的HEVC帧内预测组合快速算法[J]. 广东工业大学学报, 2024, 41(3): 131-140.

Fan Jun-yu, Song Li-feng. A Fast Combination Algorithm for HEVC Intra-Prediction Based on Neural Network[J]. 2024, 41(3): 131-140.
范俊宇, 宋立锋. 基于神经网络的HEVC帧内预测组合快速算法[J]. 广东工业大学学报, 2024, 41(3): 131-140. DOI： 10.12052/gdutxb.230067.

Fan Jun-yu, Song Li-feng. A Fast Combination Algorithm for HEVC Intra-Prediction Based on Neural Network[J]. 2024, 41(3): 131-140. DOI： 10.12052/gdutxb.230067.

摘要

为了提升高效视频编码(High Efficiency Video Coding

HEVC) 帧内编码的实时性能，本文提出的方法利用了引入偶数边长与步长的卷积核以及自注意力机制的轻量级卷积网络来预测编码树单元(Coding Tree Unit

CTU) 的帧内划分结构，从而减少了编码器对CTU进行四叉树递归遍历划分的编码时间。原始编码策略中粗模式决策通过基于残差经哈德曼变换的预测残差绝对值总和 (Sum of Absolute Transformed Difference

SATD) 的损失值来估计率失真优化过程中的率失真损失值来进行加速，但仍会耗费一定的编码时间。提出一种方法通过采样搜索的方式减少粗模式决策过程中计算的模式数，从35种模式降低到了18种模式，降低了粗模式决策过程中计算估计损失值的时间。由粗模式决策过程得到的较优的多个候选帧内模式来进行率失真优化，为了缩减粗模式决策需要计算的候选模式数，在候选模式列表中根据前后帧内预测角度模式的估计损失值的差距来筛选掉部分可能性较低的候选模式实现早停止决策，从而减少需要进行率失真优化的候选模式数量，进而减少率失真优化过程的计算时间。本文提出的算法在测试序列上平均实现78.15%的编码时间缩减，BD-PSNR为 -0.168 dB，BD-RATE为3.49%。

Abstract

To improve the real-time performance of High Efficiency Video Coding (HEVC) intra-frame encoding

a method

which utilizes a lightweight convolutional network with even-length and step-size convolutional kernels and a self-attention mechanism

is proposed to predict the intra-frame partitioning structure of Coding Tree Units(CTU)

thereby reducing the encoding time required for the encoder to perform quadtree recursive traversal partitioning on CTUs. In the original encoding strategy

Rough Mode Decision accelerates the process by estimating the rate-distortion loss value in Rate Distortion Optimization based on the Sum of Absolute Transformed Difference (SATD) -based loss value

but it still consumes a certain amount of encoding time. A proposed method reduces the number of patterns calculated in the Rough Mode Decision process through a sampling search approach

reducing the number of patterns from 35 to 18

and decreasing the time required to estimate the loss value during the Rough Mode Decision process. The more favorable multiple candidate intra-frame modes obtained from the Rough Mode Decision process are used for Rate Distortion Optimization. In order to reduce the number of candidate modes that need to be calculated in Rate Distortion Optimization

an early stopping decision is implemented by filtering out some less likely candidate modes based on the differences in the estimated loss values of the intra-frame prediction angle modes in the candidate mode list

thus reducing the number of candidate modes that need to be evaluated in Rate Distortion Optimization and consequently decreasing the computation time of the Rate Distortion Optimization process. The proposed algorithm achieves an average encoding time reduction of 78.15% on the test sequences

with a BD-PSNR of -0.168dB and a BD-RATE of 3.49%.

关键词

Keywords

references

LENG J, SUN L, IKENAGA T, et al . Content based hierarchical fast coding unit decision algorithm for HEVC[C ] //2011 International Conference on Multimedia and Signal Processing. Guilin: IEEE, 2011: 56-59.

TANG Y, WANG X D, ZHANG L J. A fast CU partition mode algorithm for HEVC [J]. Wireless Communication Technology, 2020, 29(3): 12-15.

ZHANG Y, KWONG S, JIANG G, et al . Statistical early termination model for fast mode decision and reference frame selection in multiview video coding [J ] . IEEE Transactions on Broadcasting, 2012, 58(1): 10-23.

HE S Q, YU X H, DENG Z J. Efficient H. 265/HEVC fast intra frame coding method [J]. Computer Engineering and Design, 2022, 43(9): 2601-2608.

GUO L, WANG X D, XU B W, et al . Fast intra mode prediction decision and coding unit partition algorithm based on high efficiency video coding [J ] . Journal of Computer Applications, 2018, 38(4): 1157-1163.

SHEN L, ZHANG Z, LIU Z. Effective CU size decision for HEVC intracoding [J]. IEEE Transactions on Image Processing, 2014, 23(10): 4232-4241.

NISHIKORI T, NAKAMURA T, YOSHITOME T, et al. A fast CU decision using image variance in HEVC intra coding[C ] //2013 IEEE Symposium on Industrial Electronics Applications. Kuching: IEEE, 2013: 52-56.

TANG J, PENG Y. Fast coding unit partition algorithm for HEVC based on temporal-spatial correlation and texture property [J]. Computer Digital Engineering, 2019, 47(007): 1753-1756.

WU G J, SONG L F. Fast intra mode and depth decision algorithm for HEVC [J]. Journal of Guangdong University of Technology, 2015, 32(4): 132-137.

JAMALI M, COULOMBE S, CARON F. Fast HEVC intra mode decision based on edge detection and SATD costs classification[C]//2015 Data Compression Conference. Snowbird: IEEE, 2015: 43-52.

MIN B, CHEUNG R. A fast CU size decision algorithm for the HEVC intra encoder [J]. IEEE Transactions on Circuits and Systems for Video Technology, 2014, 25(5): 892-896.

SHI M, XI S H, YI Q M. Improved algorithm for intraframe prediction mode fast selecting in high-efficiency video coding based on size of prediction units [J]. Laser Optoelectronics Progress, 2019, 56(20): 226-234.

CORREA G, ASSUNCAO P A, AGOSTINI L V, et al . Fast HEVC encoding decisions using data mining [J ] . IEEE Transactions on Circuits and Systems for Video Technology, 2014, 25(4): 660-673.

HU Q, SHI Z, ZHANG X, et al . Fast HEVC intra mode decision based on logistic regression classification[C ] //2016 IEEE International Symposium on Broadband Multimedia Systems and Broadcasting (BMSB) . Nara: IEEE, 2016: 1-4.

NAIR P S, RAO K R, NAIR M S. A machine learning approach for fast mode decision in HEVC intra prediction based on statistical features [J]. Journal of Intelligent and Fuzzy Systems, 2019, 36(3): 2095-2106.

LIU D, LIU X, LI Y. Fast CU size decisions for HEVC intra frame coding based on support vector machines[C]//2016 IEEE 14th Intl Conf on Dependable, Autonomic and Secure Computing, 14th Intl Conf on Pervasive Intelligence and Computing, 2nd Intl Conf on Big Data Intelligence and Computing and Cyber Science and Technology Congress (DASC/PiCom/DataCom/CyberSciTech) . Auckland: IEEE, 2016: 594-597.

LIU X, LI Y, LIU D, et al . An adaptive CU size decision algorithm for HEVC intra prediction based on complexity classification using machine learning [J ] . IEEE Transactions on Circuits and Systems for Video Technology, 2017, 29(1): 144-155.

RYU S, KANG J. Machine learning-based fast angular prediction mode decision technique in video coding [J]. IEEE Transactions on Image Processing, 2018, 27(11): 5525-5538.

ZHOU S R, YANG J. Fast partition algorithm of low complexity HEVC intra coding [J]. Journal of Chinese Computer Systems, 2021, 42(7): 1475-1478.

YI Q M, LIN C S, SHI M. Fast HEVC coding units partitioning algorithm based on deep learning [J]. Journal of Chinese Computer Systems, 2021, 42(2): 368-373.

LIU Z, YU X, GAO Y, et al . CU partition mode decision for HEVC hardwired intra encoder using convolution neural network [J ] . IEEE Transactions on Image Processing, 2016, 25(11): 5088-5103.

LAUDE T, OSTERMANN J. Deep learning-based intra prediction mode decision for HEVC[C]//2016 Picture Coding Symposium (PCS) . Nuremberg: IEEE, 2016: 1-5.

CHEN Z, SHI J, LI W. Learned fast HEVC intra coding[J]. 2020 IEEE Transactions on Image Processing, 2020, 29: 5431-5446.

XU M, LI T, WANG Z, et al . Reducing complexity of HEVC: a deep learning approach [J ] . IEEE Transactions on Image Processing, 2018, 27(10): 5044-5059.

JIA K B, CUI T H, LIU P Y, et al . Fast prediction algorithm in high efficiency video coding intra-mode based on deep feature learning [J ] . Journal of Electronics Information Technology, 2021, 43(7): 2023-2031.

浏览量

下载量

CSCD

文章被引用时，请邮件提醒。

提交

工具集

关联资源

针对忆阻神经网络加速器的混合粒度剪枝方法研究

基于分离结果信噪比估计与自适应调频网络的单通道语音分离技术

基于BP神经网络的车辆碳排放测算研究

基于多级残差U-Net的稀疏SPECT图像重建

基于运动特性的帧间预测模式快速选择算法