GIGA-Lens: Fast Bayesian Inference for Strong Gravitational Lens Modeling

Abstract

We present GIGA-Lens: a gradient-informed, GPU-accelerated Bayesian framework for modeling strong gravitational lensing systems, implemented in TensorFlow and JAX. The three components, optimization using multi-start gradient descent, posterior covariance estimation with variational inference, and sampling via Hamiltonian Monte Carlo, all take advantage of gradient information through automatic differentiation and massive parallelization on graphics processing units (GPUs). We test our pipeline on a large set of simulated systems and demonstrate in detail its high level of performance. The average time to model a single system on four Nvidia A100 GPUs is 105 seconds. The robustness, speed, and scalability offered by this framework make it possible to model the large number of strong lenses found in current surveys and present a very promising prospect for the modeling of $\mathcal{O}(10^5)$ lensing systems expected to be discovered in the era of the Vera C. Rubin Observatory, Euclid, and the Nancy Grace Roman Space Telescope.

arXiv comment23 pages, 13 figures, 2 tables. Submitted to ApJ

References in corpus (72)

• Array Programming with NumPy
• emcee: The MCMC Hammer
• Variational Inference: A Review for Statisticians
• A Comprehensive Measurement of the Local Value of the Hubble Constant with 1 km/s/Mpc Uncertainty from the Hubble Space Telescope and the SH0ES Team
• Large Magellanic Cloud Cepheid Standards Provide a 1% Foundation for the Determination of the Hubble Constant and Stronger Evidence for Physics Beyond LambdaCDM
• Overview of the DESI Legacy Imaging Surveys
• Parallel Tempering: Theory, Applications, and New Perspectives
• H0LiCOW XIII. A 2.4% measurement of $H_0$ from lensed quasars: $5.3\sigma$ tension between early and late-Universe probes
• The Carnegie-Chicago Hubble Program. VIII. An Independent Determination of the Hubble Constant Based on the Tip of the Red Giant Branch
• A Bayesian approach to strong lensing modelling of galaxy clusters
• The Sloan Lens ACS Survey. III - The Structure and Formation of Early-type Galaxies and their Evolution since z~1
• The Sloan Lens ACS Survey. I. A Large Spectroscopically Selected Sample of Massive Early-Type Lens Galaxies
• Particle Swarm Optimization: A survey of historical and recent developments with hybridization perspectives
• H0LiCOW V. New COSMOGRAIL time delays of HE0435-1223: $H_0$ to 3.8% precision from strong lensing in a flat $\Lambda$CDM model
• Strong Lensing by Galaxies
• Dissecting the Gravitational Lens B1608+656. II. Precision Measurements of the Hubble Constant, Spatial Curvature, and the Dark Energy Equation of State
• Gravitationally lensed quasars and supernovae in future wide-field optical imaging surveys
• Calibration of the Tip of the Red Giant Branch (TRGB)
• Gravitational detection of a low-mass dark satellite at cosmological distance
• TDCOSMO IV: Hierarchical time-delay cosmography -- joint inference of the Hubble constant and galaxy density profiles
• Two accurate time-delay distances from strong lensing: Implications for cosmology
• The population of galaxy-galaxy strong lenses in forthcoming optical imaging surveys
• Detection of lensing substructure using ALMA observations of the dusty galaxy SDP.81
• Lenstronomy: multi-purpose gravitational lens modelling software package
• Detection of a Dark Substructure through Gravitational Imaging
• Revealing the properties of dark matter in the merging cluster MACSJ0025.4-1222
• iPTF16geu: A multiply imaged, gravitationally lensed type Ia supernova
• Time Delay Cosmography
• TDCOSMO. I. An exploration of systematic uncertainties in the inference of $H_0$ from time-delay cosmography
• lenstronomy II: A gravitational lensing software ecosystem
• Cosmological Constraints from Strong Gravitational Lensing in Galaxy Clusters
• CLASH-VLT: Insights on the mass substructures in the Frontier Fields Cluster MACS J0416.1-2403 through accurate strong lens modeling
• An excess of small-scale gravitational lenses observed in galaxy clusters
• Gravitational lens modeling with basis sets
• The Strong Gravitational Lens Finding Challenge
• Fast calculation of a family of elliptical mass gravitational lens models
• Finding strong lenses in CFHTLS using convolutional neural networks
• Dark matter halos of massive elliptical galaxies at $z\sim 0.2$ are well described by the Navarro-Frenk-White profile
• Low-mass halo perturbations in strong gravitational lenses at redshift z$\sim$0.5 are consistent with CDM
• Statistics of mass substructure from strong gravitational lensing: quantifying the mass fraction and mass function
• The stellar velocity dispersion of the lens galaxy in MG2016+112 at z=1.004
• The Sloan Lens ACS Survey. XIII. Discovery of 40 New Galaxy-Scale Strong Lenses
• Finding high-redshift strong lenses in DES using convolutional neural networks
• The Sloan Lens ACS Survey. XII. Extending Strong Lensing to Lower Masses
• Detection of the Gravitational Lens Magnifying a Type Ia Supernova
• Finding Strong Gravitational Lenses in the DESI DECam Legacy Survey
• Discovering New Strong Gravitational Lenses in the DESI Legacy Imaging Surveys
• Precise Time Delays from Strongly Gravitationally Lensed Type Ia Supernovae with Chromatically Microlensed Images
• PyAutoLens: Open-Source Strong Gravitational Lensing
• Strong lensing signatures of self-interacting dark matter in low-mass halos
• The elliptical power law profile lens
• Rates and Properties of Strongly Gravitationally Lensed Supernovae and their Host Galaxies in Time-Domain Imaging Surveys
• SN Refsdal : Photometry and Time Delay Measurements of the First Einstein Cross Supernova
• HOLISMOKES -- II. Identifying galaxy-scale strong gravitational lenses in Pan-STARRS using convolutional neural networks
• How to Find Gravitationally Lensed Type Ia Supernovae
• A Spectroscopic Redshift for the Cl0024+16 Multiple Arc System: Implications for the Central Mass Distribution
• The Impact of Microlensing on the Standardisation of Strongly Lensed Type Ia Supernovae
• Efficient Multi-Start Strategies for Local Search Algorithms
• HOLISMOKES -- I. Highly Optimised Lensing Investigations of Supernovae, Microlensing Objects, and Kinematics of Ellipticals and Spirals
• The Gravitational Lens System B1422$+$231: Dark Matter, Superluminal Expansion and the Hubble Constant
• Turning Gravitationally Lensed Supernovae into Cosmological Probes
• Improved time-delay lens modelling and $H_0$ inference with transient sources
• Distances with <4% Precision from Type Ia Supernovae in Young Star-Forming Environments
• Direct Detection of Dark Matter Substructure in Strong Lens Images with Convolutional Neural Networks
• Differentiable Strong Lensing: Uniting Gravity and Neural Nets through Differentiable Probabilistic Programming
• Quantifying the Line-of-Sight Halo Contribution to the Dark Matter Convergence Power Spectrum from Strong Gravitational Lenses
• Prediction of Supernova Rates in Known Galaxy-galaxy Strong-lens Systems
• Kiloparsec Mass/Light Offsets in the Galaxy Pair-Ly$\alpha$ Emitter Lens System SDSS\,J1011$+$0143
• HST Discovery of a z = 3.9 Multiply Imaged Galaxy Behind the Complex Cluster Lens WARPS J1415.1+36 at z = 1.026
• SLITronomy: towards a fully wavelet-based strong lensing inversion technique
• HOLISMOKES -- VII. Time-delay measurement of strongly lensed Type Ia supernovae using machine learning
• Galaxy mass profiles from strong lensing I: The circular power-law model

