A Bayesian Convolutional Neural Network for Robust Galaxy Ellipticity Regression

Abstract

Cosmic shear estimation is an essential scientific goal for large galaxy surveys. It refers to the coherent distortion of distant galaxy images due to weak gravitational lensing along the line of sight. It can be used as a tracer of the matter distribution in the Universe. The unbiased estimation of the local value of the cosmic shear can be obtained via Bayesian analysis which relies on robust estimation of the galaxies ellipticity (shape) posterior distribution. This is not a simple problem as, among other things, the images may be corrupted with strong background noise. For current and coming surveys, another central issue in galaxy shape determination is the treatment of statistically dominant overlapping (blended) objects. We propose a Bayesian Convolutional Neural Network based on Monte-Carlo Dropout to reliably estimate the ellipticity of galaxies and the corresponding measurement uncertainties. We show that while a convolutional network can be trained to correctly estimate well calibrated aleatoric uncertainty, -- the uncertainty due to the presence of noise in the images -- it is unable to generate a trustworthy ellipticity distribution when exposed to previously unseen data (i.e. here, blended scenes). By introducing a Bayesian Neural Network, we show how to reliably estimate the posterior predictive distribution of ellipticities along with robust estimation of epistemic uncertainties. Experiments also show that epistemic uncertainty can detect inconsistent predictions due to unknown blended scenes.

arXiv comment17 pages, 12 figures, submitted to ECML-PKDD 2021 in the Applied Data Science track

References in corpus (11)

• What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?
• Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learning
• LSST Science Book, Version 2.0
• A Method for Weak Lensing Observations
• Rotation-invariant convolutional neural networks for galaxy morphology prediction
• Weak lensing for precision cosmology
• The Third Gravitational Lensing Accuracy Testing (GREAT3) Challenge Handbook
• Decomposition of Uncertainty in Bayesian Deep Learning for Efficient and Risk-sensitive Learning
• SCARLET: Source separation in multi-band images by Constrained Matrix Factorization
• Deblending galaxies with Variational Autoencoders: a joint multi-band, multi-instrument approach
• Effects of overlapping sources on cosmic shear estimation: Statistical sensitivity and pixel-noise bias