Abstract:A major aim of current and upcoming cosmological surveys is testing deviations from the standard $\Lambda$CDM model, but the full scientific value of these surveys will only be realised through efficient simulation methods that keep up with the increasing volume and precision of observational data. $N$-body simulations of modified gravity (MG) theories are computationally expensive since highly non-linear equations need to be solved to model the non-linear matter evolution; this represents a significant bottleneck in the path to reach the data volume and resolution attained by equivalent $\Lambda$CDM simulations. We develop a field-level, neural-network-based emulator that generates density and velocity divergence fields under the $f(R)$ gravity MG model from the corresponding $\Lambda$CDM simulated fields. Using attention mechanisms and a complementary frequency-based loss function, our model is able to learn this intricate mapping. We further use the idea of latent space extrapolation to generalise our emulator to $f(R)$ models with differing field strengths. The predictions of our emulator agree with the $f(R)$ simulations to within 5% for matter density and to within 10% for velocity divergence power spectra up to $k \sim 2\,h$ $\mathrm{Mpc}^{-1}$. But for a few select cases, higher-order statistics are reproduced with $\lesssim$10% agreement with the $f(R)$ simulations. Latent extrapolation allows our emulator to generalise to different $f(R)$ model variants without explicitly training on those variants. Given a $\Lambda$CDM simulation, the GPU-based emulator is able to reproduce the equivalent $f(R)$ realisation $\sim$600 times faster than full $N$-body simulations. This lays the foundations for a valuable tool for realistic yet rapid mock field generation and robust cosmological analyses.

Attention-based Neural Network Emulators for Multi-Probe Data Vectors Part I: Forecasting the Growth-Geometry split

Attention-Based Neural Network Emulators for Multi-Probe Data Vectors Part II: Assessing Tension Metrics

Neural Network Emulation of Spatio-temporal Data Using Linear and Nonlinear Dimensionality Reduction

Fast emulation of two-point angular statistics for photometric galaxy surveys

A field-level emulator for modified gravity

A neural network emulator for the Lyman-$α$ 1D flux power spectrum

High-accuracy emulators for observables in $Λ$CDM, $N_\mathrm{eff}$, $Σm_ν$, and $w$ cosmologies

Emulation of $f(R)$ modified gravity from $Λ$CDM using conditional GANs

Emulation of f(R) modified gravity from ΛCDM using conditional GANs

Cosmological multifield emulator

As Simple as Possible but No Simpler: Optimizing the Performance of Neural Net Emulators for Galaxy SED Fitting

High-accuracy emulators for observables in ΛCDM, N eff, Σ m ν, and w cosmologies

Field-level Emulation of Cosmic Structure Formation with Cosmology and Redshift Dependence

AI-assisted Reconstruction of Cosmic Velocity Field from Redshift-Space Spatial Distribution of Halos

Testing interacting dark energy with Stage IV cosmic shear surveys through differentiable neural emulators

On the road to percent accuracy II: calibration of the non-linear matter power spectrum for arbitrary cosmologies

Neural network based emulation of galaxy power spectrum covariances -- A reanalysis of BOSS DR12 data

Emulation of the Cosmic Dawn 21-cm Power Spectrum and Classification of Excess Radio Models Using an Artificial Neural Network

Deep Neural Emulation of the Supermassive Black-hole Binary Population

AI-assisted super-resolution cosmological simulations IV: An emulator for deterministic realizations

Learning to Predict the Cosmological Structure Formation.