Bayesian Optimization of a Free-Electron Laser
Joseph Duris, Dylan Kennedy, Adi Hanuka, J. Shtalenkova, Auralee Edelen, Panagiotis Baxevanis, Adam Egger, Tyler Cope, Mitchell McIntire, Stefano Ermon, Daniel Ratner
Abstract
The Linac coherent light source x-ray free-electron laser is a complex scientific apparatus which changes configurations multiple times per day, necessitating fast tuning strategies to reduce setup time for successive experiments. To this end, we employ a Bayesian approach to maximizing x-ray laser pulse energy by controlling groups of quadrupole magnets. A Gaussian process model provides probabilistic predictions for the machine response with respect to control parameters, enabling a balance of exploration and exploitation in the search for the global optimum. We show that the model parameters can be learned from archived scans, and correlations between devices can be extracted from the beam transport. The result is a sample-efficient optimization routine, combining both historical data and knowledge of accelerator physics to significantly outperform existing optimizers.