arXiv Machine Learning

A Pre-trained Variational Autoencoder for Gyrokinetic Plasma Turbulence Surrogate Modeling

arXiv AI
Sep 7

GyroSwin: 5D Surrogates for Gyrokinetic Plasma Turbulence Simulations

GyroSwin is a scalable 5‑dimensional neural surrogate that models nonlinear gyrokinetic plasma turbulence, a key challenge for nuclear fusion research. It extends Vision Transformers to 5D, incorporates cross‑attention and latent 3D↔5D interactions, and uses channelwise mode separation inspired by nonlinear physics. The model outperforms traditional reduced numerics in heat‑flux prediction, captures turbulent energy cascades, and cuts the computational cost of full gyrokinetic simulations by three orders of magnitude while remaining physically verifiable.

By Fabian Paischer, Gianluca Galletti, William Hornsby, Paul Setinek, Lorenzo Zanisi, Naomi Carey, Stanislas Pamela, Johannes Brandstetter
arXiv Machine Learning
Jul 15

A Shortcut to Statistically Steady-State Turbulence with Flow Matching

arXiv:2607. 13022v1 Announce Type: cross Abstract: Many nonlinear physical systems exhibit an initial transient phase in which perturbations grow before nonlinear interactions lead to a statistically steady state.

By Gianluca Galletti, Gerald Gutenbrunner, William Hornsby, Lorenzo Zanisi, Naomi Carey, Stanislas Pamela, Johannes Brandstetter, Fabian Paischer
arXiv Machine Learning
Sep 16

Stable by Construction: Variational Latent Markov Operators for Long-Horizon PDE Prediction

The paper introduces a variational framework called VAMO that incorporates latent Markov dynamics for neural PDE solvers, aiming to improve long‑horizon predictions by mitigating error accumulation. By representing physical states as latent distributions and evolving them through probabilistic transitions, the method aligns learned dynamics with a spectral geometry induced by structured Gaussian perturbations. Experiments on fluid‑dynamics benchmarks show that VAMO reduces error growth and enhances rollout stability compared to deterministic and noise‑injection baselines.

By Junyi Liao, Johann Guilleminot, Vahid Tarokh