Leveraging Prior Mean Models for Faster Bayesian Optimization of Particle Accelerators

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Boltz, Tobias, Martinez, Jose L., Xu, Connie, Baker, Kathryn R. L., Zhu, Zihan, Morgan, Jenny, Roussel, Ryan, Ratner, Daniel, Mustapha, Brahim, Edelen, Auralee L.
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916618536222720
author Boltz, Tobias
Martinez, Jose L.
Xu, Connie
Baker, Kathryn R. L.
Zhu, Zihan
Morgan, Jenny
Roussel, Ryan
Ratner, Daniel
Mustapha, Brahim
Edelen, Auralee L.
author_facet Boltz, Tobias
Martinez, Jose L.
Xu, Connie
Baker, Kathryn R. L.
Zhu, Zihan
Morgan, Jenny
Roussel, Ryan
Ratner, Daniel
Mustapha, Brahim
Edelen, Auralee L.
contents Tuning particle accelerators is a challenging and time-consuming task that can be automated and carried out efficiently using suitable optimization algorithms, such as model-based Bayesian optimization techniques. One of the major advantages of Bayesian algorithms is the ability to incorporate prior information about beam physics and historical behavior into the model used to make control decisions. In this work, we examine incorporating prior accelerator physics information into Bayesian optimization algorithms by utilizing fast executing, neural network models trained on simulated or historical datasets as prior mean functions in Gaussian process models. We show that in ideal cases, this technique substantially increases convergence speed to optimal solutions in high-dimensional tuning parameter spaces. Additionally, we demonstrate that even in non-ideal cases, where prior models of beam dynamics do not exactly match experimental conditions, the use of this technique can still enhance convergence speed. Finally, we demonstrate how these methods can be used to improve optimization in practical applications, such as transferring information gained from beam dynamics simulations to online control of the LCLS injector, and transferring knowledge gained from experimental measurements across different operating modes, such as accelerating different ion species at the ATLAS heavy ion accelerator.
format Preprint
id arxiv_https___arxiv_org_abs_2403_03225
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Leveraging Prior Mean Models for Faster Bayesian Optimization of Particle Accelerators
Boltz, Tobias
Martinez, Jose L.
Xu, Connie
Baker, Kathryn R. L.
Zhu, Zihan
Morgan, Jenny
Roussel, Ryan
Ratner, Daniel
Mustapha, Brahim
Edelen, Auralee L.
Accelerator Physics
Tuning particle accelerators is a challenging and time-consuming task that can be automated and carried out efficiently using suitable optimization algorithms, such as model-based Bayesian optimization techniques. One of the major advantages of Bayesian algorithms is the ability to incorporate prior information about beam physics and historical behavior into the model used to make control decisions. In this work, we examine incorporating prior accelerator physics information into Bayesian optimization algorithms by utilizing fast executing, neural network models trained on simulated or historical datasets as prior mean functions in Gaussian process models. We show that in ideal cases, this technique substantially increases convergence speed to optimal solutions in high-dimensional tuning parameter spaces. Additionally, we demonstrate that even in non-ideal cases, where prior models of beam dynamics do not exactly match experimental conditions, the use of this technique can still enhance convergence speed. Finally, we demonstrate how these methods can be used to improve optimization in practical applications, such as transferring information gained from beam dynamics simulations to online control of the LCLS injector, and transferring knowledge gained from experimental measurements across different operating modes, such as accelerating different ion species at the ATLAS heavy ion accelerator.
title Leveraging Prior Mean Models for Faster Bayesian Optimization of Particle Accelerators
topic Accelerator Physics
url https://arxiv.org/abs/2403.03225