Effective cosmic density field reconstruction with convolutional neural network

Abstract

We present a cosmic density field reconstruction method that augments the traditional reconstruction algorithms with a convolutional neural network (CNN). Following Shallue $\&$ Eisenstein (2022), the key component of our method is to use the $\text{reconstructed}$ density field as the input to the neural network. We extend this previous work by exploring how the performance of these reconstruction ideas depends on the input reconstruction algorithm, the reconstruction parameters, and the shot noise of the density field, as well as the robustness of the method. We build an eight-layer CNN and train the network with reconstructed density fields computed from the Quijote suite of simulations. The reconstructed density fields are generated by both the standard algorithm and a new iterative algorithm. In real space at $z=0$, we find that the reconstructed field is $90\%$ correlated with the true initial density out to $k\sim 0.5h{\mathrm{Mpc}}^{-1}$, a significant improvement over $k\sim 0.2h{\mathrm{Mpc}}^{-1}$ achieved by the input reconstruction algorithms. We find similar improvements in redshift space, including an improved removal of redshift space distortions at small scales. We also find that the method is robust across changes in cosmology. Additionally, the CNN removes much of the variance from the choice of different reconstruction algorithms and reconstruction parameters. However, the effectiveness decreases with increasing shot noise, suggesting that such an approach is best suited to high density samples. This work highlights the additional information in the density field beyond linear scales as well as the power of complementing traditional analysis approaches with machine learning techniques.

10 pages, 8 figures. Matches version accepted for publication in MNRAS

References in corpus (9)

• Improving Cosmological Distance Measurements by Reconstruction of the Baryon Acoustic Peak
• Nonlinear Evolution of Baryon Acoustic Oscillations
• On the Robustness of the Acoustic Scale in the Low-Redshift Clustering of Matter
• The clustering of galaxies in the completed SDSS-III Baryon Oscillation Spectroscopic Survey: Observational systematics and baryon acoustic oscillations in the correlation function
• The clustering of galaxies in the completed SDSS-III Baryon Oscillation Spectroscopic Survey: Baryon Acoustic Oscillations in Fourier-space
• ELUCID - Exploring the Local Universe with reConstructed Initial Density field I: Hamiltonian Markov Chain Monte Carlo Method with Particle Mesh Dynamics
• Calibrating the Baryon Oscillation Ruler for Matter and Halos
• Reconstruction within the Zeldovich approximation
• A Hybrid Deep Learning Approach to Cosmological Constraints From Galaxy Redshift Surveys

Cited by in corpus (11)

• Neural network reconstruction of density and velocity fields from the 2MASS Redshift Survey
• DarkAI: Reconstructing the density, velocity and tidal field of dark matter from a DESI-like bright galaxy sample
• Learning the Universe: Learning to Optimize Cosmic Initial Conditions with Non-Differentiable Structure Formation Models
• YOLO-CL cluster detection in the Rubin/LSST DC2 simulation
• Neural Network Reconstruction of Non-Gaussian Initial Conditions from Dark Matter Halos
• Baryon Acoustic Oscillations analyses with Density-Split Statistics
• Lagrangian space remapping and the angular momentum reconstruction from cosmic structures
• Fast Sampling of Cosmological Initial Conditions with Gaussian Neural Posterior Estimation
• On the Connection between Field-Level Inference and $n$-point Correlation Functions
• Searching optimal scales for reconstructing cosmological initial conditions using convolutional neural networks
• From Redshift to Real Space: Combining Linear Theory With Neural Networks