Abstract:We train a neural network model to predict the full phase space evolution of cosmological N-body simulations. Its success implies that the neural network model is accurately approximating the Green's function expansion that relates the initial conditions of the simulations to its outcome at later times in the deeply nonlinear regime. We test the accuracy of this approximation by assessing its performance on well-understood simple cases that have either known exact solutions or well-understood expansions. These scenarios include spherical configurations, isolated plane waves, and two interacting plane waves: initial conditions that are very different from the Gaussian random fields used for training. We find our model generalizes well to these well-understood scenarios, demonstrating that the networks have inferred general physical principles and learned the nonlinear mode couplings from the complex, random Gaussian training data. These tests also provide a useful diagnostic for finding the model's strengths and weaknesses, and identifying strategies for model improvement. We also test the model on initial conditions that contain only transverse modes, a family of modes that differ not only in their phases but also in their evolution from the longitudinal growing modes used in the training set. When the network encounters these initial conditions that are orthogonal to the training set, the model fails completely. In addition to these simple configurations, we evaluate the model's predictions for the density, displacement, and momentum power spectra with standard initial conditions for N-body simulations. We compare these summary statistics against N-body results and an approximate, fast simulation method called COLA (COmoving Lagrangian Acceleration). Our model achieves percent level accuracy at nonlinear scales of k ∼ 1 Mpc - 1 h , representing a significant improvement over COLA.

Neural Infalling Cloud Equations (NICE): Increasing the Efficacy of Subgrid Models and Scientific Equation Discovery using Neural ODEs and Symbolic Regression

Neural ODEs as a discovery tool to characterize the structure of the hot galactic wind of M82

Capturing missing physics in climate model parameterizations using neural differential equations

Neural Astrophysical Wind Models

Neural Ideal Large Eddy Simulation: Modeling Turbulence with Neural Stochastic Differential Equations

Data-Driven Equation Discovery of a Cloud Cover Parameterization

CGNSDE: Conditional Gaussian Neural Stochastic Differential Equation for Modeling Complex Systems and Data Assimilation

Data‐Driven Equation Discovery of a Cloud Cover Parameterization

Enhancing Low-Order Discontinuous Galerkin Methods with Neural Ordinary Differential Equations for Compressible Navier--Stokes Equations

Bayesian Neural Ordinary Differential Equations

Learning Governing Equations of Unobserved States in Dynamical Systems

NeuralPDE: Modelling Dynamical Systems from Data

Accelerating Simulation of Two-Phase Flows with Neural PDE Surrogates

Neural Downscaling for Complex Systems: from Large-scale to Small-scale by Neural Operator

A High Resolution Physics-informed Neural Networks for High-Dimensional Convection-Diffusion-reaction Equations.

Data-driven multiscale modeling of subgrid parameterizations in climate models

Neural Ordinary Differential Equations for Data-Driven Reduced Order Modeling of Environmental Hydrodynamics

Incomplete to complete multiphysics forecasting -- a hybrid approach for learning unknown phenomena

An extended neural ordinary differential equation network with grey system and its applications

Simple lessons from complex learning: what a neural network model learns about cosmic structure formation