Machine learning-based prediction of magnet errors in storage ring light sources
Jianhao Xu
TL;DR
This work introduces a machine-learning-based inverse-mapping framework that predicts magnet errors directly from BPM-measured optics and COD in a four-bend achromat storage-ring lattice, bypassing traditional LOCO-based inversion. By generating a large, labeled dataset with ELEGANT simulations and evaluating LR, SVM, RBFNN, and DenseNet, the study demonstrates that RBFNN and related nonlinear models can accurately recover quadrupole and sextupole errors and reconstruct ideal optics in many cases, enabling faster commissioning and online diagnostics. Correlation analyses reveal nonlinear dependencies between errors and optics, guiding feature selection and model choice, while compensation experiments show substantial reductions in beta-beating and dispersion variations, validating the practical potential of the approach for next-generation diffraction-limited rings. The work also discusses limitations and future directions, including broader error models and scalable architectures for larger facilities.
Abstract
Magnet errors in storage rings significantly degrade beam performance, impacting the brightness and stability of the light source. Therefore, beam-based correction is crucial for the safe operation of machines and the stability of radiated photons. Unlike traditional correction methods such as linear optics from closed orbit, this paper proposes a machine learning (ML) framework to directly predict quadrupole/sextupole gradient errors and misalignment from beam position monitor-measured optics functions and closed-orbit distortion data. Based on a four-bend achromat storage ring lattice, we generate training datasets through ELEGANT numerical simulations and compare regression performance of Linear Regression, Support Vector Machine, Radial Basis Function Neural Network and Densely Connected Convolutional Network. Results demonstrate that ML models can effectively predict magnet errors and reconstruct ideal optics. This approach offers a novel strategy for accelerating storage ring commissioning and optimization, online diagnostics, and dynamic compensation for next-generation diffraction-limited rings.
