skip to main content


This content will become publicly available on November 15, 2018

Title: Uncertainties in Parameters Estimated with Neural Networks: Application to Strong Gravitational Lensing

In Hezaveh et al. (2017) we showed that deep learning can be used for model parameter estimation and trained convolutional neural networks to determine the parameters of strong gravitational lensing systems. Here we demonstrate a method for obtaining the uncertainties of these parameters. We review the framework of variational inference to obtain approximate posteriors of Bayesian neural networks and apply it to a network trained to estimate the parameters of the Singular Isothermal Ellipsoid plus external shear and total flux magnification. We show that the method can capture the uncertainties due to different levels of noise in the input data, as well as training and architecture-related errors made by the network. To evaluate the accuracy of the resulting uncertainties, we calculate the coverage probabilities of marginalized distributions for each lensing parameter. By tuning a single hyperparameter, the dropout rate, we obtain coverage probabilities approximately equal to the confidence levels for which they were calculated, resulting in accurate and precise uncertainty estimates. Our results suggest that neural networks can be a fast alternative to Monte Carlo Markov Chains for parameter uncertainty estimation in many practical applications, allowing more than seven orders of magnitude improvement in speed.
ORCiD logo [1] ; ORCiD logo [1] ; ORCiD logo [1]
  1. Stanford Univ. (United States). Kavli Inst. for Particle Astrophysics and Cosmology; SLAC National Accelerator Lab., Menlo Park, CA (United States)
Publication Date:
Grant/Contract Number:
Accepted Manuscript
Journal Name:
The Astrophysical Journal. Letters
Additional Journal Information:
Journal Volume: 850; Journal Issue: 1; Journal ID: ISSN 2041-8213
Institute of Physics (IOP)
Research Org:
SLAC National Accelerator Lab., Menlo Park, CA (United States)
Sponsoring Org:
Country of Publication:
United States
OSTI Identifier: