Posterior samples of source galaxies in strong gravitational lenses with score-based priors
read the original abstract
Inferring accurate posteriors for high-dimensional representations of the brightness of gravitationally-lensed sources is a major challenge, in part due to the difficulties of accurately quantifying the priors. Here, we report the use of a score-based model to encode the prior for the inference of undistorted images of background galaxies. This model is trained on a set of high-resolution images of undistorted galaxies. By adding the likelihood score to the prior score and using a reverse-time stochastic differential equation solver, we obtain samples from the posterior. Our method produces independent posterior samples and models the data almost down to the noise level. We show how the balance between the likelihood and the prior meet our expectations in an experiment with out-of-distribution data.
This paper has not been read by Pith yet.
Forward citations
Cited by 4 Pith papers
-
Joint inference of weak lensing convergence map and cosmology with diffusion models
A transformer-based diffusion model learns the joint distribution of convergence maps and cosmology from log-normal weak lensing simulations and generates calibrated posterior samples matching MCMC results.
-
An Unconditional Representation of the Conditional Score in Infinite-Dimensional Linear Inverse Problems
Derives the conditional score exactly from an unconditional score via affine maps for linear inverse problems in infinite dimensions, shifting computation to offline training.
-
ShuffleFlow: Scalable Posterior Inference for Bayesian Inverse Imaging
ShuffleFlow is a variational inference framework that partitions images with pixel-unshuffling and models the joint posterior over sub-images using a shared conditional normalizing flow conditioned on neural field fea...
-
MIRA: A Score for Conditional Distribution Accuracy and Model Comparison
MIRA is a new analytic score for conditional distribution accuracy derived from equal probability mass assignment, enabling Bayesian model comparison via direct posterior validation.
discussion (0)
Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.