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.

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

References in corpus (7)

• Simulating galaxy formation with black hole driven thermal and kinetic feedback
• Simba: Cosmological Simulations with Black Hole Growth and Feedback
• Fast likelihood-free cosmology with neural density estimators and active learning
• The Euclid mission design
• The CAMELS Multifield Dataset: Learning the Universe's Fundamental Parameters with Artificial Intelligence
• Constraining the reionization history using deep learning from 21cm tomography with the Square Kilometre Array
• 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