[論文レビュー] High-productivity, high-performance workflow for virus-scale electrostatic simulations with Bempp-Exafmm

Tingyu Wang, Christopher D. Cooper+2 more|arXiv (Cornell University)|Mar 1, 2021

Bacteriophages and microbial interactions参考文献 26被引用数 5

ひとこと要約

本論文は、Bempp-Exafmmを用いて、境界要素法（Bempp）のPythonベースの境界要素法と、核独立型高速多重極法（kifmm）であるExafmmを組み合わせることで、ウイルススケールの電気的シミュレーションのための生産性が高く、高性能なワークフローを提示している。このフレームワークにより、O(N)スケーリングを達成するPoissin-BoltzmannソルバーのインタラクティブでJupyterベースのプロトタイピングが可能となり、1000万個の境界要素を持つジカウイルスのソルバビリティ自由エネルギーの計算に1ノードで80分を要した。

ABSTRACT

Biomolecular electrostatics is key in protein function and the chemical processes affecting it. Implicit-solvent models via the Poisson-Boltzmann (PB) equation provide insights with less computational cost than atomistic models, making large-system studies -- at the scale of viruses -- accessible to more researchers. Here we present a high-productivity and high-performance linear PB solver based on Exafmm, a fast multipole method library, and Bempp, a Galerkin boundary element method package. The workflow integrates an easy-to-use Python interface with optimized computational kernels, and can be run interactively via Jupyter notebooks, for faster prototyping. Our results show the capability of the software, confirm code correctness, and assess performance with between 8,000 and 2 million elements. Showcasing the power of this interactive computing platform, we study the conditioning of two variants of the boundary integral formulation with just a few lines of code. Mesh-refinement studies confirm convergence as $1/N$, for $N$ boundary elements, and a comparison with results from the trusted APBS code using various proteins shows agreement. Our binding energy calculations using 9 various complexes align with the results from using five other grid-based PB solvers. Performance results include timings, breakdowns, and computational complexity. Exafmm offers evaluation speeds of just a few seconds for tens of millions of points, and $\mathcal{O}(N)$ scaling. The trend observed in our performance comparison with APBS demonstrates the advantage of Bempp-Exafmm in applications involving larger structures or requiring higher accuracy. Computing the solvation free energy of a Zika virus, represented by 1.6 million atoms and 10 million boundary elements, took 80-min runtime on a single compute node (dual 20-core).

研究の動機と目的

• リーダーシップクラスの計算施設を超えた大規模なバイオ分子電気的シミュレーションのためのアクセス可能で生産性の高いプラットフォームの開発。
• 高レベルのPythonコードを用いた線形ポテンシャル・ボルツマン方程式の境界積分形式の迅速なプロトタイピングの実現。
• BemppとExafmmの統合により、1000万個の境界要素に達するウイルススケールのシステムに対する高性能な実装。
• APBSやグリッドベースの手法といった既存のソルバーと比較して、正確性とパフォーマンスの妥当性の検証。
• 標準的なHPCハードウェアを用いて、完全なウイルスの高精度なインタラクティブ電気的シミュレーションの実現可能性の提示。

提案手法

• ワークフローは、Pythonベースのガレルキン境界要素法ライブラリBemppと、高性能な核独立型高速多重極法（kifmm）ライブラリExafmm-tを組み合わせている。
• 線形ポテンシャル・ボルツマン方程式の境界積分方程式は、Bemppの高レベルなPythonインターフェース内でのガレルキン離散化により定式化されている。
• Exafmm-tは、境界要素の階層的クラスタリングと遠方相互作用の低ランク近似を用いて、行列-ベクトル積の高速化を実現している。
• アルゴリズムは、粒子から多重極（P2M）、多重極から多重極（M2M）、多重極から局所（M2L）、局所から局所（L2L）、局所から粒子（L2P）、粒子から粒子（P2P）への変換を、二重木走査で実行している。
• 直接法とJuffer型境界積分形式をサポートしており、形式の変更はわずかなコード変更で可能である。
• すべてのワークフローは、再現性とインタラクティブ利用を目的としたJupyterノートブックにパckagedされており、Dockerによるポータビリティをサポートしている。

実験結果

リサーチクエスチョン

• RQ1 生産性が高く、インタラクティブでPythonベースのワークフローが、ウイルススケールの電気的シミュレーションにおいて高性能を達成できるか？
• RQ2 Bempp-Exafmmフレームワークは、大規模なバイオ分子系においてAPBSなどの既存ソルバーと比較して、正確性とパフォーマンスで優れているか？
• RQ3 Bempp-Exafmmソルバーの計算複雑性とスケーリング特性は、最大200万個の境界要素を持つシステムに対してどのように変化するか？
• RQ4 このフレームワークは、ジカウイルスのような大規模ウイルスの溶媒自由エネルギーを効率的に計算できるか？
• RQ5 最小限のコード変更で、異なる境界積分形式の条件付けが実際の計算でどのように異なるか？

主な発見

• Bempp-Exafmmワークフローは、Exafmm-tを用いてO(N)の計算複雑性を達成し、数千万点の評価が数秒で可能となった。
• メッシュの細分化に関する研究により、境界要素数Nに対して1/Nの収束率が確認され、ソルバーの数値的正確性が裏付けられた。
• 9つのタンパク質複合体の結合エネルギー計算では、他の5つのグリッドベースPBソルバーとの結果が強く一致した。
• 1.600万原子と1000万境界要素を持つジカウイルスモデルの溶媒自由エネルギーは、1つのデュアル20コア計算ノードで80分で正しく計算された。
• APBSとのパフォーマンス比較では、より大きなシステムやより高い精度が求められる応用において、Bempp-Exafmmの優位性が示された。これは、優れたスケーリング性能と低いメモリ使用量のおかげである。
• インタラクティブなJupyterノートブックインターフェースにより、数行のコードで異なる境界積分形式の迅速なプロトタイピングと比較が可能になった。