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

Sintesi Tecnica: Blast.jl – Un algoritmo veloce ed efficiente per l'analisi 3 × 2 pt

Problema
L'avvento di indagini fotometriche e spettroscopiche di prossima generazione (ad esempio, Euclid, LSST, Roman, DESI) promette misurazioni di alta precisione della Struttura su Grande Scala (LSS). Queste indagini si affidano pesantemente alle statistiche "3 × 2 pt", che combinano l'aggregazione delle galassie, le correlazioni shear-shear del lensing debole e le cross-correlazioni galaxy-galaxy lensing. La modellazione teorica di queste statistiche richiede il calcolo degli spettri di potenza angolari ($C_\ell$) integrando il prodotto dello spettro di potenza della materia 3D con funzioni di Bessel sferiche altamente oscillanti su tre dimensioni.

L'approccio standard per ridurre il costo computazionale è l'approssimazione di Limber, che semplifica l'integrale 3D in un integrale 1D assumendo che i modi a diverse epoche non contribuiscano all'integrazione lungo la linea di vista. Tuttavia, questa approssimazione introduce errori significativi, in particolare su scale angolari grandi (bassi $\ell$). Per sfruttare appieno il potere di vincolo delle future indagini e controllare i sistemi, sono essenziali calcoli non-Limber accurati. Tuttavia, eseguire questi integrali 3D completi è computazionalmente proibitivo per i metodi di inferenza dei parametri come il Markov Chain Monte Carlo (MCMC), che richiedono milioni di valutazioni. Ad esempio, analisi recenti che utilizzavano l'approssimazione di Limber hanno richiesto 0,5 milioni di ore/CPU per convergere.

Metodologia
Gli autori presentano Blast.jl, un nuovo algoritmo progettato per calcolare efficientemente gli spettri di potenza angolari non-Limber completi senza assumere un tasso di crescita dipendente dalla scala. La metodologia centrale si basa sulla decomposizione dello spettro di potenza della materia utilizzando i polinomi di Chebyshev.

1. Decomposizione di Chebyshev: L'algoritmo approssima lo spettro di potenza della materia liscio $P(k, \chi_1, \chi_2)$ sull'intervallo $[k_{min}, k_{max}]$ utilizzando un'espansione in serie di Chebyshev. Ciò consente di separare l'integrale in componenti geometriche e componenti cosmologiche.
2. Pre-computazione dei termini geometrici: L'integrale viene ristrutturato in modo che la parte più costosa dal punto di vista computazionale — l'integrazione del prodotto di due funzioni di Bessel sferiche e dei polinomi di Chebyshev — diventi indipendente dai parametri cosmologici. Questi termini ($\tilde{T}^{AB}_{n; \ell}$) vengono calcolati una sola volta e memorizzati.
3. Valutazione efficiente: Per ogni nuovo modello cosmologico, l'algoritmo deve solo ricalcolare i coefficienti di Chebyshev dello spettro di potenza, un processo computazionalmente poco oneroso. Gli altri integrali esterni sulla distanza comovente ($\chi$) e sul rapporto $R = \chi_2/\chi_1$ vengono eseguiti utilizzando regole di quadratura standard (regola di Simpson per $\chi$ e quadratura di Clenshaw-Curtis per $R$).
4. Trasformazione di coordinate: Viene impiegata una trasformazione di variabili da $(\chi_1, \chi_2)$ a $(\chi, R)$ per campionare meglio la regione in cui $\chi_1 \approx \chi_2$, che contribuisce più significativamente all'integrale.
5. Correzione non lineare: Il metodo tratta separatamente gli spettri di potenza lineari e non lineari. Il contributo non lineare, significativo solo su piccole scale dove l'approssimazione di Limber è valida, viene corretto utilizzando un approccio basato su Limber, mentre l'integrale non-Limber completo viene applicato alla componente lineare.

Contributi Chiave

• Innovazione Algoritmica: Blast.jl introduce un approccio basato sulla decomposizione utilizzando i polinomi di Chebyshev che evita i vincoli di spaziatura della griglia logaritmica tipici dei metodi basati su FFTLog (come FKEM e matter) e la risoluzione di equazioni differenziali iterative del metodo Levin.
• Strategia di Pre-computazione: Isolando gli integrali geometrici indipendenti dalla cosmologia, l'algoritmo riduce drasticamente il tempo necessario per le valutazioni ripetute durante l'inferenza dei parametri.
• Flessibilità: A differenza di alcuni metodi concorrenti, Blast.jl non assume un tasso di crescita indipendente dalla scala, rendendolo applicabile a cosmologie con neutrini massicci o gravità modificata.
• Open Source: Il codice è pubblicamente disponibile su GitHub, accompagnato da un repository che ne dettaglia l'utilizzo.

Risultati
L'algoritmo è stato valutato rispetto alla sfida "N5K" (Non-local No-Nonsense Non-Limber Numerical Knock-out), organizzata dalla LSST Dark Energy Science Collaboration per testare i metodi di integrazione non-Limber. I test sono stati condotti su un nodo di calcolo ad alte prestazioni (Narval) con 64 thread, confrontando Blast.jl con i vincitori della sfida: FKEM (CosmoLike), matter e Levin.

• Velocità: Al requisito di accuratezza fiduciale ($\Delta\chi^2 < 0,2$ per $\ell < 200$), Blast.jl ha raggiunto un tempo di esecuzione di 0,00512 secondi utilizzando 64 core. Questo è circa 10 volte più veloce di FKEM (il precedente metodo più veloce, 0,05871 s) e circa 15 volte più veloce quando ottimizzato a 32 core (0,00360 s).
• Scalabilità: Blast.jl dimostra un'eccellente scalabilità rispetto al numero di bin di redshift tomografici. Mentre altri metodi mostrano un aumento del tempo di esecuzione con il numero di bin, il tempo di esecuzione di Blast.jl rimane effettivamente indipendente dal numero di bin (da 2 a 15).
• Accuratezza: Il metodo soddisfa i requisiti di accuratezza della sfida. Gli autori osservano che l'accuratezza può essere regolata modificando il numero di polinomi di Chebyshev e i punti di campionamento senza impattare significativamente sul tempo di esecuzione.
• Compromesso Precisione vs Velocità: Ridurre la precisione del numero in virgola mobile da Float64 a Float32 ha migliorato il tempo di esecuzione di un fattore 2 con un impatto trascurabile sull'accuratezza ($\Delta\chi^2$ è passato da 0,1054 a 0,1064).

Significatività
L'articolo sostiene che Blast.jl rappresenti l'algoritmo di integrazione non-Limber più veloce attualmente disponibile, superando i metodi allo stato dell'arte di oltre un ordine di grandezza nello scenario fiduciale. La sua principale significatività risiede nel rendere computazionalmente fattibili le analisi cosmologiche utilizzando le statistiche 3 × 2 pt per le indagini di prossima generazione, senza affidarsi all'errore dell'approssimazione di Limber.

Gli autori sottolineano che questa velocità è fondamentale per abilitare l'inferenza dei parametri basata su MCMC con i massicci dataset attesi dalle future indagini. Inoltre, il design dell'algoritmo permette estensioni future, come l'inclusione di distorsioni nello spazio di redshift, bias di magnificazione e correzioni relativistiche, nonché l'integrazione del lensing della CMB. Gli autori evidenziano anche i piani per incorporare la differenziazione automatica, che consentirebbe l'uso di metodi di inferenza basati sul gradiente, accelerando ulteriormente le pipeline di analisi cosmologica.