Machine Learning with ROOT/TMVA

¹ CERN
² Lulea University of Technology
³ Carnegie Mellon University
⁴ University of Alabama
⁵ Karlsruhe Institute of Technology

^* e-mail: stefan.wunsch@cern.ch

Published online: 16 November 2020

Abstract

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.