2-5 October 2024
Lufthansa Seeheim Conference Hotel
Europe/Berlin timezone

Cheetah – A High-speed Differentiable Beam Dynamics Simulation for Machine Learning Applications

3 Oct 2024, 16:30
20m
Living Room 1+2

Living Room 1+2

Contributed talk D-1 Beam Dynamics Simulations Sessions in Living Room 1+2

Speaker

Chenran Xu (KIT)

Description

Machine learning has emerged as a powerful solution to the modern challenges in accelerator physics. However, the limited availability of beam time and the high computational cost of simulation codes pose significant hurdles in generating the necessary data for training state-of-the-art machine learning models. Furthermore, optimisation methods can be used to tune accelerators and perform complex system identification tasks. However, they too require large numbers of samples of expensive-to-compute objective functions in order to achieve state-of-the-art performance. In this work, we introduce Cheetah, a PyTorch-based high-speed differentiable linear-beam dynamics code that enables fast collection of large datasets and sample-efficient gradient-based optimisation, while being easy to use, straightforward to extend and integrating seamlessly with widely adopted machine learning tools. Ultimately, we believe that Cheetah will simplify the development of machine learning-based methods for particle accelerators and fast-track their integration into everyday operations of accelerator facilities.

Primary authors

Chenran Xu (KIT) Jan Kaiser (DESY) Annika Eichler (DESY) Andrea Santamaria (KIT)

Presentation Materials