Symplectic neural surrogate models for beam dynamics

Abstract

Development of robust machine-learning (ML) based surrogates for particle accelerators can significantly benefit the modeling, design, optimization, monitoring and control of such accelerators. It is desirable that the surrogate models embed fundamental physical constraints to the interaction and dynamics of the beams, for which an accelerator must be designed to operate upon. We implement and train a class of phase space structure-preserving neural networks — Henon Neural Networks (HenonNets) [1], for nonlinear beam dynamics problems. It is demonstrated that the trained HenonNet model predicts the beam transfer matrix to a reasonably good accuracy while strongly maintaining the symplecticity. To explore such model’s applicability and flexibility for high brightness or intensity beams, we further test it with beam dynamics in the presence of electrostatic and radiative collective effects. Our results indicate that HenonNet may be used as a base ML model for the surrogate of complex beam dynamics, thus opening up a wide range of applications.