Mar 5 – 8, 2024
Lahan Select Gyeongju, South Korea
Asia/Seoul timezone

Bayesian Optimization with Neural Network Prior Mean Models

Mar 6, 2024, 9:20 AM
20m
Lahan Select Gyeongju, South Korea

Lahan Select Gyeongju, South Korea

Lahan Select Gyeongju, South Korea
Oral (16mins + 4 mins) Optimization & Control Optimization & Control

Speaker

Tobias Boltz (SLAC National Accelerator Laboratory)

Description

Bayesian optimization using Gaussian processes is a powerful tool to automate complex and time-consuming accelerator tuning tasks and has been demonstrated to outperform conventional methods at several facilities. In high-dimensional input spaces, however, even this sample efficient search may take a prohibitively large number of steps to reach convergence. In this contribution, we discuss the use of neural networks as a prior mean to inform the surrogate GP model and thereby speed-up convergence. We present collaborative results obtained in simulations and experiments at the Linac Coherent Light Source (LCLS) and the Argonne Tandem Linear Accelerator System (ATLAS). We show that high quality models can significantly improve optimization performance and discuss further measures to recover performance in cases where only models of limited accuracy are available.

Primary Keyword bayesian optimization

Primary author

Tobias Boltz (SLAC National Accelerator Laboratory)

Co-authors

Auralee Edelen (SLAC National Accelerator Laboratory) Brahim Mustapha (Argonne National Laboratory) Daniel Ratner (SLAC National Accelerator Laboratory) Jose Martinez-Marin (Argonne National Laboratory) Ryan Roussel (SLAC National Accelerator Laboratory)

Presentation materials