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

Abstract

We investigate the use of approximate Bayesian neural networks (BNNs) in modeling hundreds of time-delay gravitational lenses for Hubble constant ($H_0$) determination. Our BNN was trained on synthetic HST-quality images of strongly lensed active galactic nuclei (AGN) with lens galaxy light included. The BNN can accurately characterize the posterior PDFs of model parameters governing the elliptical power-law mass profile in an external shear field. We then propagate the BNN-inferred posterior PDFs into ensemble $H_0$ inference, using simulated time delay measurements from a plausible dedicated monitoring campaign. Assuming well-measured time delays and a reasonable set of priors on the environment of the lens, we achieve a median precision of $9.3$\% per lens in the inferred $H_0$. A simple combination of 200 test-set lenses results in a precision of 0.5 ${\text{km s}}^{-1}{\text{ Mpc}}^{-1}$ ($0.7\%$), with no detectable bias in this $H_0$ recovery test. The computation time for the entire pipeline -- including the training set generation, BNN training, and $H_0$ inference -- translates to 9 minutes per lens on average for 200 lenses and converges to 6 minutes per lens as the sample size is increased. Being fully automated and efficient, our pipeline is a promising tool for exploring ensemble-level systematics in lens modeling for $H_0$ inference.

21 pages (+2 appendix), 17 figures. Published in ApJ. Code at https://github.com/jiwoncpark/h0rton. Datasets, trained models, and inference results at https://zenodo.org/record/4300382

References in corpus (7)

• The Sloan Lens ACS Survey. V. The Full ACS Strong-Lens Sample
• Exoplanet population inference and the abundance of Earth analogs from noisy, incomplete catalogs
• Fast Automated Analysis of Strong Gravitational Lenses with Convolutional Neural Networks
• Uncertainties in Parameters Estimated with Neural Networks: Application to Strong Gravitational Lensing
• H0LiCOW III. Quantifying the effect of mass along the line of sight to the gravitational lens HE 0435-1223 through weighted galaxy counts
• Gravitational lensing model degeneracies: Is steepness all-important?
• The use of convolutional neural networks for modelling large optically-selected strong galaxy-lens samples

Cited by in corpus (3)

• lenstronomy II: A gravitational lensing software ecosystem
• Strong lensing time-delay cosmography in the 2020s
• Extracting cosmological parameters from N-body simulations using machine learning techniques