Abstract: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 demonstrate that this diffusion model captures a greater diversity of possible deconvolutions compared to a conditional VAE.

What problem does this paper attempt to address?

The paper aims to address the problem of Point Spread Function (PSF) deconvolution in astronomical images. Specifically, it focuses on how to recover a clearer, more detailed original image from an observed image with a specific PSF. Since PSF convolution is an irreversible process, traditional deconvolution methods can only obtain a possible solution and cannot capture the diversity of all possible solutions. To solve this problem, the paper proposes a classifier-free conditional diffusion model and compares it with a Conditional Variational Autoencoder (CVAE). The main research contents include: 1. **Problem Background**: High-fidelity galaxy models are crucial for image deblending, analyzing lens substructures, and validating optical survey results. Traditional models like the Sersic profile cannot capture the rich structures in modern observational data, thus using deep generative models (such as VAE) to represent galaxies has become a trend. 2. **Problem Challenges**: Due to the irreversibility of PSF convolution, a single deconvolution result is insufficient to describe all possibilities. Existing deep generative models like VAE and GAN can generate multiple possible solutions, but the sample diversity is insufficient. 3. **Solution**: A classifier-free conditional diffusion model is proposed for the PSF deconvolution task of galaxy images. Comparative experiments with CVAE demonstrate that this model can generate more diverse deconvolution results. 4. **Experimental Validation**: Experiments were conducted using a dataset generated by the IllustrisTNG simulation, and the results show that the diffusion model maintains high-frequency details while significantly outperforming CVAE in terms of sample diversity. In summary, by introducing a conditional diffusion model, the paper addresses the issue of insufficient diversity in the PSF deconvolution process of galaxy images, providing a better tool for subsequent astronomical data analysis.

Diffusion Models for Probabilistic Deconvolution of Galaxy Images

Can Diffusion Model Conditionally Generate Astrophysical Images?

Learning the Evolution of Physical Structure of Galaxies via Diffusion Models

Using Galaxy Evolution as Source of Physics-Based Ground Truth for Generative Models

Diffusion with Forward Models: Solving Stochastic Inverse Problems Without Direct Supervision

Probabilistic Galaxy Field Generation with Diffusion Models

Diffusion Models in Vision: A Survey

Unveil Conditional Diffusion Models with Classifier-free Guidance: A Sharp Statistical Theory

Principled Probabilistic Imaging using Diffusion Models as Plug-and-Play Priors

Bayesian Deconvolution of Astronomical Images with Diffusion Models: Quantifying Prior-Driven Features in Reconstructions

Diffusion Models in Low-Level Vision: A Survey

Diffusion Model for Generative Image Denoising

Denoising Diffusion Delensing Delight: Reconstructing the Non-Gaussian CMB Lensing Potential with Diffusion Models

Efficient Conditional Diffusion Model with Probability Flow Sampling for Image Super-resolution

Distilling Diffusion Models into Conditional GANs

Strong-Lensing Source Reconstruction with Denoising Diffusion Restoration Models

Enhancing Diffusion-Based Image Synthesis with Robust Classifier Guidance

Toward a Diffusion-Based Generalist for Dense Vision Tasks

Diffusion-HMC: Parameter Inference with Diffusion Model driven Hamiltonian Monte Carlo