GIGA-Lens: Fast Bayesian Inference for Strong Gravitational Lens Modeling
arXiv:2202.07663 · doi:10.3847/1538-4357/ac6de4
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 lensing systems expected to be discovered in the era of the Vera C. Rubin Observatory, Euclid, and the Nancy Grace Roman Space Telescope.
23 pages, 13 figures, 2 tables. Submitted to ApJ
References in corpus (24)
- Array Programming with NumPy
- 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
- A Bayesian approach to strong lensing modelling of galaxy clusters
- H0LiCOW V. New COSMOGRAIL time delays of HE0435-1223: to 3.8% precision from strong lensing in a flat CDM model
- Revealing the properties of dark matter in the merging cluster MACSJ0025.4-1222
- 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
- The Sloan Lens ACS Survey. XIII. Discovery of 40 New Galaxy-Scale Strong Lenses
- The Sloan Lens ACS Survey. XII. Extending Strong Lensing to Lower Masses
- Detection of the Gravitational Lens Magnifying a Type Ia Supernova
- PyAutoLens: Open-Source Strong Gravitational Lensing
- Strong lensing signatures of self-interacting dark matter in low-mass halos
- How to Find Gravitationally Lensed Type Ia Supernovae
- Turning Gravitationally Lensed Supernovae into Cosmological Probes
- Improved time-delay lens modelling and 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
- Quantifying the Line-of-Sight Halo Contribution to the Dark Matter Convergence Power Spectrum from Strong Gravitational Lenses
- 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 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