Journal Article

AI-assisted superresolution cosmological simulations

Croft, Rupert A. C., Di Matteo, Tiziana, Ni, Yueying, Bird, Simeon, Feng, Yu, Li, Yin

2021

Overview

Cosmological simulations of galaxy formation are limited by finite computational resources. We draw from the ongoing rapid advances in artificial intelligence (AI; specifically deep learning) to address this problem. Neural networks have been developed to learn from high-resolution (HR) image data and then make accurate superresolution (SR) versions of different low-resolution (LR) images. We apply such techniques to LR cosmological N-body simulations, generating SR versions. Specifically, we are able to enhance the simulation resolution by generating 512 times more particles and predicting their displacements from the initial positions. Therefore, our results can be viewed as simulation realizations themselves, rather than projections, e.g., to their density fields. Furthermore, the generation process is stochastic, enabling us to sample the small-scale modes conditioning on the large-scale environment. Our model learns from only 16 pairs of small-volume LR-HR simulations and is then able to generate SR simulations that successfully reproduce the HR matter power spectrum to percent level up to 16 h −1Mpc and the HR halo mass function to within 10% down to 1011 M ☉. We successfully deploy the model in a box 1,000 times larger than the training simulation box, showing that high-resolution mock surveys can be generated rapidly. We conclude that AI assistance has the potential to revolutionize modeling of small-scale galaxy-formation physics in large cosmological volumes.

• Publisher: National Academy of Sciences
• Subject: Artificial intelligence / Astronomical models / Astronomy / Computer applications / Deep learning / Galactic evolution / High resolution / Image resolution / Neural networks / Physical Sciences / Simulation / Star & galaxy formation / Stochastic processes