Focal plane wavefront control with model-based reinforcement learning

Abstract

The direct imaging of potentially habitable exoplanets is one prime science case for high-contrast imaging instruments on extremely large telescopes. Most such exoplanets orbit close to their host stars, where their observation is limited by fast-moving atmospheric speckles and quasi-static non-common-path aberrations (NCPA). Conventional NCPA correction methods often use mechanical mirror probes, which compromise performance during operation. This work presents machine-learning-based NCPA control methods that automatically detect and correct both dynamic and static NCPA errors by leveraging sequential phase diversity. We extend previous work in reinforcement learning for AO to focal plane control. A new model-based RL algorithm, Policy Optimization for NCPAs (PO4NCPA), interprets the focal-plane image as input data and, through sequential phase diversity, determines phase corrections that optimize both non-coronagraphic and post-coronagraphic PSFs without prior system knowledge. Further, we demonstrate the effectiveness of this approach by numerically simulating static NCPA errors on a ground-based telescope and an infrared imager affected by water-vapor-induced seeing (dynamic NCPAs). Simulations show that PO4NCPA robustly compensates static and dynamic NCPAs. In static cases, it achieves near-optimal focal-plane light suppression with a coronagraph and near-optimal Strehl without one. With dynamics NCPA, it matches the performance of the modal least-squares reconstruction combined with a 1-step delay integrator in these metrics. The method remains effective for the ELT pupil, vector vortex coronagraph, and under photon and background noise. PO4NCPA is model-free and can be directly applied to standard imaging as well as to any coronagraph. Its sub-millisecond inference times and performance also make it suitable for real-time low-order correction of atmospheric turbulence beyond HCI.

13 pages, 11 figures accepted by A&A

References in corpus (22)

• Angular Differential Imaging: a Powerful High-Contrast Imaging Technique
• The Subaru Coronagraphic Extreme Adaptive Optics system: enabling high-contrast imaging on solar-system scales
• Limits of Adaptive Optics for high contrast imaging
• Combining high-dispersion spectroscopy (HDS) with high contrast imaging (HCI): Probing rocky planets around our nearest neighbors
• Fundamental limitations on Earth-like planet detection with Extremely Large Telescopes
• The Asymmetric Pupil Fourier Wavefront Sensor
• Lyot-based Low Order Wavefront Sensor for Phase-mask Coronagraphs: Principle, Simulations and Laboratory Experiments
• Post-coronagraphic tip-tilt sensing for vortex phase masks: the QACITS technique
• Calibration of quasi-static aberrations in exoplanet direct-imaging instruments with a Zernike phase-mask sensor. III. On-sky validation in VLT/SPHERE
• Coronagraphic Low Order Wavefront Sensor: Principle and Application to a Phase-Induced Amplitude Coronagraph
• Adaptive Optics control using Model-Based Reinforcement Learning
• Differential Imaging with a Multicolor Detector Assembly: A New ExoPlanet Finder Concept
• Direct characterization of young giant exoplanets at high spectral resolution by coupling SPHERE and CRIRES+
• Nonlinear wavefront reconstruction with convolutional neural networks for Fourier-based wavefront sensors
• Fast & Furious focal-plane wavefront sensing
• Advancing European High-Contrast Imaging R&D Towards the Habitable Worlds Observatory
• Three years of harvest with the vector vortex coronagraph in the thermal infrared
• Focal Plane Wavefront Sensing using Machine Learning: Performance of Convolutional Neural Networks compared to Fundamental Limits
• Towards on-sky adaptive optics control using reinforcement learning
• Implicit electric field Conjugation: Data-driven focal plane control
• On-sky verification of Fast and Furious focal-plane wavefront sensing: Moving forward toward controlling the island effect at Subaru/SCExAO
• A deep learning approach for focal-plane wavefront sensing using vortex phase diversity