Search papers, labs, and topics across Lattice.
The paper introduces a Latent Evolution Model (LEM), a hybrid machine learning framework combining autoencoders and transformers, to address multiple challenges in accelerator virtual beam diagnostics. The LEM uses a Conditional Variational Autoencoder (CVAE) to project high-dimensional phase spaces into lower-dimensional representations, and a transformer to learn temporal dynamics in the latent space. The framework tackles forward modeling, inverse problems (predicting upstream phase spaces and RF settings), and tuning problems (optimizing RF settings via Bayesian optimization to minimize beam loss).
A single Latent Evolution Model can solve the trifecta of forward modeling, inverse problems, and tuning for accelerator beam diagnostics, offering a unified approach to traditionally separate challenges.
Virtual beam diagnostics relies on computationally intensive beam dynamics simulations where high-dimensional charged particle beams evolve through the accelerator. We propose Latent Evolution Model (LEM), a hybrid machine learning framework with an autoencoder that projects high-dimensional phase spaces into lower-dimensional representations, coupled with transformers to learn temporal dynamics in the latent space. This approach provides a common foundational framework addressing multiple interconnected challenges in beam diagnostics. For \textit{forward modeling}, a Conditional Variational Autoencoder (CVAE) encodes 15 unique projections of the 6D phase space into a latent representation, while a transformer predicts downstream latent states from upstream inputs. For \textit{inverse problems}, we address two distinct challenges: (a) predicting upstream phase spaces from downstream observations by utilizing the same CVAE architecture with transformers trained on reversed temporal sequences along with aleatoric uncertainty quantification, and (b) estimating RF settings from the latent space of the trained LEM using a dedicated dense neural network that maps latent representations to RF parameters. For \textit{tuning problems}, we leverage the trained LEM and RF estimator within a Bayesian optimization framework to determine optimal RF settings that minimize beam loss. This paper summarizes our recent efforts and demonstrates how this unified approach effectively addresses these traditionally separate challenges.