There is a newer version of the record available.

Published September 6, 2019 | Version v1

A fast inference engine for Boosted Decision Trees

Authors/Creators

  • 1. ROR icon European Organization for Nuclear Research

Description

Decision trees and derivatives such as Boosted Decision Trees, Adaboost, XG- boost, Random forest are widely used in the world, and are now part of the High Energy Physics toolbox. However, High Energy Physics has some spe- cial requirements, it needs the inference of such tree to be as fast as possible, to be used during reconstruction of events or even trigger. In this work, we design and implement a fast inference engine for decision trees in C++ and benchmark it against XGBoost C API, showing a >15x improvement event-by-event and a 3x improvement for batched versions. The code is freely available on github. This is realized in the context of the CERN summer student program for the ROOT-TMVA project.

Files

AA_main.pdf

Files (379.6 kB)

Name Size Download all
md5:6a2683d10dc814c547f190cb2a122371
379.6 kB Preview Download

Additional details

Identifiers

CDS Report Number
CERN-STUDENTS-Note-2019-183

Linked records