Probabilistic Mass Mapping with Neural Score Estimation

Abstract

Weak lensing mass-mapping is a useful tool to access the full distribution of dark matter on the sky, but because of intrinsic galaxy ellipticies and finite fields/missing data, the recovery of dark matter maps constitutes a challenging ill-posed inverse problem. We introduce a novel methodology allowing for efficient sampling of the high-dimensional Bayesian posterior of the weak lensing mass-mapping problem, and relying on simulations for defining a fully non-Gaussian prior. We aim to demonstrate the accuracy of the method on simulations, and then proceed to applying it to the mass reconstruction of the HST/ACS COSMOS field. The proposed methodology combines elements of Bayesian statistics, analytic theory, and a recent class of Deep Generative Models based on Neural Score Matching. This approach allows us to do the following: 1) Make full use of analytic cosmological theory to constrain the 2pt statistics of the solution. 2) Learn from cosmological simulations any differences between this analytic prior and full simulations. 3) Obtain samples from the full Bayesian posterior of the problem for robust Uncertainty Quantification. We demonstrate the method on the $\kappa$TNG simulations and find that the posterior mean significantly outperfoms previous methods (Kaiser-Squires, Wiener filter, Sparsity priors) both on root-mean-square error and in terms of the Pearson correlation. We further illustrate the interpretability of the recovered posterior by establishing a close correlation between posterior convergence values and SNR of clusters artificially introduced into a field. Finally, we apply the method to the reconstruction of the HST/ACS COSMOS field and yield the highest quality convergence map of this field to date.

Submitted to A&A, 20 pages, 15 figures, comments are welcome

References in corpus (13)

• Array Programming with NumPy
• COSMOS Photometric Redshifts with 30-bands for 2-deg2
• The XMM-Newton wide-field survey in the COSMOS field: VI. Statistical properties of clusters of galaxies
• COSMOS: 3D weak lensing and the growth of structure
• TensorFlow Distributions
• KiDS-450: Cosmological Constraints from Weak Lensing Peak Statistics - II: Inference from Shear Peaks using N-body Simulations
• KiDS-450: Cosmological Constraints from Weak Lensing Peak Statistics-I: Inference from Analytical Prediction of High Signal-to-Noise Ratio Convergence Peaks
• Cosmological Constraints From Weak Lensing Peak Statistics With CFHT Stripe 82 Survey
• Cosmic shear analysis of archival HST/ACS data: I. Comparison of early ACS pure parallel data to the HST/GEMS Survey
• Efficient Wiener filtering without preconditioning
• Starlet l1-norm for weak lensing cosmology
• Sparse reconstruction of the merging A520 cluster system
• A tomographic spherical mass map emulator of the KiDS-1000 survey using conditional generative adversarial networks

Cited by in corpus (22)

• The DAWES review 10: The impact of deep learning for the analysis of galaxy surveys
• Astronomia ex machina: a history, primer, and outlook on neural networks in astronomy
• IllustrisTNG in the HSC-SSP: image data release and the major role of mini mergers as drivers of asymmetry and star formation
• Beyond Gaussian Noise: A Generalized Approach to Likelihood Analysis with non-Gaussian Noise
• Diffusion-HMC: Parameter Inference with Diffusion-model-driven Hamiltonian Monte Carlo
• Gravitational-Wave Parameter Estimation in non-Gaussian noise using Score-Based Likelihood Characterization
• Uncertainty-Aware Blob Detection with an Application to Integrated-Light Stellar Population Recoveries
• Galaxy Imaging with Generative Models: Insights from a Two-Models Framework
• Generative modelling for mass-mapping with fast uncertainty quantification
• Tackling the Problem of Distributional Shifts: Correcting Misspecified, High-Dimensional Data-Driven Priors for Inverse Problems
• Dark from light (DfL): Inferring halo properties from luminous tracers with machine learning trained on cosmological simulations. I. Method, proof of concept & preliminary testing
• The spatially-resolved effect of mergers on the stellar mass assembly of MaNGA galaxies
• Stochastic Super-resolution of Cosmological Simulations with Denoising Diffusion Models
• Neural style transfer of weak lensing mass maps
• Impact of weak-lensing mass-mapping algorithms on cosmology inference
• Distribution-free uncertainty quantification for inverse problems: application to weak lensing mass mapping
• Bayesian imaging inverse problem with scattering transform
• Denoising weak lensing mass maps with diffusion model: systematic comparison with generative adversarial network
• Fast Sampling of Cosmological Initial Conditions with Gaussian Neural Posterior Estimation
• A plug-and-play approach with fast uncertainty quantification for weak lensing mass mapping
• Advancing weak lensing mass mapping with a mask-aware HEALPix transformer
• Generative modeling of convergence maps based on predicted one-point statistics