Abstract:Theoretical uncertainty limits our ability to extract cosmological information from baryonic fields such as the thermal Sunyaev-Zel'dovich (tSZ) effect. Being sourced by the electron pressure field, the tSZ effect depends on baryonic physics that is usually modeled by expensive hydrodynamic simulations. We train neural networks on the IllustrisTNG-300 cosmological simulation to predict the continuous electron pressure field in galaxy clusters from gravity-only simulations. Modeling clusters is challenging for neural networks as most of the gas pressure is concentrated in a handful of voxels and even the largest hydrodynamical simulations contain only a few hundred clusters that can be used for training. Instead of conventional convolutional neural net (CNN) architectures, we choose to employ a rotationally equivariant DeepSets architecture to operate directly on the set of dark matter particles. We argue that set-based architectures provide distinct advantages over CNNs. For example, we can enforce exact rotational and permutation equivariance, incorporate existing knowledge on the tSZ field, and work with sparse fields as are standard in cosmology. We compose our architecture with separate, physically meaningful modules, making it amenable to interpretation. For example, we can separately study the influence of local and cluster-scale environment, determine that cluster triaxiality has negligible impact, and train a module that corrects for mis-centering. Our model improves by 70 % on analytic profiles fit to the same simulation data. We argue that the electron pressure field, viewed as a function of a gravity-only simulation, has inherent stochasticity, and model this property through a conditional-VAE extension to the network. This modification yields further improvement by 7 %, it is limited by our small training set however. (abridged)

Predicting the turbulent transport of cosmic rays via neural networks

Predicting the turbulent transport of cosmic rays via neural networks

Neural networks for turbulent transport prediction in a simplified model of tokamak plasmas

Prediction and Anomaly Detection of accelerated particles in PIC simulations using neural networks

Deep Learning Cosmic Ray Transport from Density Maps of Simulated, Turbulent Gas

Cosmic Velocity Field Reconstruction Using AI

Application of a Neural Network to Store and Compute the Optical Properties of Non-Spherical Particles

Predicting atmospheric optical properties for radiative transfer computations using neural networks

deepCool: Fast and Accurate Estimation of Cooling Rates in Irradiated Gas with Artificial Neural Networks

Advancements in neural network techniques for electric and magnetic field reconstruction: Application to ion radiography

Artificial neural networks for cosmic gamma-ray propagation in the universe

Artificial Neural Network in Cosmic Landscape

Prediction of turbulent heat transfer using convolutional neural networks

Neural network emulation of flow in heavy-ion collisions at intermediate energies

Toward using GANs in astrophysical Monte-Carlo simulations

Predicting the Thermal Sunyaev-Zel'dovich Field using Modular and Equivariant Set-Based Neural Networks

Fast Inference Using Automatic Differentiation and Neural Transport in Astroparticle Physics

Alternative Artificial Neural Network Structures for Turbulent Flow Velocity Field Prediction

Neural Network predictions of inclusive electron-nucleus cross sections

Predictions of turbulent shear flows using deep neural networks