Lya2pcf: an efficient pipeline to estimate two- and three-point correlation functions of the Lyman-$\alpha$ forest

TL;DR

Lya2pcf is a GPU-optimized pipeline that efficiently computes two- and three-point correlation functions of the Lyman-alpha forest, enabling advanced cosmological analyses with large datasets.

Contribution

The paper introduces Lya2pcf, a novel GPU-accelerated pipeline for calculating two- and three-point correlation functions from Lyman-alpha forest data, including the first measurement of the anisotropic three-point function on large samples.

Findings

Lya2pcf significantly reduces computation time compared to PICCA.

First measurement of anisotropic three-point correlation function on large spectroscopic data.

Demonstrates the potential for incorporating three-point statistics into future cosmological analyses.

Abstract

Studying the matter distribution in the universe through the Lyman-$\alpha$ forest allows us to constrain small-scale physics in the high-redshift regime. Spectroscopic quasar surveys are generating increasingly large datasets that require efficient algorithms to compute correlation functions. Moreover, cosmological analyses based on Lyman-$\alpha$ forests can significantly benefit from incorporating higher-order statistics alongside traditional two-point correlations. In this work, we present Lya2pcf, a pipeline designed to compute three-dimensional two-point and three-point correlation functions using Lyman-$\alpha$ forest data. The code implements standard algorithms widely used in current spectroscopic surveys for computing the two-point correlation function with its distortion matrix, covariance matrices; and it naturally extends the two-point estimator to three-point correlations. Thanks to GPU optimization, Lya2pcf achieves a substantial reduction in computational time for both the two-point correlation function and its distortion matrix when compared to the widely used PICCA code. We apply Lya2pcf to data from the Sloan Digital Sky Survey (SDSS) sixteenth data release (DR16) and a Dark Energy Spectroscopic Instrument Year-5 (DESI Y5) mock dataset, demonstrating overall performance gains over PICCA, especially on GPUs. We show the first measurement of the anisotropic three-point correlation function on a large spectroscopic sample for all possible triangles with scales up to 80 Mpc/h. The estimator's fast computation and the resulting signal-to-noise ratio -- above one for many triangle configurations -- demonstrate the viability of incorporating three-point statistics into future cosmological inference analyses, particularly with the larger datasets expected from Stage IV spectroscopic surveys.