Scalable Bayesian Inference for Finding Strong Gravitational Lenses [IMA]

http://arxiv.org/abs/2211.10479


Finding strong gravitational lenses in astronomical images allows us to assess cosmological theories and understand the large-scale structure of the universe. Previous works on lens detection do not quantify uncertainties in lens parameter estimates or scale to modern surveys. We present a fully amortized Bayesian procedure for lens detection that overcomes these limitations. Unlike traditional variational inference, in which training minimizes the reverse Kullback-Leibler (KL) divergence, our method is trained with an expected forward KL divergence. Using synthetic GalSim images and real Sloan Digital Sky Survey (SDSS) images, we demonstrate that amortized inference trained with the forward KL produces well-calibrated uncertainties in both lens detection and parameter estimation.

Read this paper on arXiv…

Y. Patel and J. Regier
Tue, 22 Nov 22
3/83

Comments: Accepted to the NeurIPS 2022 Workshop on Machine Learning and the Physical Sciences