Enhance the HEVC Fast Intra CU Mode Decision Based on Convolutional Neural Network by Corner Power Estimation

Liangliang Chang,Zhenyu Liu,Libo Wang,Xiaobo Li
DOI: https://doi.org/10.1109/dcc.2018.00053
2018-01-01
Abstract:HEVC (high efficiency video coding) uses block coding regime to fulfill the goal that allocates coding unit (CU) size according to the image complexity. It maintains a competitive visual quality with low bitrate. The process to decide the CU size, which is called rate distortion optimization (RDO), consumes a large amount of calculations. In this paper, we embed CNN into HEVC codec on efficiently performing the CU size decision and decreasing the amount of CU for full RDO process. Firstly, a corner-edge power estimate algorithm is proposed for early CU size decision and preparing balanced data for CNN training and inference. Secondly, the CNN with a deeper structure and stronger expressive ability is used in our mission. We proposed a method to adapt the learning rate according to different CNN layers for more effective training. Our algorithm saves overall 59.7% Intra encoding time, whereas the average BDBR increasing is merely 2.4%.
What problem does this paper attempt to address?