1. GIGA-Lens: Fast Bayesian Inference for Strong Gravitational Lens Modeling
- Author
-
A. Gu, X. Huang, W. Sheu, G. Aldering, A. S. Bolton, K. Boone, A. Dey, A. Filipp, E. Jullo, S. Perlmutter, D. Rubin, E. F. Schlafly, D. J. Schlegel, Y. Shu, S. H. Suyu, Laboratoire d'Astrophysique de Marseille (LAM), and Aix Marseille Université (AMU)-Institut national des sciences de l'Univers (INSU - CNRS)-Centre National d'Études Spatiales [Toulouse] (CNES)-Centre National de la Recherche Scientifique (CNRS)
- Subjects
Einstein rings ,FOS: Computer and information sciences ,Cosmology and Nongalactic Astrophysics (astro-ph.CO) ,[SDU.ASTR]Sciences of the Universe [physics]/Astrophysics [astro-ph] ,310 ,Galaxies and Cosmology ,FOS: Physical sciences ,Astronomy and Astrophysics ,GPU computing ,Bayesian statistics ,Statistics - Applications ,ddc ,Cosmology ,Markov chain Monte Carlo ,[SDU]Sciences of the Universe [physics] ,Space and Planetary Science ,Strong gravitational lensing ,Applications (stat.AP) ,Astrophysics - Instrumentation and Methods for Astrophysics ,Instrumentation and Methods for Astrophysics (astro-ph.IM) ,Astrophysics - Cosmology and Nongalactic Astrophysics - 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., Comment: 23 pages, 13 figures, 2 tables. Submitted to ApJ
- Published
- 2022
- Full Text
- View/download PDF