Workshop: 2nd ICML Workshop on Machine Learning for Astrophysics

Diffusion Models for Probabilistic Deconvolution of Galaxy Images

Zhiwei Xue · Yuhang Li · Yash Patel · Jeffrey Regier


Telescopes capture images with a particular point-spread function (PSF). Inferring what an image would have looked like with a much sharper PSF, a problem known as PSF deconvolution, is ill-posed because PSF convolution is not an invertible transformation. Deep generative models are appealing for PSF deconvolution because they can infer a posterior distribution over candidate images that, if convolved with the PSF, could have generated the observation. However, classical deep generative models such as VAEs and GANs often provide inadequate sample diversity. As an alternative, we propose a classifier-free conditional diffusion model for PSF deconvolution of galaxy images. We empirically demonstrate that this diffusion model captures a greater diversity of possible deconvolutions compared to a conditional VAE.

Chat is not available.