Differentiable Optics with dLux I: Deep calibration of Flat Field and Phase Retrieval with Automatic Differentiation
arXiv:2406.08703 · doi:10.1117/1.JATIS.9.2.028007
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, dLux, which uses the machine learning framework JAX to achieve GPU 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.
13 pages, 6 figures
References in corpus (13)
- Composable Effects for Flexible and Accelerated Probabilistic Programming in NumPyro
- Fast computation of Lyot-style coronagraph propagation
- Beyond the Kepler/K2 bright limit: variability in the seven brightest members of the Pleiades
- Analysis of a JWST NIRSpec Lab Time Series: Characterizing Systematics, Recovering Exoplanet Transit Spectroscopy, and Constraining a Noise Floor
- A PSF-based Approach to TESS High quality data Of Stellar clusters (PATHOS) -- I. Search for exoplanets and variable stars in the field of 47 Tuc
- Fisher Matrices and Confidence Ellipses: A Quick-Start Guide and Software
- High precision astrometry with a diffractive pupil telescope
- Kernel phase imaging with VLT/NACO: high-contrast detection of new candidate low-mass stellar companions at the diffraction limit
- The Near Infrared Imager and Slitless Spectrograph for the James Webb Space Telescope -- IV. Aperture Masking Interferometry
- Phase Retrieval and Design with Automatic Differentiation
- Rethinking data-driven point spread function modeling with a differentiable optical model
- The Near Infrared Imager and Slitless Spectrograph for JWST -- V. Kernel Phase Imaging and Data Analysis
- Exoplanet Imaging Data Challenge: benchmarking the various image processing methods for exoplanet detection