[論文レビュー] Krylov space solvers for shifted linear systems

Beat Jegerlehner|ArXiv.org|Dec 15, 1996

Matrix Theory and Algorithms参考文献 8被引用数 100

ひとこと要約

本論文は、CG、CR、BiCGstab、およびCG-Mのための、シフトされた多項式を用いた一元的なフレームワークを提示する。これにより、1つの線形系の行列ベクトル積のみを用いて、複数のシフトされた線形系 $(A + \sigma)x = b$ を同時に解くことが可能になる。主な貢献は、特にスタッガードフェルミオンにおいて、格子QCD計算の最適化を実現し、複数のクォーク質量を解く際に顕著な高速化を達成することにある。

ABSTRACT

We investigate the application of Krylov space methods to the solution of shifted linear systems of the form (A+σ) x - b = 0 for several values of σsimultaneously, using only as many matrix-vector operations as the solution of a single system requires. We find a suitable description of the problem, allowing us to understand known algorithms in a common framework and developing shifted methods basing on short recurrence methods, most notably the CG and the BiCGstab solvers. The convergence properties of these shifted solvers are well understood and the derivation of other shifted solvers is easily possible. The application of these methods to quark propagator calculations in quenched QCD using Wilson and Clover fermions is discussed and numerical examples in this framework are presented. With the shifted CG method an optimal algorithm for staggered fermions is available.

研究の動機と目的

• 複数のシフトされた系 $(A + \sigma)x = b$ を、最小限の計算コストで同時に解くことができる一元的なフレームワークを構築すること。
• クェンチドQCDにおけるクォーク伝播関数計算の計算ボトルネックを解消すること。ここで $\sigma$（クォーク質量）の複数の値を効率的に逆数化する必要がある。
• CG、BiCGstab、CRなどの短記憶Krylov法を、各質量値あたり2つの追加ベクトルで済むような、シフトされたバージョンに拡張すること。
• 特に丸め誤差や32ビット演算においても、数値的安定性と収束性を保証すること。
• ウィルソン、クローラー、スタッガードフェルミオンを用いた実際の格子QCDシミュレーションにおいて、実用的効率性とスケーラビリティを示すこと。

提案手法

• 行列 $A$ に対して生成される多項式 $P_n^\sigma(z + \sigma) = c_n^\sigma P_n(z)$ を用いて、$P_n^\sigma(A)$ が生成するベクトルが $P_n(A)$ のスケーリング版であるようにシフトされた多項式を定義し、$\sigma$ 値間で再利用可能となるようにする。
• Krylov部分空間の不変性 $\mathcal{K}_n(A, v_0) = \mathcal{K_n}(A + \sigma, v_0)$ を用いて、各 $\sigma$ に対して追加の行列ベクトル積を必要とせずに、有効な反復ベクトルを生成することを保証する。
• CG、CR、BiCGstab、CG-Mのシフト版を、アルゴリズムに多項式構造を伝搬させる再帰関係を導出することで構築する。
• 線形前処理（式 4.63 における $a=0$）を実装し、特に小さなクォーク質量において収束を安定化させ、停滞を避けるために条件 (4.62) を保証する。
• ウィルソン、クローラー、スタッガードフェルミオンを用いたクォーク伝播関数計算にこの手法を適用し、タドループ・イニシャライズドパラメータを用い、非局所源および点源を用いてテストする。
• 特に32ビット演算における丸め誤差の影響を受ける可能性のあるシフトされた残差の監視を通じて、収束性と安定性を検証する。

実験結果

リサーチクエスチョン

• RQ1 CG や BiCGstab などの標準的な短記憶解法から、共通の多項式フレームワークを用いて、シフトされたKrylov部分空間法を体系的に導出可能か？
• RQ2 複数のシフトされた系 $(A + \sigma)x = b$ を、最も遅い1つの系に必要な行列ベクトル積のみで解く方法は何か？
• RQ3 丸め誤差がシフトされた残差の収束に与える影響は何か？実装上、どのようにこれを軽減できるか？
• RQ4 非局所源を伴う格子QCDにおいて、質量数や質量の間隔の増加に伴うシフト解法のスケーリング特性は？
• RQ5 非常に小さなクォーク質量に対して、高次多項式前処理を効果的に適用できるか？

主な発見

• スタッガードフェルミオンでは、効率的な前処理が得られないため、シフトされたCG-MおよびCR-M法が最適であり、顕著な高速化を実現できる。
• メモリ制約が許容される場合、ウィルソンおよびクローラー・フェルミオンでは、BiCGstab-M法が最良の選択肢であり、安定した収束性と高い効率性を示す。
• 非局所源では、クォーク質量が近接している場合、改善要因が顕著に高くなる。$\sum_{i=1}^{n}N^{\rm cont}_{i} - 2N^{\rm zero\text{-}guess}_{n} \gg 0$ は、大きな利得を示している。
• 数値的テストでは、丸め誤差の影響により、一部のケースでシフトされた残差が $\approx 10^{-2}$ で停滞することが判明し、特に大きな格子サイズで顕著になるため、残差の監視が不可欠である。
• 線形前処理により、小さな質量における収束が安定化され、適切な実装のもとで32ビット演算でも実用的である。
• このフレームワークは、他のKrylov法への容易な拡張を可能にし、極めて小さな質量に対して高次前処理を効果的に適用可能である。