Fast computation of temperature and polarization coupling matrices

TL;DR

The paper introduces a fast, exact method for computing CMB mode-coupling matrices using optimized Wigner-3j symbol evaluations, significantly accelerating calculations while maintaining accuracy.

Contribution

It presents a novel approach that reduces computational cost of coupling matrices by exploiting analytic structures and recurrence relations, enabling efficient parallelization on modern hardware.

Findings

Achieves 6-25x speedup over standard methods.

Maintains numerical accuracy in coupling matrix calculations.

Supports efficient parallelization on CPUs and GPUs.

Abstract

We present a fast and exact method for computing CMB mode-coupling matrices based on an optimised evaluation of Wigner-3j symbols. The method exploits analytic structure in the relevant Wigner-3j symbol configurations appearing in temperature and polarization coupling matrices, expressing all required quantities in terms of a small set of recurrence-generated values which are precomputed and stored in lookup tables. This approach reduces the computational cost of constructing the full coupling matrices whilst maintaining numerical accuracy. We demonstrate the performance of the threej_cosmo implementation using realistic survey masks from current CMB experiments. Relative to standard recursion-based approaches used in existing pseudo-C_l pipelines, the method achieves speedups of 6-25x in practical coupling-matrix constructions, with the largest gains occurring at high multipoles. The algorithm admits efficient parallelisation on both CPUs and GPUs, the latter providing additional acceleration, up to a further order of 50 on modern hardware, without altering the underlying formalism. Beyond full matrix construction, the approach is naturally suited to applications in which only a restricted set of l3 modes is required for each (l1,l2) pair, such as in the computation of band-limited coupling matrices and analytic covariance terms. These features make threej_cosmo a practical backend for pseudo-C_l estimation and related calculations in next-generation CMB analysis pipelines.