DeepAI AI Chat
Log In Sign Up

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

by   A. Gu, et al.

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 ๐’ช(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.


page 5

page 11

page 15

page 22

page 23

โˆ™ 11/30/2020

Large-Scale Gravitational Lens Modeling with Bayesian Neural Networks for Accurate and Precise Inference of the Hubble Constant

We investigate the use of approximate Bayesian neural networks (BNNs) in...
โˆ™ 11/10/2019

Modular Deep Learning Analysis of Galaxy-Scale Strong Lensing Images

Strong gravitational lensing of astrophysical sources by foreground gala...
โˆ™ 05/17/2021

GPU-Accelerated Hierarchical Bayesian Inference with Application to Modeling Cosmic Populations: CUDAHM

We describe a computational framework for hierarchical Bayesian inferenc...
โˆ™ 05/11/2019

Massive parallelization boosts big Bayesian multidimensional scaling

Big Bayes is the computationally intensive co-application of big data an...
โˆ™ 06/06/2019

Tensor Processing Units for Financial Monte Carlo

Monte Carlo methods are core to many routines in quantitative finance su...
โˆ™ 10/26/2020

Hierarchical Inference With Bayesian Neural Networks: An Application to Strong Gravitational Lensing

In the past few years, approximate Bayesian Neural Networks (BNNs) have ...