[論文レビュー] Rational Tracer: a Tool for Faster Rational Function Reconstruction

Vitaly Magerya|arXiv (Cornell University)|Nov 7, 2022

Parallel Computing and Optimization Techniques被引用数 5

ひとこと要約

Rational Tracer (Ratracer) は、式のトレースを評価から分離することで、高エネルギー物理学における有理関数再構成を高速化し、最適化された C++ トレースを用いて IBP 減少におけるオーバーヘッドを低減する。Kira + FireFly より最大 5 倍の高速化と、データ構造管理を排除することでプローブ時間に 20 倍の高速化を達成し、フェルミオン積分係数の効率的かつ並列化可能な再構成を可能にする。

ABSTRACT

Rational Tracer (Ratracer) is a tool to simplify complicated arithmetic expressions using modular arithmetics and rational function reconstruction, with the main idea of separating the construction of expressions (via tracing, i.e. recording the list of operations) and their subsequent evaluation during rational reconstruction. Ratracer can simplify arithmetic expressions (provided as text files), solutions of linear equation systems (specifically targeting Integration-by-Parts (IBP) relations between Feynman integrals), and even more generally: arbitrary sequences of rational operations, defined in C++ using the provided library ratracer.h. Any of these can also be automatically expanded into series prior to reconstruction. This paper describes the usage of Ratracer specifically focusing on IBP reduction, and demonstrates its performance benefits by comparing with Kira+FireFly and Fire6. Specifically, Ratracer achieves a typical ~10x probe time and ~5x overall time speedup over Kira+FireFly, and even higher if only a few terms in $\varepsilon$ need to be reconstructed.

研究の動機と目的

• Kira や FireFly のような従来のソルバーにおけるデータ構造管理が原因で生じる IBP 減少における性能ボトルネックを解消すること。
• フェルミオン積分の有理関数再構成における中間式の膨張と計算オーバーヘッドを低減すること。
• トレース生成と評価を分離することで、モジュラ評価からの係数再構成をより高速かつスケーラブルに可能にすること。
• 特に二ループおよび三ループレベルにおける高多重性・マルチスケールのフェルミオン積分計算における性能向上を図ること。
• トレースベースの再構成を高エネルギー物理学ワークフローに統合できるモジュラーかつ拡張可能な C++ ライブラリおよびコマンドラインツールを提供すること。

提案手法

• Ratracer はガウスの消去法の過程で算術演算をトレースとして記録し、複雑なデータ構造を維持するのではなく、モジュラ演算のシーケンスとして保存する。
• 出力位置がマークされたカスタムバイナリフォーマットを用いてトレースを保存することで、効率的なメモリレイアウトと最適化を可能にする。
• トレースは不要コードの削除、定数畳み込み、共通部分式の削除、メモリレイアウトの再編成により最適化される。
• 再構成は、変数値を素数を法とする整数または有理数に設定してトレースを評価し、その後に有理数再構成を実行することで行われる。
• システムは部分的出力再構成と、出力のサブセットを複数のマシンに分散させることによる並列化をサポートする。
• Kira や FireFly のような既存の IBP ソルバーと統合され、それらの再構成段階をより高速なトレースベースの評価に置き換える。

実験結果

リサーチクエスチョン

• RQ1 トレース生成と評価を分離することで、IBP 減少における有理関数再構成の計算オーバーヘッドを著しく低減できるか？
• RQ2 トレース最適化とメモリレイアウトの改善は、高エネルギー物理学計算におけるモジュラ算術評価の速度にどの程度の向上をもたらすか？
• RQ3 Ratracer のパフォーマンスは、Kira + FireFly および Kira + Fermat と比較して、合計実行時間とプローブ効率の観点でどの程度優れているか？
• RQ4 ε のべき乗で係数を展開することの再構成速度とメモリ使用量に与える影響は何か？
• RQ5 プローブ時間が著しく短縮された場合、FireFly における有理数再構成のオーバーヘッドがパフォーマンスボトルネックになる可能性はあるか？

主な発見

• Ratracer は Kira より最大 20 倍のプローブ時間の高速化を達成し、個々のモジュラ評価のコストを顕著に低減している。
• Kira + FireFly と比較して、困難な IBP 減少の合計実行時間は最大 5 倍の高速化が達成され、特に質量付きの五点二ループトポロジーのような難易度の高いケースで顕著である。
• 一部のベンチマーク、例えば二質量付きの三ループダイヤモンドでは、FireFly の効率がプローブ頻度の高さにより 3% 未満にまで低下しており、再構成のオーバーヘッドが支配的であることが示唆されている。
• 係数を O(ε⁰) に展開すると 2.5–5.5 倍の高速化が得られ、O(ε¹) に展開すると 1.4–4 倍の高速化が得られ、高次の展開の利点が明らかになっている。
• Kira + Fermat はほとんどのケースで Kira + FireFly よりも高速であるが、非平面型ボックスでは Fermat のシングルプロセスボトルネックが遅延を引き起こしている。
• 出力サブセットの独立した再構成を可能にすることで、効率的な並列化が実現され、不要コードの削除により出力ごとの評価コストが低減されている。