Cited by in corpus (21)

• Strong lensing time-delay cosmography in the 2020s
• Sensitivity of strong lensing observations to dark matter substructure: a case study with Euclid
• Using wavelets to capture deviations from smoothness in galaxy-scale strong lenses
• TDCOSMO X. Automated Modeling of 9 Strongly Lensed Quasars and Comparison Between Lens Modeling Software
• New Strong Gravitational Lenses from the DESI Legacy Imaging Surveys Data Release 9
• One never walks alone: the effect of the perturber population on subhalo measurements in strong gravitational lenses
• HOLISMOKES -- IX. Neural network inference of strong-lens parameters and uncertainties from ground-based images
• HOLISMOKES -- X. Comparison between neural network and semi-automated traditional modeling of strong lenses
• LeMoN: Lens Modelling with Neural networks -- I. Automated modelling of strong gravitational lenses with Bayesian Neural Networks
• Modeling lens potentials with continuous neural fields in galaxy-scale strong lenses
• Exploiting the diversity of modeling methods to probe systematic biases in strong lensing analyses
• El Gordo needs El Anzuelo: Probing the structure of cluster members with multi-band extended arcs in JWST data
• Lens Modeling of STRIDES Strongly Lensed Quasars using Neural Posterior Estimation
• A Differentiable, End-to-End Forward Model for 21 cm Cosmology: Estimating the Foreground, Instrument, and Signal Joint Posterior
• DESI-253.2534+26.8843: A New Einstein Cross Spectroscopically Confirmed with VLT/MUSE and Modeled with GIGA-Lens
• Simulation-guided galaxy evolution inference: A case study with strong lensing galaxies
• Analytic auto-differentiable $\Lambda$CDM cosmography
• Accelerating lensed quasar discovery and modeling with physics-informed variational autoencoders
• Multi-band analysis of strong gravitationally lensed post-blue nugget candidates from the Kilo-Degree Survey
• Differentiable Modeling of Planet and Substellar Atmosphere: High-Resolution Emission, Transmission, and Reflection Spectroscopy with ExoJAX2
• JAXtronomy: A JAX port of lenstronomy