Differentiable design of the PIAA-ZWFS: a flexible wavefront sensor that approaches the fundamental limit
Abstract
Extreme adaptive optics (AO) is necessary for high contrast astronomy at scales of the habitable zone of nearby systems. We seek to evaluate wavefront sensors that approach fundamental limits of wavefront sensing, enabling adaptive optics systems to run faster or on fainter targets. We present the phase-induced amplitude apodisation Zernike wavefront sensor (PIAA-ZWFS): an adaptation of the conventional Zernike wavefront sensor (ZWFS) that leverages lossless apodisation of the pupil to concentrate the starlight in the focal plane. We optimise and evaluate the sensor with a differentiable modelling framework, drawing on concepts from Bayesian experimental design to minimise the variance of a maximum likelihood estimator that uses the system in the high Strehl regime. Our architecture shows state-of-the-art performance in simulation for different apertures, bandwidths, photon fluxes and source sizes, closing the gap to the fundamental limit by a factor 10 (2.5) compared to the conventional ZWFS (optimised ZWFS) in a typical photon-limited case. For extended sources, we show that even an ideal point source sensor rapidly becomes sub-optimal, and our system outperforms it for stellar diameters larger than 0.8λ/D. We verify that these gains do not come at the cost of dynamic range with either linear or non-linear reconstructors. Finally, we present a proof that there must be a trade-off between the information gained about amplitude and phase errors for any wavefront sensor. The PIAA-ZWFS is a viable wavefront sensor operating near the fundamental sensitivity limits.
Related
- Gaussian processes on ray-guided transformed uniform grids for fast, flexible, and auto-differentiable adaptive source reconstruction in lens modelling
- Model-based low-order wavefront sensing and control with a fully physical digital twin on MagAO-X
- Exploring reinforcement learning to enhance focal-plane wavefront control for vortex coronagraphs
- Reinforcement learning for post-coronagraphic wavefront control
- Wavefront-Conditioned Photon Measurements for Exoplanet Spectroscopy: Contrast Limits and Conditional Gains
BibTeX
@misc{taras2026differentiable,
title = {{Differentiable design of the PIAA-ZWFS: a flexible wavefront sensor that approaches the fundamental limit}},
author = {A. K. Taras and S. Y. Haffert and L. Desdoigts},
howpublished = {arXiv},
year = {2026},
month = oct,
doi = {10.48550/arxiv.2606.28136},
eprint = {2606.28136},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2606.28136},
note = {CC BY-SA 4.0},
}
RIS
TY - UNPB
TI - Differentiable design of the PIAA-ZWFS: a flexible wavefront sensor that approaches the fundamental limit
AU - A. K. Taras
AU - S. Y. Haffert
AU - L. Desdoigts
PY - 2026
DA - 2026/10/16
PB - arXiv
DO - 10.48550/arxiv.2606.28136
UR - https://arxiv.org/abs/2606.28136
AB - Extreme adaptive optics (AO) is necessary for high contrast astronomy at scales of the habitable zone of nearby systems. We seek to evaluate wavefront sensors that approach fundamental limits of wavefront sensing, enabling adaptive optics systems to run faster or on fainter targets. We present the phase-induced amplitude apodisation Zernike wavefront sensor (PIAA-ZWFS): an adaptation of the conventional Zernike wavefront sensor (ZWFS) that leverages lossless apodisation of the pupil to concentrate the starlight in the focal plane. We optimise and evaluate the sensor with a differentiable modelling framework, drawing on concepts from Bayesian experimental design to minimise the variance of a maximum likelihood estimator that uses the system in the high Strehl regime. Our architecture shows state-of-the-art performance in simulation for different apertures, bandwidths, photon fluxes and source sizes, closing the gap to the fundamental limit by a factor 10 (2.5) compared to the conventional ZWFS (optimised ZWFS) in a typical photon-limited case. For extended sources, we show that even an ideal point source sensor rapidly becomes sub-optimal, and our system outperforms it for stellar diameters larger than 0.8λ/D. We verify that these gains do not come at the cost of dynamic range with either linear or non-linear reconstructors. Finally, we present a proof that there must be a trade-off between the information gained about amplitude and phase errors for any wavefront sensor. The PIAA-ZWFS is a viable wavefront sensor operating near the fundamental sensitivity limits.
ER -