ICALEPCS 2025 - The 20th International Conference on Accelerator and Large Experimental Physics Control Systems

Bayesian active learning for converging posteriors in latent variable inference for control systems

WEPD075

24 Sept 2025, 16:30

1h 30m

Palmer House Hilton Chicago

Poster Presentation MC13: Artificial Intelligence & Machine Learning WEPD Posters

Speaker

Kilean Hwang (Facility for Rare Isotope Beams)

Description

Inferring latent variables, such as Courant-Snyder parameters in particle accelerators, is challenging due to noisy, partial observations that often produce multi-modal posterior distributions, despite the true latent variable being unique. We present a Bayesian Active Learning (BAL) framework to enhance latent variable inference in simulation-equipped control systems. BAL actively selects control settings (e.g., quadrupole magnet configurations) to maximize information gain, efficiently refining multi-modal posteriors into unimodal ones for improved inference accuracy. Using an ensemble of physics-informed beam envelope simulations in PyTorch, our approach approximates posterior sampling and mutual information to guide data acquisition. This interpretable framework holds broad potential for improving latent variable inference in control systems.

Funding Agency

Work supported by the U.S. Department of Energy, under Award Number DE-SC0024707 and used resources of FRIB Operations, a DOE Office of Science
User Facility under Award Number DE-SC0023633.