MadVfold: accelerating NLO event generation and reducing negative weights with SIMD vectorization and GPUs

粒子物理学という極めて重要な世界において、科学者たちは大型ハドロン衝突型加速器（LHC）を用いて陽子を衝突させ、初期宇宙の状態を再現しています。その破片を理解するために、彼らは衝突中に何が起こるべきかを予測するコンピュータ・シミュレーションに頼っています。しかし、これらの予測を極めて高い精度で計算することは、気の遠くなるような作業です。「次世代（next-to-leading-order）」と呼ばれる最も正確な手法は、非常に低速で計算コストがかさみます。これらのシミュレーションにおける大きな障害は、「負の重み（negative weights）」の出現です。これは、シミュレートされたイベントの一部が負の値を持つという数学的な特異現象です。これらの負の値が生み出すノイズを打ち消して明確な像を得るために、研究者は通常よりもはるかに膨大な数のイベントを生成しなければならず、膨大な計算能力と時間を消費することになります。

これを解決するために、物理学者は「フォールディング（folding）」と呼ばれる手法を開発しました。複雑な積分の値を単一の点でサンプリングすることで推定しようとする場面を想像してみてください。それはノイズが多く、不正確になり得ます。フォールディングは、単一のイベントの運動学的変数の異なるバリエーションに対して、同じ複雑な関数を複数回計算し、その結果を平均化することに似ています。このアプローチは、積分のより安定した推定値を提供し、統計的なノイズを平滑化して負の重みの必要性を軽減します。効果的ではあるものの、この手法自体は、コンピュータに単一のイベントに対して同じ複雑な計算を何度も実行させる必要があるため、非常に低速です。CERNのアンドレア・ヴァラッシ氏による新しいアプローチは、現代のコンピュータ・ハードウェアを利用して、これらの繰り返される計算を一つずつではなく、同時に実行することで、このプロセスを高速化することを目指しています。

ヴァラッシ氏の仕事は、「ベクトル化されたフォールディング（vectorized folding）」と呼ばれる手法を導入しています。これは、現代のグラフィックスカードや高度な中央演算処理装置（CPU）の並列処理能力を活用するものです。特定のイベントのバリエーションを一つずつ計算するのではなく、数十のバリエーションをグループ化します。そして、このバッチ全体をコンピュータのプロセッサに送り、プロセッサはそれらすべてを同時に計算します。これは、イベントを逐次的に処理するという従来のシミュレーションのやり方からの大きな転換です。この並列能力を活用するようにソフトウェアを再構成することで、彼は、広く使用されているMadgraph5_aMC@NLOシミュレーション・パッケージで動作するように設計された、MADVFOLDという新しいツールを生み出しました。

この再構成の結果は驚くべきものです。電子と陽電子の衝突によってボトムクォークが生成される過程を含むテストにおいて、新しい手法はシミュレートされたイベントの生成に必要な時間を劇的に短縮しました。現代のデータセンターで見られる高度なプロセッサを使用した際、ベクトル化されたフォールディング技術は、標準的な手法よりも計算を約6倍から9倍高速化しました。計算をグラフィックス・プロセッシング・ユニット（GPU）にオフロードした場合、その高速化はさらに顕著になり、元の手法の10倍近くに達しました。これらの利得は、基礎となる物理学を変更したのではなく、コンピュータが数学を実行する方法を変更し、フォールディングの重労働をより効率的に処理できるようにしたことによって達成されました。

この研究では、この並列アプローチをフォールディング技術を用いないイベント生成に適用できるかどうかも調査されました。たとえ、標準的なシミュレーションの高速化のみを目的とした「アンフォールド（unfolded）」のシナリオであっても、新しいソフトウェアは3倍の速度向上を実現しました。これは、フォールディングをサポートするためのアーキテクチャ上の変更が、シミュレーション・パイプライン全体をより効率的にするという、より幅広い利益をもたらすことを示唆しています。この研究は、研究者がコードの記述とテストに大規模言語モデルを多用するという斬新なプロセスを用いて開発され、この手法によってプロジェクトを構想から実用的なソフトウェアへとわずか約6週間で進めることができました。

目覚ましい高速化にもかかわらず、論文ではフォールディングは依然として計算コストの高い手法であると指摘しています。新しいハードウェア・アクセラレーションを用いても、フォールディングを伴うシミュレーションを実行することは、フォールディングなしで実行する場合よりも明らかに時間がかかります。しかし、フォールディングは負の重みを持つイベントの数を減少させ、その結果、統計的に有意な結果に達するために必要な総イベント数を減らすことができるため、このトレードオフはしばしば不可欠です。フォールディングのプロセス自体を大幅に高速化することで、この新しい手法は、今後の高輝度LHCプログラムを含む将来の実験において、この必要なトレードオフをより管理しやすいものにします。本研究は、ソフトウェアがさらなるテストと改良の準備ができているものの、これは今後予想される膨大なデータセットに対して、高精度な粒子物理学シミュレーションをより実現可能なものにするための重要な一歩であると結論付けています。