cDVAE: VAE-guided diffusion for particle accelerator beam 6D phase space projection diagnostics

Imaging the 6D phase space of a beam in a particle accelerator in a single shot is currently impossible. Single shot beam measurements only exist for certain 2D beam projections and these methods are destructive. A virtual diagnostic that can generate an accurate prediction of a beam's 6D phase space would be incredibly useful for precisely controlling the beam. In this work, a generative conditional diffusion- based approach to creating a virtual diagnostic of all 15 unique 2D projections of a beam's 6D phase space is developed. The diffusion process is guided by a combination of scalar parameters and images that are converted to low-dimensional latent vector representation by a variational autoencoder (VAE). We demonstrate that conditional diffusion guided by a VAE (cDVAE) can accurately reconstruct all 15 of the unique 2D projections of a charged particle beam's 6D phase space for the HiRES compact accelerator.