[論文レビュー] The FHI-aims Code: All-electron, ab initio materials simulations towards the exascale

Volker Blüm, Mariana Rossi+2 more|arXiv (Cornell University)|Aug 25, 2022

Machine Learning in Materials Science被引用数 10

ひとこと要約

本論文では、数値原子中心オルビタル（NAOs）に基づく、エクサスケールコンピューティングを想定した高精度な全電子ab initio電子構造コードFHI-aimsを提示する。このコードは、コア電子の完全な取り扱いを可能とし、マルチスケールおよびAI駆動のモデリングワークフローをサポートしており、多様な物質系において高い並列効率と精度を達成する。

ABSTRACT

FHI-aims is a quantum mechanics software package based on numeric atom-centered orbitals (NAOs) with broad capabilities for all-electron electronic-structure calculations and ab initio molecular dynamics. It also connects to workflows for multi-scale and artificial intelligence modeling.

研究の動機と目的

• エクサスケールHPC環境に適したスケーラブルで、全電子ab initio電子構造コードの開発。
• 数値原子中心オルビタル（NAOs）を用いて、コア電子を完全に取り扱う高精度な複雑な物質のシミュレーションの実現。
• 物質設計のためのマルチスケールおよび人工知能駆動のモデリングワークフローへのFHI-aimsの統合。
• 多様な物質および化学的環境において、高い並列効率と数値的安定性を確保すること。

提案手法

• FHI-aimsは、数値原子中心オルビタル（NAOs）を基底関数として採用し、全電子計算における体系的な基底関数の完全性と高精度を実現する。
• 局在化された基底関数を用いた線形スケーリングアプローチを採用しており、大規模並列アーキテクチャでの効率的計算を可能にする。
• コア電子および価電子を完全に取り扱う全電子Kohn-Sham密度汎関数理論（DFT）を実装する。
• 適応的時間ステップとエネルギー保存を備えたab initio分子動力学（AIMD）シミュレーションをサポートする。
• 標準化されたインターフェースを通じて、外部のマルチスケールモデリングおよび機械学習応用向けワークフローへの統合を実現する。
• 最適化されたメモリアクセスパターン、負荷バランス、通信を回避するアルゴリズムにより、エクサスケール環境への対応を達成する。

実験結果

リサーチクエスチョン

• RQ1 どのようにしてエクサスケールHPC環境において、全電子ab initioシミュレーションをスケーラブルかつ効率的に行えるか？
• RQ2 数値原子中心オルビタルを用いた大規模物質シミュレーションにおいて、FHI-aimsの性能と精度はいかほどか？
• RQ3 FHI-aimsはどのようにしてマルチスケールおよびAI駆動の物質モデリングパイプラインへのシームレスな統合を可能にするか？
• RQ4 多様な物質および化学的環境において、どの程度の並列効率と数値的安定性が達成できるか？
• RQ5 FHI-aimsはエクサスケールのハードウェアリソースにスケーリングしても、高い精度を維持できるか？

主な発見

• FHI-aimsは大規模HPCシステムにおいて高い並列効率を達成し、数万コアまでの強スケーリングを示した。
• 体系的な基底関数収束を伴う全電子精度を実現し、電子的・構造的・熱力学的性質の信頼性の高い予測が可能である。
• 数値的結果は、遷移金属酸化物や複雑な半導体を含む広範な物質系において、実験データおよびベンチマーク手法と良好に一致した。
• FHI-aimsのマルチスケールおよびAIワークフローへの統合により、物質の発見と物性予測が加速された。
• コードの設計により、エネルギー保存と長時間スケールにわたる堅牢性を備えた長時間のab initio分子動力学シミュレーションが可能となった。
• FHI-aimsはエクサスケールシステム向けに生産環境で利用可能であり、最適化されたメモリアクセスおよび通信パターンにより、現代のHPCアーキテクチャ間でのパフォーマンスポータビリティが保証された。