BLAST: Beyond Limber Angular power Spectra Toolkit. A fast and efficient algorithm for 3x2 pt analysis

Résumé Technique : Blast.jl – Un algorithme rapide et efficace pour l'analyse 3 × 2 pt

Énoncé du Problème
L'avènement des relevés photométriques et spectroscopiques de nouvelle génération (par exemple, Euclid, LSST, Roman, DESI) promet des mesures de haute précision de la structure à grande échelle (LSS). Ces relevés reposent largement sur les statistiques « 3 × 2 pt », qui combinent le regroupement des galaxies (clustering), les corrélations de cisaillement (shear-shear) du lentillage gravitationnel faible, et les corrélations croisées de lentillage matière-galaxie. La modélisation théorique de ces statistiques nécessite le calcul des spectres de puissance angulaire ($C_\ell$) en intégrant le produit du spectre de puissance de matière 3D et de fonctions sphériques de Bessel hautement oscillantes sur trois dimensions.

L'approche standard pour réduire le coût computationnel est l'approximation de Limber, qui simplifie l'intégrale 3D en une intégrale 1D en supposant que les modes à différentes époques ne contribuent pas à l'intégration le long de la ligne de visée. Cependant, cette approximation introduit des erreurs significatives, particulièrement sur les grandes échelles angulaires (faibles $\ell$). Pour exploiter pleinement le pouvoir de contrainte des futurs relevés et contrôler les biais systématiques, des calculs non-Limber précis sont essentiels. Pourtant, effectuer ces intégrales 3D complètes est prohibitivement coûteux pour les méthodes d'inférence de paramètres comme la méthode de Monte Carlo par chaînes de Markov (MCMC), qui nécessite des millions d'évaluations. Par exemple, des analyses récentes utilisant l'approximation de Limber ont nécessité 0,5 million d'heures-CPU pour converger.

Méthodologie
Les auteurs présentent Blast.jl, un nouvel algorithme conçu pour calculer efficacement les spectres de puissance angulaire complets non-Limber sans supposer un taux de croissance dépendant de l'échelle. La méthodologie centrale repose sur la décomposition du spectre de puissance de la matière à l'aide de polynômes de Chebyshev.

1. Décomposition de Chebyshev : L'algorithme approxime le spectre de puissance de la matière lisse $P(k, \chi_1, \chi_2)$ sur l'intervalle $[k_{min}, k_{max}]$ à l'aide d'une série de développement de Chebyshev. Cela permet de séparer l'intégrale en composantes géométriques et composantes cosmologiques.
2. Pré-calcul des termes géométriques : L'intégrale est restructurée de telle sorte que la partie la plus coûteuse en termes de calcul — l'intégration du produit de deux fonctions sphériques de Bessel et de polynômes de Chebyshev — devienne indépendante des paramètres cosmologiques. Ces termes ($\tilde{T}^{AB}_{n; \ell}$) sont calculés une seule fois et stockés.
3. Évaluation efficace : Pour tout nouveau modèle cosmologique, l'algorithme doit seulement recalculer les coefficients de Chebyshev du spectre de puissance, un processus qui est peu coûteux en calcul. Les autres intégrales sur la distance comobile ($\chi$) et sur le rapport $R = \chi_2/\chi_1$ sont effectuées à l'aide de règles de quadrature standards (règle de Simpson pour $\chi$ et quadrature de Clenshaw-Curtis pour $R$).
4. Transformation de coordonnées : Un changement de variables de $(\chi_1, \chi_2)$ vers $(\chi, R)$ est employé pour mieux échantillonner la région où $\chi_1 \approx \chi_2$, laquelle contribue le plus significativement à l'intégrale.
5. Correction non-linéaire : La méthode traite les spectres de puissance linéaires et non-linéaires séparément. La contribution non-linéaire, qui est significative uniquement sur les petites échelles où l'approximation de Limber est valide, est corrigée à l'aide d'une approche basée sur Limber, tandis que l'intégrale complète non-Limber est appliquée à la composante linéaire.

