Predicting dark matter halo formation in N-body simulations with deep regression networks

Abstract

Dark matter haloes play a fundamental role in cosmological structure formation. The most common approach to model their assembly mechanisms is through N-body simulations. In this work we present an innovative pathway to predict dark matter halo formation from the initial density field using a Deep Learning algorithm. We implement and train a Deep Convolutional Neural Network (DCNN) to solve the task of retrieving Lagrangian patches from which dark matter halos will condense. The volumetric multi-label classification task is turned into a regression problem by means of the euclidean distance transformation. The network is complemented by an adaptive version of the watershed algorithm to form the entire protohalo identification pipeline. We show that splitting the segmentation problem into two distinct sub-tasks allows for training smaller and faster networks, while the predictive power of the pipeline remains the same. The model is trained on synthetic data derived from a single full N-body simulation and achieves deviations of ~10% when reconstructing the dark matter halo mass function at z=0. This approach represents a promising framework for learning highly non-linear relations in the primordial density field. As a practical application, our method can be used to produce mock dark matter halo catalogues directly from the initial conditions of N-body simulations.

arXiv comment10 pages, 8 figures, 1 table, submitted to MNRAS, comments welcome

References in corpus (19)

• Ellipsoidal collapse and an improved model for the number and spatial distribution of dark matter haloes
• ParseNet: Looking Wider to See Better
• The Connection between Galaxies and their Dark Matter Halos
• How do galaxies populate Dark Matter halos?
• Learning Deconvolution Network for Semantic Segmentation
• HOP: A New Group-Finding Algorithm for N-body Simulations
• Physics-guided Neural Networks (PGNN): An Application in Lake Temperature Modeling
• Evolution of the Mass Function of Dark Matter Haloes
• Learning to Predict the Cosmological Structure Formation
• Painting galaxies into dark matter halos using machine learning
• Machine learning cosmological structure formation
• Classifying the Large Scale Structure of the Universe with Deep Neural Networks
• A volumetric deep Convolutional Neural Network for simulation of mock dark matter halo catalogues
• A Hybrid Deep Learning Approach to Cosmological Constraints From Galaxy Redshift Surveys
• An interpretable machine learning framework for dark matter halo formation
• From Dark Matter to Galaxies with Convolutional Networks
• Painting halos from cosmic density fields of dark matter with physically motivated neural networks
• Neural physical engines for inferring the halo mass distribution function
• The galaxy - halo connection in low mass halos

Cited by in corpus (23)

• Large-scale dark matter simulations
• Super-resolution emulator of cosmological simulations using deep physical models
• Learning effective physical laws for generating cosmological hydrodynamics with Lagrangian Deep Learning
• AI-assisted super-resolution cosmological simulations II: Halo substructures, velocities and higher order statistics
• Machine learning constraints on deviations from general relativity from the large scale structure of the Universe
• Machine Learning improved fits of the sound horizon at the baryon drag epoch
• Machine Learning meets the redshift evolution of the CMB Temperature
• Machine learning forecasts of the cosmic distance duality relation with strongly lensed gravitational wave events
• Simulation-based inference of dynamical galaxy cluster masses with 3D convolutional neural networks
• Nonlinear 3D Cosmic Web Simulation with Heavy-Tailed Generative Adversarial Networks
• Bayesian Inference of Initial Conditions from Non-Linear Cosmic Structures using Field-Level Emulators
• A semi-supervised approach to dark matter searches in direct detection data with machine learning
• Testing the $\Lambda$CDM paradigm with growth rate data and machine learning
• Machine learning cosmic inflation
• Halo assembly bias from a deep learning model of halo formation
• AI-assisted super-resolution cosmological simulations III: Time evolution
• Emulating Sunyaev-Zeldovich Images of Galaxy Clusters using Auto-Encoders
• Deep learning insights into cosmological structure formation
• Complementary consistency test of the Copernican principle via Noether's theorem and machine learning forecasts
• Dark Matter Subhalos, Strong Lensing and Machine Learning
• Learning the Universe: Learning to Optimize Cosmic Initial Conditions with Non-Differentiable Structure Formation Models
• Variational autoencoder for generating realistic $N$-body simulations for dark matter halos
• Segmenting proto-halos with vision transformers