Distributed Machine Learning Workflow with PanDA and iDDS in LHC ATLAS

¹ Brookhaven National Laboratory, Upton, NY, USA
² University of Wisconsin-Madison, Madison, USA
³ University of Texas at Arlington, Arlington, TX, USA
⁴ University of Pittsburgh, Pittsburgh, PA, USA

^* e-mail: wguan2@bnl.gov

Published online: 6 May 2024

Abstract

Machine Learning (ML) has become one of the important tools for High Energy Physics analysis. As the size of the dataset increases at the Large Hadron Collider (LHC), and at the same time the search spaces become bigger and bigger in order to exploit the physics potentials, more and more computing resources are required for processing these ML tasks. In addition, complex advanced ML workflows are developed in which one task may depend on the results of previous tasks. How to make use of vast distributed CPUs/GPUs in WLCG for these big complex ML tasks has become a popular research area. In this paper, we present our efforts enabling the execution of distributed ML workflows on the Production and Distributed Analysis (PanDA) system and intelligent Data Delivery Service (iDDS). First, we describe how PanDA and iDDS deal with large-scale ML workflows, including the implementation to process workloads on diverse and geographically distributed computing resources. Next, we report real-world use cases, such as HyperParameter Optimization, Monte Carlo Toy confidence limits calculation, and Active Learning. Finally, we conclude with future plans.