26 June 2023 Differentiable optics with ∂Lux: I—deep calibration of flat field and phase retrieval with automatic differentiation
Author Affiliations +
Abstract

The sensitivity limits of space telescopes are imposed by uncalibrated errors in the point spread function, photon-noise, background light, and detector sensitivity. These are typically calibrated with specialized wavefront sensor hardware and with flat fields obtained on the ground or with calibration sources, but these leave vulnerabilities to residual time-varying or non-common path aberrations and variations in the detector conditions. It is, therefore, desirable to infer these from science data alone, facing the prohibitively high dimensional problems of phase retrieval and pixel-level calibration. We introduce a new Python package for physical optics simulation, ∂ Lux, which uses the machine learning framework Jax to achieve graphics processing unit acceleration and automatic differentiation (autodiff), and apply this to simulating astronomical imaging. In this first of a series of papers, we show that gradient descent enabled by autodiff can be used to simultaneously perform phase retrieval and calibration of detector sensitivity, scaling efficiently to inferring millions of parameters. This new framework enables high dimensional optimization and inference in data analysis and hardware design in astronomy and beyond, which we explore in subsequent papers in this series.

© 2023 Society of Photo-Optical Instrumentation Engineers (SPIE)
Louis Desdoigts, Benjamin J. S. Pope, Jordan Dennis, and Peter G. Tuthill "Differentiable optics with ∂Lux: I—deep calibration of flat field and phase retrieval with automatic differentiation," Journal of Astronomical Telescopes, Instruments, and Systems 9(2), 028007 (26 June 2023). https://doi.org/10.1117/1.JATIS.9.2.028007
Received: 16 December 2022; Accepted: 30 May 2023; Published: 26 June 2023
Lens.org Logo
CITATIONS
Cited by 1 scholarly publication.
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Calibration

Point spread functions

Data modeling

Phase retrieval

Optical aberrations

Mathematical optimization

Machine learning

Back to Top