Mar 5 – 8, 2024
Lahan Select Gyeongju, South Korea
Asia/Seoul timezone

AI/ML Coupling & Surrogates in BLAST Accelerator Modeling Codes

Mar 7, 2024, 5:40 PM
20m
Lahan Select Gyeongju, South Korea

Lahan Select Gyeongju, South Korea

Lahan Select Gyeongju, South Korea
Oral (16mins + 4 mins) Methods Methods

Speaker

Dr Axel Huebl (Lawrence Berkeley National Laboratory)

Description

Detailed modeling of particle accelerators can benefit from parallelization on modern compute hardware such as GPUs and can often be distributed to large supercomputers. Providing production-quality implementations, the Beam, Plasma & Accelerator Simulation Toolkit (BLAST) provides multiple modern codes to cover the widely different time and length scales between conventional accelerator elements and advanced, plasma-based elements. The Exascale code WarpX provides electromagnetic and -static, t-based particle-in-cell routines, advanced algorithms and is highly scalable. For beam-dynamics, the s-based ImpactX code provides an efficient implementation for tracking relative to a nominal reference trajectory, including space charge. Yet, integrated modeling of "hybrid" beamlines integrating both detailed plasma models and large-scale transport at full detail require exchange between both codes and are limited by the computational speed of the most-detailed element, usually the plasma element.

In this work, we present an alternative approach to coupling particle-in-cell models and codes beyond direct data exchange or reduced details for accelerator modeling. In particular, we investigate and demonstrate detailed data-driven modeling based on high-quality WarpX simulations that were used to train surrogate models for the beam transport code ImpactX. We describe new workflows, illuminate predictive quality, performance and applicability to central research topics in advanced accelerator research, such as staging of laser-wakefield accelerators.

Primary Keyword surrogate model architecture
Secondary Keyword MLOps
Tertiary Keyword digital twins

Primary authors

Dr Axel Huebl (Lawrence Berkeley National Laboratory) Dr Ryan T Sandberg (Lawrence Berkeley National Laboratory) Dr Remi Lehe (Lawrence Berkeley National Laboratory) Dr Chad E Mitchell (Lawrence Berkeley National Laboratory) Dr Marco Garten (Lawrence Berkeley National Laboratory) Dr Ji Qiang (Lawrence Berkeley National Laboratory) Dr Jean-Luc Vay (Lawrence Berkeley National Laboratory)

Presentation materials