Hostname: page-component-76d6cb85b7-2r2wp Total loading time: 0 Render date: 2026-07-21T09:08:28.579Z Has data issue: false hasContentIssue false

The machine learning approach to moment closure relations for plasma: a review

Published online by Cambridge University Press:  21 July 2026

Samuel Burles*
Affiliation:
School of Physical and Chemical Sciences, Queen Mary University of London, London E1 4NS, UK
Enrico Camporeale
Affiliation:
School of Physical and Chemical Sciences, Queen Mary University of London, London E1 4NS, UK Space Weather TREC, University of Colorado, Boulder, CO 80303, USA
*
Corresponding author: Samuel Burles, s.burles@qmul.ac.uk

Abstract

The requirement for large-scale global simulations of plasma is an ongoing challenge in both space and laboratory plasma physics. Any simulation based on a fluid model inherently requires a closure relation for the high-order plasma moments. This review compiles and analyses the recent surge of machine learning (ML) approaches developing improved plasma closure models capable of capturing kinetic phenomena within plasma fluid models. We survey two methodological families: neural network surrogates (from multi-layer perceptrons to Fourier neural operators, the latter recently reproducing both linear and nonlinear Landau damping online within a fluid solver) and equation discovery methods such as sparse regression; and organise the studies by whether they are tested offline against reference data or online within a time-evolving solver. We outline the challenges associated with ML closures, including off-diagonal pressure-tensor accuracy, generalisation beyond the training distribution and stable integration into large-scale simulations, and the directions future research might take to address them.

Information

Type
Review Article
Creative Commons
Creative Common License - CCCreative Common License - BY
This is an Open Access article, distributed under the terms of the Creative Commons Attribution licence (https://creativecommons.org/licenses/by/4.0/), which permits unrestricted re-use, distribution and reproduction, provided the original article is properly cited.
Copyright
© The Author(s), 2026. Published by Cambridge University Press
Figure 0

Figure 1. Spatial distribution of the ion inertial length di$d_i$ in a global magnetosphere simulation, illustrating the large variation in kinetic scales across the simulation domain. Regions where di$d_i$ is comparable to the local grid resolution require kinetic treatment, motivating the embedded kinetic-fluid simulation approach. Reproduced from Tóth et al. (2016b, figure 2).

Figure 1

Table 1. Summary of analytical plasma closure models discussed in this review, organised by closure class. `Closure class’ indicates the type of fluid model and the order at which the fluid hierarchy is truncated: 5-moment (scalar pressure), 10-moment (full pressure tensor) or higher-order closures such as the heat flux or the fourth-order moment.Table 1 long description.

Figure 2

Figure 2. Schematic of the SINDy algorithm. Time-series data of state variables are collected (left), a library of candidate nonlinear terms is constructed (centre), and sparsity-promoting regression identifies the minimal set of terms that best describe the dynamics (right), yielding an interpretable governing equation. Reproduced from Brunton et al. (2016, figure 1).

Figure 3

Figure 3. Schematic of the PDE-Net 2.0 architecture for data-driven PDE discovery. The network is constructed by stacking multiple δt$\delta t$ blocks into a feed-forward deep network, where all blocks share the same learnable parameters. Each δt$\delta t$ block takes the current state Ut$U_t$ as input and produces an updated state U~t+δt$\tilde {U}_{t+\delta t}$ by combining two components: (i) learnable convolution filters that approximate spatial differential operators (e.g. ∂x$\partial _x$, ∂xx$\partial _{xx}$), with constraints on their moment matrices to ensure correspondence with finite-difference stencils; and (ii) a symbolic neural network (SymNet) that approximates the nonlinear response function F(U,∇U,∇2U,…)$F(U, \boldsymbol{\nabla }U, \boldsymbol{\nabla} ^2 U, \ldots )$ of the underlying PDE. The forward Euler temporal discretisation makes the architecture ResNet-like, enabling long-term prediction by propagating the input through successive δt$\delta t$ blocks. During training, the accumulated prediction error ‖U(t+nδt)−U~(t+nδt)‖2$\|U(t + n\delta t) - \tilde {U}(t + n\delta t)\|^2$ is minimised jointly over all blocks, and sparsity-promoting regularisation on the symNet parameters allows irrelevant terms to be driven to zero, thereby recovering an interpretable analytic form of the governing PDE. Reproduced from Long et al. (2019, figure 2).

Figure 4

Figure 4. Overview of the PINN methodology for learning an implicit plasma closure from one-dimensional Landau-damping kinetic simulation data. (a) Spatiotemporal evolution of the kinetic simulation data, showing the density, bulk velocity, pressure, heat flux and electric field. (b) Subdomains from which the training data are sparsely sampled, restricted to early simulation times to test the model’s ability to extrapolate. (c) Architecture of the PINN: the network takes time and spatial coordinates as inputs and outputs all fluid variables simultaneously, with the loss function incorporating the fluid moment equations as physics-informed residual constraints alongside the data loss. The gPINNp variant additionally penalises the gradient of the pressure residual only, exploiting the expected sensitivity of the pressure evolution to the closure. (d) Model predictions for each fluid variable, demonstrating good agreement with the kinetic simulation even at late times beyond the training window. The closure is learned implicitly within the coupled system of equations rather than as a standalone module. Reproduced from Qin et al. (2023, figure 1).

Figure 5

Figure 5. Architecture of the FNO. Top: the full network pipeline from input function through lifting (P$P$), iterative Fourier layers and projection (Q$Q$) to the output function. Bottom: detail of a single Fourier layer, showing the parallel paths of the spectral convolution (F→Rℓ→F−1$\mathcal{F} \to R_\ell \to \mathcal{F}^{-1}$, where high-frequency modes above kmax$k_{\mathrm{max}}$ are truncated) and the local linear transform Wℓ$W_\ell$, combined through a pointwise nonlinearity σ$\sigma$. Reproduced from Kovachki et al. (2023, figure 5).

Figure 6

Figure 6. Accuracy–complexity Pareto front for closure models discovered via sparse regression (SINDy) applied to fully kinetic PIC simulation data. The horizontal axis gives the model complexity, measured by the number of non-zero terms retained in the sparse regression, while the vertical axis shows the fraction of variance unexplained (FVU). Orange circles denote training-set error and blue triangles denote test-set error, evaluated over a 30-fold cross-validation procedure. At each level of complexity, the spread of the markers reflects the variability of the discovered model across different cross-validation realisations; models whose terms are consistently identified across folds (low spread) are more robust than those exhibiting high variability between realisations. Annotations along the Pareto front identify the physical approximations corresponding to each level of complexity: from the simplest model (neglecting heat flux) through progressive inclusion of pressure anisotropy, off-diagonal pressure-tensor terms and Joule heating. The sharp knee at five non-zero terms marks the optimal model, where the Pareto front breaks clearly and the discovered terms are consistent across cross-validation folds; beyond this point, additional terms yield diminishing improvements in accuracy and the model form becomes less stable. This hierarchy demonstrates how sparse regression can recover and rank well-known closure approximations directly from kinetic data, providing a data-driven framework for guiding the development of new closure models. Reproduced from Alves & Fiuza (2022).

Figure 7

Table 2. Summary of ML plasma closure studies discussed in § 3. `Online’ testing denotes embedding the trained model inside a running fluid solver; offline’ denotes validation against a kinetic or analytic reference dataset without time stepping.Table 2 long description.