Skip to main content
SpringerLink
Log in

A fast algorithm of intra prediction modes pruning for HEVC based on decision trees and a new three-step search

  • Published:
Multimedia Tools and Applications Aims and scope Submit manuscript

Abstract

The High Efficiency Video Coding (HEVC) standard is a new generation video coding scheme, succeeding to H.264/AVC. HEVC requires only 50 % bitrate of H.264/AVC at the same perceptual quality by adopting new coding tools and more flexible block structures. HEVC specifies 35 different intra prediction directions that can be associated to different block sizes. Each possible combination needs to be tested within the Rate Distortion (RD) process to enable selecting the optimal intra mode and block splitting depth. This leads to a significant processing weight and therefore any improvement that might be achieved will bring significative increase in the computational efficiency of the algorithm. This paper proposes a novel intra prediction modes pruning method based on decision trees and a new three-step search algorithm, aiming at achieving higher encoding efficiency compared to the standard—HEVC. This fast algorithm is composed of two algorithms. The first algorithm is a modes pruning algorithm depending on decision trees. We first calculate variances of the above side, the left side and all the reference samples of all the PUs (Prediction Units), which are used to divide the PUs into three groups of different candidate intra prediction modes. The first group only includes Planar mode and DC mode, the optimal mode will be selected from the two modes. The second and third groups include 19 and 35 intra modes, respectively. Then the decision trees are trained using the information obtained previously by the software WEKA. The classification process has an accuracy of 85.29 %. The second algorithm is a three-step search algorithm which is defined to be suitable for prediction units classified into class two and class three after the execution of decision trees. The detailed implementations of three-step search algorithms for prediction units belong to those two classes are subtly different. Experimental results verify that, compared with the reference software HM15.0, on average, the proposed algorithm reduces the encoding time by 37.87 % with a slightly decreasing of BD-PSNR (0.058 dB) and increasing of BD-Rate (1.19 %).

This is a preview of subscription content, log in via an institution to check access.

Access this article

Log in via an institution

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Institutional subscriptions

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7
Fig. 8
Fig. 9
Fig. 10
Fig. 11

Similar content being viewed by others

