A PyTorch Benchmark for High-Contrast Imaging Post Processing

Chia-Lin Ko,Ewan S. Douglas,Justin Hom
2024-09-25
Abstract:Direct imaging of exoplanets is a challenging task that involves distinguishing faint planetary signals from the overpowering glare of their host stars, often obscured by time-varying stellar noise known as "speckles". The predominant algorithms for speckle noise subtraction employ principal-based point spread function (PSF) fitting techniques to discern planetary signals from stellar speckle noise. We introduce torchKLIP, a benchmark package developed within the machine learning (ML) framework PyTorch. This work enables ML techniques to utilize extensive PSF libraries to enhance direct imaging post-processing. Such advancements promise to improve the post-processing of high-contrast images from leading-edge astronomical instruments like the James Webb Space Telescope and extreme adaptive optics systems.
Earth and Planetary Astrophysics,Instrumentation and Methods for Astrophysics,Solar and Stellar Astrophysics
What problem does this paper attempt to address?