HIFlow: Generating Diverse HI Maps and Inferring Cosmology while Marginalizing over Astrophysics using Normalizing Flows

Abstract

A wealth of cosmological and astrophysical information is expected from many ongoing and upcoming large-scale surveys. It is crucial to prepare for these surveys now and develop tools that can efficiently extract most information. We present HIFlow: a fast generative model of the neutral hydrogen (HI) maps that is conditioned only on cosmology (${\Omega }_m$ and ${\sigma }_8$) and designed using a class of normalizing flow models, the Masked Autoregressive Flow (MAF). HIFlow is trained on the state-of-the-art simulations from the Cosmology and Astrophysics with MachinE Learning Simulations (CAMELS) project. HIFlow has the ability to generate realistic diverse maps without explicitly incorporating the expected 2D maps structure into the flow as an inductive bias. We find that HIFlow is able to reproduce the CAMELS average and standard deviation HI power spectrum (Pk) within a factor of $\lesssim$ 2, scoring a very high $R^2>90\%$. By inverting the flow, HIFlow provides a tractable high-dimensional likelihood for efficient parameter inference. We show that the conditional HIFlow on cosmology is successfully able to marginalize over astrophysics at the field level, regardless of the stellar and AGN feedback strengths. This new tool represents a first step toward a more powerful parameter inference, maximizing the scientific return of future HI surveys, and opening a new avenue to minimize the loss of complex information due to data compression down to summary statistics.

arXiv commentupdated: 14 pages, 10 figures, a new section on inference has been added during revision. Accepted for publication in ApJ

References in corpus (25)

• emcee: The MCMC Hammer
• LSST: from Science Drivers to Reference Design and Anticipated Data Products
• LOFAR: The LOw-Frequency ARray
• Simulating Galaxy Formation with the IllustrisTNG Model
• Simulating galaxy formation with black hole driven thermal and kinetic feedback
• Radiative transfer in a clumpy universe: IV. New synthesis models of the cosmic UV/X-ray background
• Simba: Cosmological Simulations with Black Hole Growth and Feedback
• Hydrogen Epoch of Reionization Array (HERA)
• Reionization and the Cosmic Dawn with the Square Kilometre Array
• The CAMELS project: Cosmology and Astrophysics with MachinE Learning Simulations
• Fast likelihood-free cosmology with neural density estimators and active learning
• Automatic physical inference with information maximising neural networks
• Deep learning from 21-cm tomography of the Cosmic Dawn and Reionization
• The Euclid mission design
• Constraining the astrophysics and cosmology from 21cm tomography using deep learning with the SKA
• Simulation-Based Inference of Reionization Parameters From 3D Tomographic 21 cm Lightcone Images
• The CAMELS Multifield Dataset: Learning the Universe's Fundamental Parameters with Artificial Intelligence
• Identifying Reionization Sources from 21cm Maps using Convolutional Neural Networks
• Deep Generative Models for Galaxy Image Simulations
• Machine learning astrophysics from 21 cm lightcones: impact of network architectures and signal contamination
• Removing Astrophysics in 21 cm maps with Neural Networks
• Constraining the reionization history using deep learning from 21cm tomography with the Square Kilometre Array
• HInet: Generating neutral hydrogen from dark matter with neural networks
• ARTIST: Fast radiative transfer for large-scale simulations of the epoch of reionisation
• Reionization with Simba: How much does astrophysics matter in modeling cosmic reionization?

Cited by in corpus (13)

• The CAMELS project: public data release
• $\mathrm{S}\mathrm{IM}\mathrm{BIG}$: Mock Challenge for a Forward Modeling Approach to Galaxy Clustering
• Efficient parameter inference for gravitational wave signals in the presence of transient noises using temporal and time-spectral fusion normalizing flow
• Learning the Universe: Cosmological and Astrophysical Parameter Inference with Galaxy Luminosity Functions and Colours
• Deep learning-driven likelihood-free parameter inference for 21-cm forest observations
• Modeling Post-Reionization HI Distributions in Fuzzy Dark Matter Cosmologies Using Conditional Normalizing Flows
• Massive $\nu$s through the CNN lens: interpreting the field-level neutrino mass information in weak lensing
• Cosmology from Point Clouds with Dark Matter Halos from the Quijote Simulations
• The Cosmological analysis of X-ray cluster surveys VII. Bypassing scaling relations with Lagrangian Deep Learning and Simulation-based inference
• CASCO: Cosmological and AStrophysical parameters from Cosmological simulations and Observations -- II. Constraining cosmology and astrophysical processes with early- and late-type galaxies
• Detecting Modeling Bias with Continuous Time Flow Models on Weak Lensing Maps
• Large, fast and accurate HI intensity maps with latent overlap diffusion
• Accelerating reionization constraints: An ANN-emulator framework for the SCRIPT Semi-numerical Model