Rethinking data-driven point spread function modeling with a differentiable optical model
This addresses a critical challenge in astronomy for researchers using space telescopes, by providing a high-fidelity PSF model that improves data analysis for scientific goals like weak gravitational lensing, though it is a novel method rather than incremental.
The paper tackles the problem of modeling the spatially varying point spread function (PSF) for space telescopes, where direct measurements are unavailable or degraded, by introducing WaveDiff, a differentiable optical model that shifts modeling from pixels to wavefronts. The result is a 6-fold reduction in pixel reconstruction errors at observation resolution and up to 250-fold reductions in size errors, enabling accurate PSF estimation without special calibration data.
In astronomy, upcoming space telescopes with wide-field optical instruments have a spatially varying point spread function (PSF). Specific scientific goals require a high-fidelity estimation of the PSF at target positions where no direct measurement of the PSF is provided. Even though observations of the PSF are available at some positions of the field of view (FOV), they are undersampled, noisy, and integrated into wavelength in the instrument's passband. PSF modeling represents a challenging ill-posed problem, as it requires building a model from degraded observations that can infer a super-resolved PSF at any wavelength and position in the FOV. Our model, coined WaveDiff, proposes a paradigm shift in the data-driven modeling of the point spread function field of telescopes. We change the data-driven modeling space from the pixels to the wavefront by adding a differentiable optical forward model into the modeling framework. This change allows the transfer of complexity from the instrumental response into the forward model. The proposed model relies on stochastic gradient descent to estimate its parameters. Our framework paves the way to building powerful, physically motivated models that do not require special calibration data. This paper demonstrates the WaveDiff model in a simplified setting of a space telescope. The proposed framework represents a performance breakthrough with respect to the existing state-of-the-art data-driven approach. The pixel reconstruction errors decrease 6-fold at observation resolution and 44-fold for a 3x super-resolution. The ellipticity errors are reduced at least 20 times, and the size error is reduced more than 250 times. By only using noisy broad-band in-focus observations, we successfully capture the PSF chromatic variations due to diffraction. Code available at https://github.com/tobias-liaudat/wf-psf.