Bridging the gap between machine learning and particle accelerator physics with high-speed, differentiable simulations
Jan Kaiser,Chenran Xu,Annika Eichler,Andrea Santamaria Garcia
DOI: https://doi.org/10.1103/physrevaccelbeams.27.054601
2024-05-29
Physical Review Accelerators and Beams
Abstract:Machine learning has emerged as a powerful solution to the modern challenges in accelerator physics. However, the limited availability of beam time, the computational cost of simulations, and the high dimensionality of optimization problems pose significant challenges in generating the required data for training state-of-the-art machine learning models. In this work, we introduce c heetah, a p y t orch-based high-speed differentiable linear beam dynamics code. c heetah enables the fast collection of large datasets by reducing computation times by multiple orders of magnitude and facilitates efficient gradient-based optimization for accelerator tuning and system identification. This positions c heetah as a user-friendly, readily extensible tool that integrates seamlessly with widely adopted machine learning tools. We showcase the utility of c heetah through five examples, including reinforcement learning training, gradient-based beamline tuning, gradient-based system identification, physics-informed Bayesian optimization priors, and modular neural network surrogate modeling of space charge effects. The use of such a high-speed differentiable simulation code will simplify the development of machine learning-based methods for particle accelerators and fast-track their integration into everyday operations of accelerator facilities. https://doi.org/10.1103/PhysRevAccelBeams.27.054601 Published by the American Physical Society under the terms of the Creative Commons Attribution 4.0 International license. Further distribution of this work must maintain attribution to the author(s) and the published article's title, journal citation, and DOI. Published by the American Physical Society
physics, particles & fields, nuclear