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

Autor: Jamieson D; Max-Planck-Institut für Astrophysik, Karl-Schwarzschild-Straße 1, 85748 Garching, Germany.; Department of Physics and Astronomy, Stony Brook University, Stony Brook, NY 11794-3800, USA., Li Y; Center for Computational Astrophysics, Flatiron Institute, 162 5th Avenue, New York, NY 10010, USA.; Center for Computational Mathematics, Flatiron Institute, 162 5th Avenue, New York, NY 10010, USA., He S; Center for Computational Astrophysics, Flatiron Institute, 162 5th Avenue, New York, NY 10010, USA.; Department of Physics, Carnegie Mellon University, Pittsburgh, PA 15213, USA., Villaescusa-Navarro F; Center for Computational Astrophysics, Flatiron Institute, 162 5th Avenue, New York, NY 10010, USA.; Department of Astrophysical Sciences, Princeton University, 4 Ivy Lane, Princeton, NJ 08544, USA., Ho S; Center for Computational Astrophysics, Flatiron Institute, 162 5th Avenue, New York, NY 10010, USA.; Department of Astrophysical Sciences, Princeton University, 4 Ivy Lane, Princeton, NJ 08544, USA.; Department of Physics and Center for Data Science, New York University, 60 5th Ave, New York, NY 10011, USA.; Department of Physics, Carnegie Mellon University, 10 40th St, Pittsburgh, PA 15201, USA., de Oliveira RA; Center for Computational Astrophysics, Flatiron Institute, 162 5th Avenue, New York, NY 10010, USA.; Centro de Ciências Exatas, Universidade Federal do Espírito Santo, Av. Fernando Ferrari, 514, Goiabeiras, Vitória 29075-910, ES, Brazil., Spergel DN; Simons Foundation, 160 5th Avenue, New York, NY 10010, USA.
Jazyk: angličtina
Zdroj: PNAS nexus [PNAS Nexus] 2022 Nov 09; Vol. 2 (4), pp. pgac250. Date of Electronic Publication: 2022 Nov 09 (Print Publication: 2023).
DOI: 10.1093/pnasnexus/pgac250
Abstrakt: 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.
(© The Author(s) 2022. Published by Oxford University Press on behalf of National Academy of Sciences.)
Databáze: MEDLINE