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

技術要約：Blast.jl – 3 × 2 pt 解析のための高速かつ効率的なアルゴリズム

問題提起
次世代のフォトメトリックおよび分光サーベイ（Euclid、LSST、Roman、DESIなど）の到来は、大規模構造（LSS）の高精度な測定を約束している。これらのサーベイは、銀河のクラスタリング、弱重力レンズによるシェア・シェア相関、および銀河・銀河重力レンズの相互相関を組み合わせた「3 × 2 pt」統計に大きく依存している。これらの統計の理論的モデリングには、3次元の物質パワースペクトルと高度に振動する球面ベッセル関数の積を積分することにより、角パワースペクトル（$C_\ell$）を計算する必要がある。

計算コストを削減するための標準的なアプローチはリムベル近似（Limber approximation）であり、これは異なる時代におけるモードが視線方向の積分に寄与しないと仮定することで、3次元積分を1次元積分へと簡略化するものである。しかし、この近似は、特に大きな角度スケール（低 $\ell$）において重大な誤差を生じさせる。将来のサーベイの制約力を最大限に活用し、系統誤差を制御するためには、正確な非リムベル計算が不可欠である。しかし、パラメータ推論のためのマルコフ連鎖モンテカルロ法（MCMC）のような手法では、数百万回の評価が必要となるため、これらの完全な3次元積分を実行することは計算量的に極めて困難である。例えば、最近のリムベル近似を用いた解析では、収束までに0.5百万CPU時間がかかった。

手法
著者らは、スケール依存の成長率を仮定することなく、完全な非リムベル角パワースペクトルを効率的に計算するために設計された新しいアルゴリズム、Blast.jlを提示する。コアとなる手法は、チェビシェフ多項式を用いた物質パワースペクトルの分解に基づいている。

1. チェビシェフ分解: アルゴリズムは、区間 $[k_{min}, k_{max}]$ 上の滑らかな物質パワースペクトル $P(k, \chi_1, \chi_2)$ をチェビシェフ級数展開を用いて近似する。これにより、積分を幾何学的成分と宇宙論的成分に分離することが可能になる。
2. 幾何学的項の事前計算: 積分は、最も計算負荷の高い部分、すなわち2つの球面ベッセル関数とチェビシェフ多項式の積の積分が、宇宙論パラメータに依存しないようになるように再構成される。これらの項（$\tilde{T}^{AB}_{n; \ell}$）は一度だけ計算され、保存される。
3. 効率的な評価: 新しい宇宙論モデルに対しては、パワースペクトルのチェビシェフ係数を再計算するだけでよいが、これは計算コストが極めて低い。残りの共動距離（$\chi$）および比率 $R = \chi_2/\chi_1$ に関する外側の積分は、標準的な求積則（$\chi$ に対してはシンプソンの則、$R$ に対してはクランシャ・カーティス求積法）を用いて実行される。
4. 座標変換: $(\chi_1, \chi_2)$ から $(\chi, R)$ への変数変換を採用することで、積分への寄与が最も大きい領域（$\chi_1 \approx \chi_2$ の領域）をより適切にサンプリングする。
5. 非線形補正: 本手法は、線形パワースペクトルと非線形パワースペクトルを個別に扱う。リムベル近似が有効な小スケールにおいて重要となる非線形寄与については、リムベルベースのアプローチを用いて補正を行う一方、線形成分には完全な非リムベル積分を適用する。

主な貢献

• アルゴリズムの革新性: Blast.jlは、FFTLogベースの手法（FKEMやmatterなど）のような対数グリッド間隔の制約を回避し、またレヴィン法（Levin method）のような反復的な微分方程式の求解を必要としない、チェビシェフ多項式を用いた分解ベースのアプローチを導入している。
• 事前計算戦略: 宇宙論に依存しない幾何学的積分を分離することで、パラメータ推論中の繰り返しの評価に必要な時間を劇的に削減している。
• 柔軟性: 一部の競合手法とは異なり、Blast.jlはスケール不変な成長率を仮定しないため、大質量ニュートリノや修正重力を含む宇宙論にも適用可能である。
• オープンソース: コードはGitHubで公開されており、その使用方法を詳述したリポジトリも用意されている。

結果
アルゴリズムの評価は、非リムベル積分手法のベンチマークとしてLSSTダークエネルギー科学コラボレーションによって組織された「N5K」（Non-local No-Nonsense Non-Limber Numerical Knock-out）チャレンジに対して行われた。テストは、64スレッドを備えた高性能コンピューティングノード（Narval）上で実施され、Blast.jlをこれまでの勝者であるFKEM (CosmoLike)、matter、およびLevinと比較した。

• 速度: 基準となる精度要件（$\ell < 200$ で $\Delta\chi^2 < 0.2$）において、Blast.jlは64コアを使用して0.00512秒の実行時間を達成した。これは、以前の最速手法であったFKEM（0.05871秒）よりも約10倍速く、32コアに最適化した場合には約15倍高速（0.00360秒）である。
• スケーリング: Blast.jlは、トマティック・レッドシフト・ビンの数に対する優れたスケーリングを示す。他の手法はビンの数が増えると実行時間が増加する傾向にあるが、Blast.jlの実行時間は（2から15の範囲で）実質的にビンの数に依存しない。
• 精度: 本手法は、チャレンジの精度要件を満たしている。著者らは、チェビシェフ多項式の数とサンプリング点を調整することで、実行時間に大きな影響を与えることなく精度を調整できると述べている。
• 精度と速度のトレードオフ: 浮動小数点精度をFloat64からFloat32に下げることで、精度への影響を無視できる範囲に留めつつ（$\Delta\chi^2$ は0.1054から0.1064へ増加）、実行時間を2倍に向上させた。

意義
本論文は、Blast.jlが現在利用可能な中で最も高速な非リムベル積分アルゴリズムであり、基準となるシナリオにおいて既存の最先端手法を1桁以上のオーダーで上回っていると主張している。その主な意義は、エラーの多いリムベル近似に頼ることなく、次世代サーベイにおける3 × 2 pt 統計を用いた宇宙論解析を計算量的に実行可能にすることにある。

著者らは、この速度が、将来のサーベイで予想される膨大なデータセットを用いたMCMCベースのパラメータ推論を可能にするために極めて重要であることを強調している。さらに、本アルゴリズムの設計は、赤方偏移空間歪み、増光バイアス、および相対論的補正の導入、さらにはCMBレンズ効果の統合といった将来的な拡張を可能にしている。また、勾配ベースの推論手法の使用を可能にする自動微分（automatic differentiation）の組み込みについても計画を進めていることを強調しており、これにより宇宙論解析パイプラインをさらに加速させることが期待される。