Abstract:One of the open problems in scientific computing is the long-time integration of nonlinear stochastic partial differential equations (SPDEs), especially with arbitrary initial data. We address this problem by taking advantage of recent advances in scientific machine learning and the spectral dynamically orthogonal (DO) and borthogonal (BO) methods for representing stochastic processes. The recently introduced DO/BO methods reduce the SPDE to solving a system of deterministic PDEs and a system of stochastic ordinary differential equations. Specifically, we propose two new physics-informed neural networks (PINNs) for solving time-dependent SPDEs, namely the neural network (NN)-DO/BO methods. The proposed methods incorporate the DO/BO constraints into the loss function (along with the modal decomposition of the SPDE) with an implicit form instead of generating explicit expressions for the temporal derivatives of the DO/BO modes. Hence, the NN-DO/BO methods can overcome some of the drawbacks of the original DO/BO methods. For example, we do not need the assumption that the covariance matrix of the random coefficients is invertible as in the original DO method, and we can remove the assumption of no eigenvalue crossing as in the original BO method. Moreover, the NN-DO/BO methods can be used to solve time-dependent stochastic inverse problems with the same formulation and same computational complexity as for forward problems. We demonstrate the capability of the proposed methods via several numerical examples, namely: (1) A linear stochastic advection equation with deterministic initial condition: we obtain good results with the proposed methods, while the original DO/BO methods cannot be applied directly in this case. (2) Long-time integration of the stochastic Burgers' equation: we show the good performance of NN-DO/BO methods, especially the effectiveness of the NN-BO approach for such problems with many eigenvalue crossings during the whole time evolution, while the original BO method fails. (3) Nonlinear reaction diffusion equation: we consider both the forward problem and the inverse problems, including very noisy initial point values, to investigate the flexibility of the NN-DO/BO methods in handling inverse and mixed type problems. Taken together, these simulation results demonstrate that the NN-DO/BO methods can be employed to effectively quantify uncertainty propagation in a wide range of physical problems, but future work should address the efficiency issue of PINNs for forward problems.

Numerical Methods for Solving Moment Equations in Kinetic Theory of Neuronal Network Dynamics

Numerical Methods for Solving Kinetic Equations of Neuronal Network Dynamics

KINETIC THEORY FOR NEURONAL NETWORK DYNAMICS

Moment neural network and an efficient numerical method for modeling irregular spiking activity

Mean-Field and Kinetic Descriptions of Neural Differential Equations

Dynamics of Moment Neuronal Networks

Learning Invariance Preserving Moment Closure Model for Boltzmann-BGK Equation

A spectral method for a Fokker-Planck equation in neuroscience with applications in neural networks with learning rules

An Effective Kinetic Representation of Fluctuation-Driven Neuronal Networks with Application to Simple and Complex Cells in Visual Cortex

Learning fast, accurate, and stable closures of a kinetic theory of an active fluid

Theory of moment propagation for quantum dynamics in single-particle description

A Stable and Scalable Method for Solving Initial Value PDEs with Neural Networks

Direct Flux Gradient Approximation to Moment Closure of Kinetic Equations

Classical Solutions for a Nonlinear Fokker-Planck Equation Arising in Computational Neuroscience

Deep learning approximations for non-local nonlinear PDEs with Neumann boundary conditions

Numerical analysis of coupled dynamical biological networks: Modeling electrical information exchange among nerve cells using finite volume method

Kinetic and macroscopic equations for action potential in neural networks

Vanishing moment method and moment solutions for second order fully nonlinear partial differential equations

Learning in Modal Space: Solving Time-Dependent Stochastic PDEs Using Physics-Informed Neural Networks