Towards Practical Field-Level Inference for Weak Lensing

技術要約：弱重力レンズ効果における実用的なフィールドレベル推論に向けて

問題提起
弱重力レンズ効果は、宇宙における投影された物質分布の統計的な尺度を提供する。しかし、構造形成が非線形領域に入ると（現在の観測に基づけば $z \sim 0.5$ 付近）、宇宙の密度場は高次の相関や形態学的特徴を発達させ、これらは従来の2点統計量（例：パワースペクトル）では十分に捉えることができない。ピーク数、ミンコフスキー汎関数、スキャッタリング変換など、様々な代替要約統計量が提案されてきたが、フィールドレベル推論（Field-Level Inference: FLI）は、観測された弱重力レンズ場から直接宇宙論パラメータを推定することにより、より一般的なアプローチを提供する。

物理的に動機付けられたフォワードモデルを用いる際、FLIの実用的な性能を従来の2点統計解析と比較することには、重要なギャップが存在する。これまでのFLI研究は、簡略化された統計モデル（例：対数正規場）や、特定のフレームワーク（例：ラグランジュ摂動論を用いたBORG）に依存しており、これらは現実的なシミュレーションの複雑さを完全には捉えきれていない可能性がある。さらに、一致したデータとシミュレーションを適用した場合に、暗黙的（シミュレーションベース）な推論手法と明示的（サンプリングベース）な推論手法が、一貫した宇宙論的制約をもたらすかどうかを検証する必要がある。

手法
著者らは、同一のフォワードモデリング・パイプラインを用いて、3つの推論アプローチの制御された比較を行っている。これらは、弱重力レンズマップを生成するために用いられる：

1. パワースペクトルに基づく推論： ビン分割された自己および相互パワースペクトルを要約統計量として使用する、シミュレーションベースの推論（SBI）アプローチ。
2. 暗黙的FLI： ニューラル密度推定器が、シミュレーションされたデータ場ペアから直接事後分布を学習し、全場を要約ベクトルへと圧縮するか、あるいは尤度比を学習するSBIアプローチ。
3. 明示的FLI： ハミルトニアンモンテカルロ法（HMC/NUTS）を用いたサンプリングベースのアプローチであり、宇宙論パラメータ（$\vec{\theta}$）と白色化された潜在的な初期条件（$\vec{u}$）の結合事後分布をサンプリングする。

フォワードモデル
本研究では、勾配ベースのサンプリングのための微分可能性を確保するため、JAXで実装された2つの異なるフォワードモデルを利用している：

• ラグランジュ摂動論（LPT）： 具体的にはゼルドビッチ近似（1LPT）であり、計算効率は高いが線形領域に限定される。
• 粒子メッシュ（PM）N体進化： Matter fieldの進化をより正確に捉えるために、BullFrogインテグレータ（LPTに基づいたリープフロッグ・スキーム）を利用したPM N体モデル。このモデルは、LPTよりも正確に非線形構造の成長を捉える。

シミュレーションは、ライトコーンがシミュレーション体積内に留まることを保証するため、2つの狭いソース赤方偏移ビン（$z=0.7$ および $z=0.8$）に対して収束マップを生成する。解析では、シミュレーショングリッドに固有の有限解像度のアーティファクトを軽減するため、保守的なスケールカット（$\ell_{max} = 400$ および $\ell_{max} = 1000$）を採用している。

主な貢献

• FLIパラダイムの直接比較： 本論文は、同一の基礎となる物理学（LPTおよびPM）を用いて、暗黙的および明示的なFLIをパワースペクトルベースの推論と比較する制御された実験を確立している。
• 明示的FLIの検証： 著者らは、高次元の結合事後分布（宇宙論 ＋ 潜在場）を、LPTおよびより現実的なPMモデルの両方に対して、明示的FLIが正常にサンプリングできることを示している。また、暗黙的手法と明示的手法が極めて一貫した事後分布を与えることを確認し、回収されたフィールドレベルの制約の妥当性を検証している。
• 情報利得の定量化： マップレベルの解析が、2点統計量と比較して提供する宇宙論的情報の利得を定量化している。
• ベンチマーク： 著者らは、NVIDIA A100 GPU上でのフォワードモデルの計算ベンチマークを提供し、解像度、計算コスト、および推論に必要なステップ数のトレードオフを明らかにしている。

結果

• 一貫性： 暗黙的および明示的なFLIアプローチは、宇宙論パラメータ（$\Omega_m$ および $S_{8}^{lin}$）に対して一貫した事後分布を生成する。カバレッジテストにより、暗黙的解析の較正が確認されている。
• 情報利得： パワースペクトルベースの解析と比較して、FLIは宇宙論的情報の顕著な利得を示す。この利得は、PMベースのフォワードモデルを使用し、かつ小さなスケール（$\ell_{max}=1000$ まで）を解析に含めた場合に特に顕著である。
• モデル依存性： 結果は、情報利得がフォワードモデルの忠実度に敏感であることを示している。非線形構造を捉えるPMモデルは、LPTモデルよりもタイトな制約を与え、これは非線形効果が支配的な小スケールにおいて特に顕著である。
• 解像度の限界： 著者らは、PMモデルが $\ell \simeq 500$ までは非線形の解析的予測と一致するものの、シミュレーションの有限解像度のために小スケールではバイアスが低くなることを指摘している。したがって、堅牢な推論を確実にするために、本研究ではスケールカットを課している。

意義と主張
本論文は、フィールドレベル推論が、弱重力レンズ観測から宇宙論的情報を抽出するための実行可能かつ強力な手法であり、従来の2点統計量よりも大幅な改善をもたらすと主張している。暗黙的手法と明示的手法の合致は、フィールドレベルの制約の堅牢性に対する信頼を与えるものである。

しかし、著者らは観測データへの即時適用については慎重な姿勢を崩していない。彼らは、「PMベースの明示的FLIを観測データセットに適用するためには、残された課題に対処しなければならない」と明言している。これらの課題には、高解像度フォワードモデルの計算コストや、シミュレーションと実際のデータとの比較において生じる系統誤差およびモデルの不一致の処理が含まれる。本研究は、十分に正確なフォワードモデルがあれば、FLIが弱重力レンズマップに含まれる非ガウス的な情報を解き放つことができることを示す概念実証として機能している。