7–12 May 2023
Venice, Italy
Europe/Zurich timezone

Differentiable beam optics optimization and measurement

WEPL004
10 May 2023, 16:30
2h
Sala Laguna

Sala Laguna

Poster Presentation MC5.D01: Beam Optics Lattices, Correction Schemes, Transport Wednesday Poster Session

Speakers

Nikita Kuklev (Argonne National Laboratory) Vadim Sajaev (Argonne National Laboratory)

Description

Particle accelerators require extensive optics measurement and correction. Due to the complexity of analytic treatments, numerical optimizations are often employed. A disadvantage of this approach is the lack of gradients, limiting optimization methods to derivative-free ones such as simplex or genetic algorithms. We explore a reformulation of beam optics that preserves gradient information by making use of efficient automatic differentiation tools from machine learning frameworks. First, standard beam dynamics computations are converted to a graph of operations on tensors that calculates objectives. Backpropagation is then performed to find parameter gradients and higher order derivatives. Using gradient-aware optimizer algorithms, we showed improved performance in beamline optics matching over existing tools. We also demonstrated an important use of differentiable models in Bayesian inference, whereby probabilistic estimates of magnet parameters and linear optics functions can be obtained from experimental measurements. Our results on test problems showed robust performance and estimates in agreement with standard LOCO methods.

Funding Agency

The work is supported by the U.S. Department of Energy, Office of Science, Office of Basic Energy Sciences, under Contract No. DE-AC02-06CH11357.

I have read and accept the Privacy Policy Statement Yes

Primary author

Nikita Kuklev (Argonne National Laboratory)

Presentation materials

There are no materials yet.