GyroSwin: 5D Surrogates for Gyrokinetic Plasma Turbulence Simulations

• URL: http://arxiv.org/abs/2510.07314v2
• Date: Fri, 10 Oct 2025 11:45:38 GMT
• Title: GyroSwin: 5D Surrogates for Gyrokinetic Plasma Turbulence Simulations
• Abstract summary: Plasma turbulence is governed by the nonlinear gyrokinetic equation.<n> reduced-order models are often employed in practice to approximate turbulent transport of energy.<n>We introduce GyroSwin, the first scalable 5D neural surrogate that can model 5D nonlinear gyrokinetic simulations.
• Score: 18.320829792574738
• License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
• Abstract: Nuclear fusion plays a pivotal role in the quest for reliable and sustainable energy production. A major roadblock to viable fusion power is understanding plasma turbulence, which significantly impairs plasma confinement, and is vital for next-generation reactor design. Plasma turbulence is governed by the nonlinear gyrokinetic equation, which evolves a 5D distribution function over time. Due to its high computational cost, reduced-order models are often employed in practice to approximate turbulent transport of energy. However, they omit nonlinear effects unique to the full 5D dynamics. To tackle this, we introduce GyroSwin, the first scalable 5D neural surrogate that can model 5D nonlinear gyrokinetic simulations, thereby capturing the physical phenomena neglected by reduced models, while providing accurate estimates of turbulent heat transport. GyroSwin (i) extends hierarchical Vision Transformers to 5D, (ii) introduces cross-attention and integration modules for latent 3D$\leftrightarrow$5D interactions between electrostatic potential fields and the distribution function, and (iii) performs channelwise mode separation inspired by nonlinear physics. We demonstrate that GyroSwin outperforms widely used reduced numerics on heat flux prediction, captures the turbulent energy cascade, and reduces the cost of fully resolved nonlinear gyrokinetics by three orders of magnitude while remaining physically verifiable. GyroSwin shows promising scaling laws, tested up to one billion parameters, paving the way for scalable neural surrogates for gyrokinetic simulations of plasma turbulence.

Related papers

• A Shortcut to Statistically Steady-State Turbulence with Flow Matching [18.320829792574738]
  We introduce GyroFlow, a latent generative model that directly estimates steady-state statistics of gyrokinetic turbulence in 5D phase space.<n>To evaluate generation quality we propose FGyD, a distributional metric computed in the latent space of a pretrained gyrokinetic model.
  arXiv  Detail & Related papers  (2026-07-14T17:58:13Z)
• EFF-Grasp: Energy-Field Flow Matching for Physics-Aware Dexterous Grasp Generation [54.09734511705173]
  EFF-Grasp is a Flow-Matching synthesis-based framework for physics-aware dexterous grasp generation.<n>We reformulate grasp as a deterministic ordinary differential equation (ODE) process, which enables efficient and stable generation through smooth probability flows.<n> EFF-Grasp achieves superior performance in grasp quality and physical feasibility, while requiring substantially fewer sampling steps than diffusion-based baselines.
  arXiv  Detail & Related papers  (2026-03-17T06:11:56Z)
• Physics Informed Neural Networks for design optimisation of diamond particle detectors for charged particle fast-tracking at high luminosity hadron colliders [70.66815108184498]
  Future high-luminosity hadron colliders demand tracking detectors with extreme radiation tolerance, high spatial precision, and sub-nanosecond timing.<n>3D diamond pixel sensors offer these capabilities due to diamond's radiation hardness and high carrier mobility.<n>We model the phenomenon through a 3rd-order, 3+1D PDE derived as a quasi-stationary approximation of Maxwell's equations.
  arXiv  Detail & Related papers  (2025-09-25T13:09:28Z)
• Generative Pre-trained Autoregressive Diffusion Transformer [74.25668109048418]
  GPDiT is a Generative Pre-trained Autoregressive Diffusion Transformer.<n>It unifies the strengths of diffusion and autoregressive modeling for long-range video synthesis.<n>It autoregressively predicts future latent frames using a diffusion loss, enabling natural modeling of motion dynamics.
  arXiv  Detail & Related papers  (2025-05-12T08:32:39Z)
• Particle-based plasma simulation using a graph neural network [0.0]
  The model achieves high accuracy with a time step longer than conventional simulation by two orders of magnitude.<n>This work demonstrates that complex plasma dynamics can be learned and shows promise for the development of fast differentiable simulators.
  arXiv  Detail & Related papers  (2025-03-01T01:07:37Z)
• 5D Neural Surrogates for Nonlinear Gyrokinetic Simulations of Plasma Turbulence [9.788870512075171]
  We propose a method for training neural surrogates for 5D gyrokinetic simulations.<n>Our method extends a hierarchical vision transformer to five dimensions and is trained on the 5D distribution function for the adiabatic electron approximation.<n>Our work paves the way towards neural surrogates for plasma turbulence simulations to accelerate deployment of commercial energy production via nuclear fusion.
  arXiv  Detail & Related papers  (2025-02-11T11:25:10Z)
• Geometric Trajectory Diffusion Models [58.853975433383326]
  Generative models have shown great promise in generating 3D geometric systems. Existing approaches only operate on static structures, neglecting the fact that physical systems are always dynamic in nature. We propose geometric trajectory diffusion models (GeoTDM), the first diffusion model for modeling the temporal distribution of 3D geometric trajectories.
  arXiv  Detail & Related papers  (2024-10-16T20:36:41Z)
• Unfolding Time: Generative Modeling for Turbulent Flows in 4D [49.843505326598596]
  This work introduces a 4D generative diffusion model and a physics-informed guidance technique that enables the generation of realistic sequences of flow states. Our findings indicate that the proposed method can successfully sample entire subsequences from the turbulent manifold. This advancement opens doors for the application of generative modeling in analyzing the temporal evolution of turbulent flows.
  arXiv  Detail & Related papers  (2024-06-17T10:21:01Z)
• Plasma Surrogate Modelling using Fourier Neural Operators [57.52074029826172]
  Predicting plasma evolution within a Tokamak reactor is crucial to realizing the goal of sustainable fusion. We demonstrate accurate predictions of evolution plasma using deep learning-based surrogate modelling tools, viz., Neural Operators (FNO) We show that FNO has a speedup of six orders of magnitude over traditional solvers in predicting the plasma dynamics simulated from magnetohydrodynamic models. FNOs can also predict plasma evolution on real-world experimental data observed by the cameras positioned within the MAST Tokamak.
  arXiv  Detail & Related papers  (2023-11-10T10:05:00Z)
• Fast Dynamic 1D Simulation of Divertor Plasmas with Neural PDE Surrogates [3.6443770850509423]
  Managing divertor plasmas is crucial for operating reactor scale tokamak devices due to heat and particle flux constraints on the divertor target. We address this lack of fast simulators using neural PDE surrogates, data-driven neural network-based surrogate models trained using solutions generated with a classical numerical method. We simulate a realistic TCV divertor plasma with dynamics induced by upstream density ramps and provide an exploratory outlook towards fast transients.
  arXiv  Detail & Related papers  (2023-05-30T11:20:14Z)
• Unsupervised Discovery of Inertial-Fusion Plasma Physics using Differentiable Kinetic Simulations and a Maximum Entropy Loss Function [77.34726150561087]
  We create a differentiable solver for the plasma kinetics 3D partial-differential-equation and introduce a domain-specific objective function. We apply this framework to an inertial-fusion relevant configuration and find that the optimization process exploits a novel physical effect.
  arXiv  Detail & Related papers  (2022-06-03T15:27:33Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.