gallifrey: JAX-based Gaussian process structure learning for astronomical time series

Abstract

Gaussian processes (GPs) have become a common tool in astronomy for analysing time series data, particularly in exoplanet science and stellar astrophysics. However, choosing the appropriate covariance structure for a GP model remains a challenge in many situations, limiting model flexibility and performance. This work provides an introduction to recent advances in GP structure learning methods, which enable the automated discovery of optimal GP kernels directly from the data, with the aim of making these methods more accessible to the astronomical community. We present gallifrey, a JAX-based Python package that implements a sequential Monte Carlo algorithm for Bayesian kernel structure learning. This approach defines a prior distribution over kernel structures and hyperparameters, and efficiently samples the GP posterior distribution using a novel involutive Markov chain Monte Carlo procedure. We applied gallifrey to common astronomical time series tasks, including stellar variability modelling, exoplanet transit modelling, and transmission spectroscopy. We show that this methodology can accurately interpolate and extrapolate stellar variability, recover transit parameters with robust uncertainties, and derive transmission spectra by effectively separating the background from the transit signal. When compared with traditional fixed-kernel approaches, we show that structure learning has advantages in terms of accuracy and uncertainty estimation. Structure learning can enhance the performance of GP regression for astronomical time series modelling. We discuss a road map for algorithmic improvements in terms of scalability to larger datasets, so that the methods presented here can be applied to future stellar and exoplanet missions such as PLATO.

16 pages, 10 figures, 3 tables, accepted in A&A, code available at https://github.com/ChrisBoettner/gallifrey, documentation available at https://chrisboettner.github.io/gallifrey/

References in corpus (24)

• The Transiting Exoplanet Survey Satellite
• The PLATO 2.0 Mission
• Fast and scalable Gaussian process modeling with applications to astronomical time series
• astroquery: An Astronomical Web-Querying Package in Python
• The effect of red noise on planetary transit detection
• Planets and Stellar Activity: Hide and Seek in the CoRoT-7 system
• Juliet: a versatile modelling tool for transiting and non-transiting exoplanetary systems
• A simple method to estimate radial velocity variations due to stellar activity using photometry
• A Gaussian process framework for modelling instrumental systematics: application to transmission spectroscopy
• An ultrahot gas-giant exoplanet with a stratosphere
• Inferring probabilistic stellar rotation periods using Gaussian processes
• 197 Candidates and 104 Validated Planets in K2's First Five Fields
• The Deep Blue Color of HD189733b: Albedo Measurements with Hubble Space Telescope/Space Telescope Imaging Spectrograph at Visible Wavelengths
• A uniform analysis of HD209458b Spitzer/IRAC lightcurves with Gaussian process models
• Precise time-series photometry for the Kepler-2.0 mission
• Reliable inference of exoplanet light curve parameters using deterministic and stochastic systematics models
• NGTS clusters survey. I. Rotation in the young benchmark open cluster Blanco 1
• Improving transit characterisation with Gaussian process modelling of stellar variability
• Quasi-periodic Gaussian Processes for stellar activity: from physical to kernel parameters
• Mapping stellar surfaces II: An interpretable Gaussian process model for light curves
• The PLATO Solar-like Light-curve Simulator: A tool to generate realistic stellar light-curves with instrumental effects representative of the PLATO mission
• LRG-BEASTS: sodium absorption and Rayleigh scattering in the atmosphere of WASP-94A b using NTT/EFOSC2
• Bayesian Synthesis of Probabilistic Programs for Automatic Data Modeling
• LRG-BEASTS: Evidence for clouds in the transmission spectrum of HATS-46 b