arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.14781hep-exphysics.comp-ph
收藏

MLaaS4HEP:面向HEP的机器学习即服务

MLaaS4HEP: Machine Learning as a Service for HEP

Valentin Kuznetsov, Luca Giommi, Daniele Bonacorsi

摘要:

机器学习(ML)将在CERN即将推出的高亮度LHC(HL-LHC)计划的成功中发挥重要作用。在未来十年中,LHC实验将收集前所未有的大量数据,这将需要新的方法来训练和使用ML模型。在本文中,我们讨论了HEP的机器学习即服务管道(MLaaS 4 HEP),它提供了三个独立的层:数据流层,用于以本地ROOT数据格式读取高能物理(HEP)数据;数据训练层,用于使用分布式ROOT文件训练ML模型;数据推理层,用于通过HTTP协议使用预先训练的ML模型进行预测。这种模块化设计开辟了通过从远程存储设施(例如,全球LHC计算网格(WLCG)基础设施)读取ROOT文件来大规模训练数据的可能性,并将数据馈送到用户喜爱的ML框架。作为TensorFlow即服务(TFaaS)实现的推理层可以提供对HEP域内部或外部的现有基础设施和应用程序中的预训练ML模型的轻松访问。特别是,我们证明了使用的MLaaS 4 HEP架构的物理用例,即最初使用定制的Ntuples进行CMS中的$t\bar{t}$希格斯分析。我们提供了使用分布式ROOT文件训练ML模型的详细信息,讨论了MLaaS和TFaaS方法用于选定物理分析的性能,并将结果与传统方法进行了比较。

英文摘要:

Machine Learning (ML) will play a significant role in the success of the upcoming High-Luminosity LHC (HL-LHC) program at CERN. An unprecedented amount of data at the exascale will be collected by LHC experiments in the next decade, and this effort will require novel approaches to train and use ML models. In this paper, we discuss a Machine Learning as a Service pipeline for HEP (MLaaS4HEP) which provides three independent layers: a data streaming layer to read High-Energy Physics (HEP) data in their native ROOT data format; a data training layer to train ML models using distributed ROOT files; a data inference layer to serve predictions using pre-trained ML models via HTTP protocol. Such modular design opens up the possibility to train data at large scale by reading ROOT files from remote storage facilities, e.g. World-Wide LHC Computing Grid (WLCG) infrastructure, and feed the data to the user's favorite ML framework. The inference layer implemented as TensorFlow as a Service (TFaaS) may provide an easy access to pre-trained ML models in existing infrastructure and applications inside or outside of the HEP domain. In particular, we demonstrate the usage of the MLaaS4HEP architecture for a physics use-case, namely the $t\bar{t}$ Higgs analysis in CMS originally performed using custom made Ntuples. We provide details on the training of the ML model using distributed ROOT files, discuss the performance of the MLaaS and TFaaS approaches for the selected physics analysis, and compare the results with traditional methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