References

  1. Bjontegaard G (2001) Calculation of average PSNR differences between RD-curves. Doc. VCEG-M33 ITU-T Q6/16, Austin, TX, USA

  2. Bossen F (2012) Common test conditions and software reference configurations. In: Joint Collaborative Team on Video Coding (JCT-VC) of ITU-T SG16 WP3 and ISO/IEC JTC1/SC29/WG11 10th Meeting, Stockholm, Sweden

  3. Brahmasury Jain H, Rao KR (2014) Fast intra mode decision in high efficiency video coding. Polibits 50:5–12

    Article  Google Scholar 

  4. Cancellier LH, Seidel I (2014) Energy-efficient Hadamard-based SATD architectures. Integrated Circuits and Systems Design (SBCCI), 2014 27th Symposium on IEEE, 1–6

  5. Cen YF, Wang WL, Yao XW (2015) A fast CU depth decision mechanism for HEVC. Inf Process Lett 115(9):719–724

    Article  MathSciNet  MATH  Google Scholar 

  6. Chau LP, Jing X (2003) Efficient three-step search algorithm for block motion estimation in video coding. IEEE International Conference on Acoustics, Speech, & Signal Processing III:421–424

    Google Scholar 

  7. Correa G, Assuncao P, Agostini L, Silva Cruz L (2014) A method for early-splitting of HEVC inter blocks based on decision trees. Proceedings of the 22nd European on Signal Processing Conference (EUSIPCO), p 276–280.

  8. Correa G, Assuncao P, Agostini LV, Silva Cruz L (2015) Fast HEVC encoding decisions using data mining. IEEE Transactions on Circuits and Systems for Video Technology 25:660–673

    Article  Google Scholar 

  9. Elecard HEVC Analyzer [online]: http://www.elecard.com/en/download/Products.html.

  10. Garrido-Cantos R, Cock JD, Martínez JL, Leuven SV, Cuenca P, Garrido A (2013) Low complexity transcoding algorithm from H.264/AVC-to-SVC using data mining. Eurasip Journal on Advances in Signal Processing 2013:1–24

    Article  Google Scholar 

  11. Gwon D, Choi H, Youn JM (2015) HEVC fast intra mode decision based on edge and SATD cost. Multimedia and Broadcasting (APMediaCast), Asia Pacific Conference on IEEE, 1–5

  12. High Efficiency Video Coding (HEVC) [online]: https://hevc.hhi.fraunhofer.de/svn/svn_HEVC Software/.

  13. Jing X, Chau LP (2004) An efficient three-step search algorithm for block motion estimation. IEEE Transactions on Multimedia 6(3):435–438

    Article  Google Scholar 

  14. Kim I-K, Min J, Lee T, Han W-J, Park J (2012) Block partitioning structure in the HEVC standard. IEEE Transactions on Circuits and Systems for Video Technology 22:1697–1706

    Article  Google Scholar 

  15. Lainema J, Bossen F, Han WJ, Min J, Ugur K (2012) Intra coding of the HEVC standard. IEEE Transactions on Circuits and Systems for Video Technology 22:1792–1801

    Article  Google Scholar 

  16. Lei H, Yang Z, Lei H et al (2013) Fast intra prediction mode decision for high efficiency video coding. Proceedings of International Symposium on Computer Communication Control & Automation 68(12):34–37

    Google Scholar 

  17. Lim K, Lee J, Kim S, Lee S (2015) Fast PU skip and split termination algorithm for HEVC intra prediction. IEEE Transactions on Circuits and Systems for Video Technology 25:1335–1346

    Article  Google Scholar 

  18. Lin YC, Lai JC (2014) Edge density early termination algorithm for HEVC coding tree block. IEEE International Symposium on Computer, Consumer and Control (IS3C), p 39–42

  19. Min B, Cheung RCC (2015) A fast CU size decision algorithm for the HEVC intra encoder. IEEE Transactions on Circuits and Systems for Video Technology 25:892–896

    Article  Google Scholar 

  20. Quinlan JR (1993) C4.5: programs for machine learning. Morgan Kaufmann, 1993

  21. Ruiz-Coll D, Adzic V, Fernandez-Escribano G, Kalva H, Luis Martinez J, Cuenca P (2014) Fast partitioning algorithm for HEVC Intra frame coding using machine learning. IEEE International Conference on Image Processing (ICIP), p 4112–4116

  22. Sang JP (2016) CU encoding depth prediction, early CU splitting termination and fast mode decision for fast HEVC intra-coding. Signal Process Image Commun 42:79–89

    Article  Google Scholar 

  23. Sharabayko MP, Markov NG (2014) Entropy-based intra-coding RDO estimation for HEVC Strategic Technology (IFOST). 2014 9th IEEE International Forum, p 56–59

  24. Shen L, Liu Z, Zhang X, Zhao W, Zhang Z (2013) An effective CU size decision method for HEVC encoders. IEEE Transactions on Multimedia 15:465–470

    Article  Google Scholar 

  25. Shen L, Zhang Z, Liu Z (2014) Effective CU size decision for HEVC intracoding. IEEE Transactions on Image Processing A Publication of the IEEE Signal Processing Society 23(10):4232–4241

    Article  MathSciNet  Google Scholar 

  26. Shi Y, Au OC, Zhang X, Zhang H, Ma R, Jia L (2013) Content based fast prediction unit quadtree depth decision algorithm for HEVC. IEEE International Symposium on Circuits and Systems (ISCAS), p 225–228

  27. Shi W, Jiang X, Song T, Shimamoto T (2014) Edge information based fast selection algorithm for intra prediction of HEVC. IEEE Asia Pacific Conference on Circuits and Systems (APCCAS), p 17–20

  28. Sullivan GJ, Ohm J-R, Han W-J, Wiegand T (2012) Overview of the high efficiency video coding standard. IEEE Transactions on Circuits and Systems for Video Technology 22:1649–1668

    Article  Google Scholar 

  29. Utgoff PE (1989) Incremental induction of decision trees. Mach Learn 4:161–186

    Article  Google Scholar 

  30. Wang LL, Siu WC (2013) Novel adaptive algorithm for intra prediction with compromised modes skipping and signaling processes in HEVC. IEEE Transactions on Circuits and Systems for Video Technology 23:1686–1694

    Article  Google Scholar 

  31. Wiegand T, Sullivan GJ, Luthra A (2013) Overview of the H.264/AVC video coding standard. Proceedings of SPIE - The International Society for Optical Engineering 13:417–431

    Google Scholar 

  32. Witten IH, Hall M, Frank E et al (2009) The WEKA data mining software: an update. Acm Sigkdd Explorations Newsletter 11:10–18

    Article  Google Scholar 

  33. Yun Z, Sam K, Xu W et al (2015) Machine learning-based coding unit depth decisions for flexible complexity allocation in high efficiency video coding. IEEE Trans Image Process 24:2225–2238

    Article  MathSciNet  Google Scholar 

  34. Zhang Y, Kwong S, Wang X, Yuan H, Pan Z, Xu L (2015a) Machine learning-based coding unit depth decisions for flexible complexity allocation in high efficiency video coding. IEEE Trans Image Process 24:2225–2238

    Article  MathSciNet  Google Scholar 

  35. Zhang Q, Sun J, Duan Y, et al (2015b) A two-stage fast CU size decision method for HEVC intracoding. Multimedia Signal Processing (MMSP), 2015 I.E. 17th International Workshop on IEEE

  36. Zhao L, Zhang L, Ma S, et al. (2011) Fast mode decision algorithm for intra prediction in HEVC. Visual Communications and Image Processing (VCIP), IEEE. IEEE, 1–4

  37. Zhou W, Zhou X, Lian X et al (2015) An efficient interpolation filter VLSI architecture for HEVC standard. EURASIP Journal on Advances in Signal Processing 2015(95):1–12

    Google Scholar 

Download references

Acknowledgments

This project is funded by the National Natural Science Foundation of China (NSFC) under grants No. 61375025, No. 61075011, and No. 60675018, also the Scientific Research Foundation for the Returned Overseas Chinese Scholars from the State Education Ministry of China. We express our appreciations to the reviewers for their thorough review and very helpful comments, which help improving this paper.

Author information

Authors and Affiliations

  1. Department of Measurement Control and Information Technology, School of Instrumentation Science and Optoelectronics Engineering, Beihang University, Beijing, 100191, China

    Shiping Zhu & Chunyan Zhang

Authors
  1. Shiping Zhu
    View author publications

    You can also search for this author in PubMed Google Scholar

  2. Chunyan Zhang
    View author publications

    You can also search for this author in PubMed Google Scholar

Corresponding author

Correspondence to Shiping Zhu.

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Zhu, S., Zhang, C. A fast algorithm of intra prediction modes pruning for HEVC based on decision trees and a new three-step search. Multimed Tools Appl 76, 21707–21728 (2017). https://doi.org/10.1007/s11042-016-4056-0

Download citation

  • Received:

  • Revised:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s11042-016-4056-0

Keywords

Search

Navigation