Encoder-decoder neural network for solving the nonlinear Fokker-Planck-Landau collision operator in XGC
An encoder-decoder neural network has been used to accelerate the numerical solving of a partial integro-differential equation, the Fokker-Planck-Landau collision operator. This is part of the governing equation in the massively parallel particle-in-cell code, XGC, which is used to study turbulence in fusion energy devices. The neural network emphasizes physics-inspired learning, where it is taught to respect physical conservation constraints of the collision operator by including them in the training loss, along with the L2 loss. The run time for the current Picard iterative solver of the operator is $O(n^2)$, where n is the number of plasma species. As the XGC1 code begins to attack problems including a larger number of species, the collision operator will become expensive computationally, making the neural network solver even more important, especially since the training only scales as $O(n)$. A wide enough range of collisionality has been considered in the training data to ensure the full domain of collision physics is captured. Eventual work will include expansion of the network to include multiple plasma species.
Code (0)
등록된 구현이 없습니다.
Tasks
DecoderSimilar Papers 제목 키워드 기반
Fokker-Planck to Callan-Symanzik: evolution of weight matrices under training
The dynamical evolution of a neural network during training has been an incredibly fascinating subject of study. First principal derivation of generic evolution of variables in statistical physics systems has proved usef…
Invasion patterns in competitive systems
Stochastic reaction-diffusion equations are a popular modelling approach for studying interacting populations in a heterogeneous environment under the influence of environmental fluctuations. Although the theoretical bas…
Large-Scale Wasserstein Gradient Flows
Wasserstein gradient flows provide a powerful means of understanding and solving many diffusion equations. Specifically, Fokker-Planck equations, which model the diffusion of probability measures, can be understood as gr…
Solving time dependent Fokker-Planck equations via temporal normalizing flow
In this work, we propose an adaptive learning approach based on temporal normalizing flows for solving time-dependent Fokker-Planck (TFP) equations. It is well known that solutions of such equations are probability densi…
A convergent scheme for the Bayesian filtering problem based on the Fokker--Planck equation and deep splitting
A numerical scheme for approximating the nonlinear filtering density is introduced and its convergence rate is established, theoretically under a parabolic H\"ormander condition, and empirically in two numerical examples…