On learning latent dynamics of the AUG plasma state

In this work, we demonstrate the utility of state representation learning applied to modeling the time evolution of electron density and temperature profiles at ASDEX-Upgrade (AUG). The proposed model is a deep neural network, which learns to map the high dimensional profile observations to a lower dimensional state. The mapped states, alongside the original profile's corresponding machine parameters, are used to learn a forward model to propagate the state in time. We show that this approach is able to predict AUG discharges using only a selected set of machine parameters. The state is then further conditioned to encode information about the confinement regime, which yields a simple baseline linear classifier, while still retaining the information needed to predict the evolution of profiles. We, then, discuss the potential use cases and limitations of state representation learning algorithms applied to fusion devices.