Contributions Clés

• Innovation Algorithmique : Blast.jl introduit une approche de décomposition utilisant les polynômes de Chebyshev qui évite les contraintes d'espacement de grille logarithmique des méthodes basées sur FFTLog (comme FKEM et matter) ainsi que la résolution d'équations différentielles itératives de la méthode de Levin.
• Stratégie de Pré-calcul : En isolant les intégrales géométriques indépendantes de la cosmologie, l'algorithme réduit drastiquement le temps requis pour les évaluations répétées lors de l'inférence de paramètres.
• Flexibilité : Contrairement à certaines méthodes concurrentes, Blast.jl ne suppose pas un taux de croissance indépendant de l'échelle, ce qui le rend applicable aux cosmologies avec des neutrinos massifs ou de la gravité modifiée.
• Open Source : Le code est disponible publiquement sur GitHub, accompagné d'un dépôt détaillant son utilisation.

Résultats
L'algorithme a été évalué par rapport au défi « N5K » (Non-local No-Nonsense Non-Limber Numerical Knock-out), organisé par le LSST Dark Energy Science Collaboration pour tester les méthodes d'intégration non-Limber. Les tests ont été menés sur un nœud de calcul haute performance (Narval) avec 64 threads, comparant Blast.jl aux vainqueurs du défi : FKEM (CosmoLike), matter, et Levin.

• Vitesse : À l'exigence de précision fiduciaire ($\Delta\chi^2 < 0,2$ pour $\ell < 200$), Blast.jl a atteint un temps d'exécution de 0,00512 secondes en utilisant 64 cœurs. C'est environ 10 fois plus rapide que FKEM (la méthode la plus rapide auparavant, 0,05871 s) et environ 15 fois plus rapide lorsqu'il est optimisé pour 32 cœurs (0,00360 s).
• Mise à l'échelle (Scaling) : Blast.jl démontre une excellente mise à l'échelle par rapport au nombre de compartiments de redshift tomographiques. Alors que les autres méthodes montrent une augmentation du temps d'exécution avec le nombre de compartiments, le temps d'exécution de Blast.jl reste effectivement indépendant du nombre de compartiments (allant de 2 à 15).
• Précision : La méthode répond aux exigences de précision du défi. Les auteurs notent que la précision peut être ajustée en modifiant le nombre de polynômes de Chebyshev et les points d'échantillonnage sans impact significatif sur le temps d'exécution.
• Compromis Précision vs Vitesse : Réduire la précision de la virgule flottante de Float64 à Float32 a amélioré le temps d'exécution d'un facteur 2 avec un impact négligeable sur la précision ($\Delta\chi^2$ est passé de 0,1054 à 0,1064).

Signification
L'article affirme que Blast.jl représente l'algorithme d'intégration non-Limber le plus rapide actuellement disponible, surpassant les méthodes de pointe d'un ordre de grandeur dans le scénario fiduciaire. Sa principale importance réside dans le fait de rendre les analyses cosmologiques utilisant les statistiques 3 × 2 pt informatiquement réalisables pour les relevés de prochaine génération sans dépendre de l'approximation erronée de Limber.

Les auteurs soulignent que cette vitesse est cruciale pour permettre l'inférence de paramètres par MCMC avec les ensembles de données massifs attendus pour les futurs relevés. De plus, la conception de l'algorithme permet des extensions futures, telles que l'inclusion des distorsions de vitesse en redshift (RSD), du biais de magnification, et des corrections relativistes, ainsi que l'intégration du lentillage du CMB. Les auteurs mentionnent également des plans pour incorporer la différenciation automatique, ce qui permettrait l'utilisation de méthodes d'inférence basées sur le gradient, accélérant davantage les pipelines d'analyse cosmologique.