Machine Learning with ROOT/TMVA

Kim Albertsson, Sitong An, Sergei Gleyzer, Lorenzo Moneta, Joana Niermann, Stefan Wunsch, Luca Zampieri, Omar Andres Zapata Mesa, C. Doglioni, D. Kim, G.A. Stewart, L. Silvestris (+2 others)
2020 EPJ Web of Conferences  
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
more » ... nteroperability 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.
doi:10.1051/epjconf/202024506019 fatcat:cng5i4q42fhwfdhxzc7d2jeefe