Strong-lensing source reconstruction with variationally optimised Gaussian processes

Abstract

Strong-lensing images provide a wealth of information both about the magnified source and about the dark matter distribution in the lens. Precision analyses of these images can be used to constrain the nature of dark matter. However, this requires high-fidelity image reconstructions and careful treatment of the uncertainties of both lens mass distribution and source light, which are typically difficult to quantify. In anticipation of future high-resolution datasets, in this work we leverage a range of recent developments in machine learning to develop a new Bayesian strong-lensing image analysis pipeline. Its highlights are: (A) a fast, GPU-enabled, end-to-end differentiable strong-lensing image simulator; (B) a new, statistically principled source model based on a computationally highly efficient approximation to Gaussian processes that also takes into account pixellation; and (C) a scalable variational inference framework that enables simultaneously deriving posteriors for tens of thousands of lens and source parameters and optimising hyperparameters via stochastic gradient descent. Besides efficient and accurate parameter estimation and lens model uncertainty quantification, the main aim of the pipeline is the generation of training data for targeted simulation-based inference of dark matter substructure, which we will exploit in a companion paper.

24 pages, 20 figures

References in corpus (8)

• Array Programming with NumPy
• The Sloan Lens ACS Survey. V. The Full ACS Strong-Lens Sample
• Bayesian Strong Gravitational-Lens Modeling on Adaptive Grids: Objective Detection of Mass Substructure in Galaxies
• Fast Automated Analysis of Strong Gravitational Lenses with Convolutional Neural Networks
• The Sloan Lens ACS Survey. XIII. Discovery of 40 New Galaxy-Scale Strong Lenses
• Mining for Dark Matter Substructure: Inferring subhalo population properties from strong lenses with machine learning
• Comparison of classical and Bayesian imaging in radio interferometry
• SLITronomy: towards a fully wavelet-based strong lensing inversion technique

Cited by in corpus (9)

• Using wavelets to capture deviations from smoothness in galaxy-scale strong lenses
• Strong Lensing by Galaxies
• Efficient Gravitational Wave Template Bank Generation with Differentiable Waveforms
• One never walks alone: the effect of the perturber population on subhalo measurements in strong gravitational lenses
• Pixelated Reconstruction of Foreground Density and Background Surface Brightness in Gravitational Lensing Systems using Recurrent Inference Machines
• Modeling lens potentials with continuous neural fields in galaxy-scale strong lenses
• Measuring the Substructure Mass Power Spectrum of 23 SLACS Strong Galaxy-Galaxy Lenses with Convolutional Neural Networks
• Analytic auto-differentiable $\Lambda$CDM cosmography
• Source REconstruction of Arcs behind Cluster Halos (SourceREACH): A New Source Reconstruction Algorithm Optimized for Giant Arcs and Galaxy Cluster Lenses