Efficient Dynamic and Momentum Aperture Optimization for Lattice Design Using Multipoint Bayesian Algorithm Execution

TL;DR

This paper introduces multipointBAX, a neural-network-based Bayesian optimization method that significantly reduces computational costs in storage ring design, enabling more efficient and higher-quality optimization for light sources and colliders.

Contribution

The paper presents multipointBAX, a novel multipoint Bayesian algorithm that drastically decreases simulation costs in storage ring optimization compared to traditional black-box methods.

Findings

Achieves Pareto optimal solutions with over 100x fewer simulations.

Demonstrates effectiveness on a fourth-generation light source design.

Positions multipointBAX as a scalable alternative for large-scale scientific facility design.

Abstract

We demonstrate that multipoint Bayesian algorithm execution can overcome fundamental computational challenges in storage ring design optimization. Dynamic (DA) and momentum (MA) optimization is a multipoint, multiobjective design task for storage rings, ultimately informing the flux of x-ray sources and luminosity of colliders. Current state-of-art black-box optimization methods require extensive particle-tracking simulations for each trial configuration; the high computational cost restricts the extent of the search to $\sim 10^3$ configurations, and therefore limits the quality of the final design. We remove this bottleneck using multipointBAX, which selects, simulates, and models each trial configuration at the single particle level. We demonstrate our approach on a novel design for a fourth-generation light source, with neural-network powered multipointBAX achieving equivalent Pareto front results using more than two orders of magnitude fewer tracking computations compared to genetic algorithms. The significant reduction in cost positions multipointBAX as a promising alternative to black-box optimization, and we anticipate multipointBAX will be instrumental in the design of future light sources, colliders, and large-scale scientific facilities.