Cargando…

A fast inference engine for Boosted Decision Trees

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 poss...

Descripción completa

Detalles Bibliográficos
Autor principal: Zampieri, Luca
Lenguaje:eng
Publicado: 2019
Materias:
Acceso en línea:http://cds.cern.ch/record/2688585
_version_ 1780963677200973824
author Zampieri, Luca
author_facet Zampieri, Luca
author_sort Zampieri, Luca
collection CERN
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.
id cern-2688585
institution Organización Europea para la Investigación Nuclear
language eng
publishDate 2019
record_format invenio
spelling cern-26885852019-11-27T14:19:29Zhttp://cds.cern.ch/record/2688585engZampieri, LucaA fast inference engine for Boosted Decision TreesComputing and ComputersDecision 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.CERN-STUDENTS-Note-2019-183oai:cds.cern.ch:26885852019-09-06
spellingShingle Computing and Computers
Zampieri, Luca
A fast inference engine for Boosted Decision Trees
title A fast inference engine for Boosted Decision Trees
title_full A fast inference engine for Boosted Decision Trees
title_fullStr A fast inference engine for Boosted Decision Trees
title_full_unstemmed A fast inference engine for Boosted Decision Trees
title_short A fast inference engine for Boosted Decision Trees
title_sort fast inference engine for boosted decision trees
topic Computing and Computers
url http://cds.cern.ch/record/2688585
work_keys_str_mv AT zampieriluca afastinferenceengineforboosteddecisiontrees
AT zampieriluca fastinferenceengineforboosteddecisiontrees