EPJ Web of Conferences (Jan 2020)

Machine Learning with ROOT/TMVA

  • Albertsson Kim,
  • An Sitong,
  • Gleyzer Sergei,
  • Moneta Lorenzo,
  • Niermann Joana,
  • Wunsch Stefan,
  • Zampieri Luca,
  • Zapata Mesa Omar Andres

DOI
https://doi.org/10.1051/epjconf/202024506019
Journal volume & issue
Vol. 245
p. 06019

Abstract

Read online

ROOT provides, through TMVA, machine learning tools for data analysis at HEP experiments and beyond. We present recently included features in TMVA and the strategy for future developments in the diversified machine learning landscape. Focus is put on fast machine learning inference, which enables analysts to deploy their machine learning models rapidly on large scale datasets. The new developments are paired with newly designed C++ and Python interfaces supporting modern C++ paradigms and full interoperability in the Python ecosystem. We present as well a new deep learning implementation for convolutional neural network using the cuDNN library for GPU. We show benchmarking results in term of training time and inference time, when comparing with other machine learning libraries such as Keras/Tensorflow.