[論文レビュー] A Parallel Orbital-updating Based Optimization Method for Electronic Structure Calculations

Xiaoying Dai, Zhuang Liu+2 more|arXiv (Cornell University)|Oct 25, 2015

Advanced Chemical Physics Studies参考文献 3被引用数 3

ひとこと要約

本稿では、Kohn-Sham密度汎関数理論に基づく大規模な電子構造計算のための並列軌道更新最適化手法を提案する。各軌道方向に総エネルギー汎関数を独立して最小化し、近似的な正規直交化を適用することで、二重レベルの並列化を実現し、スーパーコンピュータ上で65%を超える並列効率を示し、高い効率性とスケーラビリティを達成した。

ABSTRACT

In this paper, we propose a parallel optimization method for electronic structure calculations based on a single orbital-updating approximation. It is shown by our numerical experiments that the method is efficient and reliable for atomic and molecular systems of large scale over supercomputers.

研究の動機と目的

• スーパーコンピュータ上で大規模系の効率的かつスケーラブルな電子構造計算を実現すること。
• 特に小ギャップを示す系において収束性が著しく悪いことが問題となる伝統的な自己無撞着場（SCF）手法の限界を克服すること。
• Stiefel多様体上での直接エネルギー最小化に基づく、スーパーコンピュータに適した最適化アルゴリズムを開発すること。
• グローバル最適化を独立した軌道部分問題に分解することで、二重レベルの並列化を実現すること。
• 標準的なSCF法や勾配型手法と比較して、収束の安定性と計算効率を向上させること。

提案手法

• 軌道の正規直交性を制約として、Stiefel多様体上でのKohn-Sham総エネルギー最小化問題を定式化する。
• グローバルエネルギー最小化問題を、各々が1つの軌道方向に沿ったエネルギー最小化を担うN個の独立部分問題に分解する。
• Stiefel多様体上での勾配型ステップを用いて各部分問題を近似的に解き、全固有値問題の計算を回避する。
• 二重レベルの並列化を実装する：独立した部分問題を並列に処理し、各部分問題をプロセッサ間で並列に解く。
• 不正確なラインサーチと近似的な正規直交化を用いて効率性を維持し、各反復で完全な直交化を実行しない。
• 直交化頻度を制御するパrameter $ n_{org} $ を有する二重レベル並列版を導入し、並列効率を向上させる。

実験結果

リサーチクエスチョン

• RQ1 単一軌道最小化に基づく並列軌道更新戦略は、大規模な電子構造問題において、従来のSCF手法よりも優れたスケーラビリティを達成できるか？
• RQ2 提案手法は、大規模系において標準的な最適化手法やSCF手法と比較して、収束速度と精度の点で優れているか？
• RQ3 現代のスーパーコンピュータ上で、数千電子を有する系に対して、軌道更新手法の達成可能な並列効率はどの程度か？
• RQ4 軌道部分問題の不正確な解法が、全体の収束性およびエネルギーの正確性にどのように影響を与えるか？
• RQ5 小ギャップや複雑な電子構造を示す系に適用した場合、本手法は収束性と信頼性を維持できるか？

主な発見

• 提案された並列軌道更新最適化手法は、大規模系において65%を超える並列効率を達成した。アラニン鎖（$N=132$）および $C_{120}$（$N=240$）系では最大82%の並列効率を記録した。
• $C_{1419}H_{556}$ 系（3116電子、180万グリッド点）では、155反復でエネルギー $-8.43085432 \times 10^3$ ハートリーに収束し、残差は $4.43 \times 10^{-5}$ であった。並列時間は61%短縮された。
• ベンゼン、アスピリン、$C_{60}$、$C_{120}$、2JMO、FAS2、および大規模な炭素鎖など多様な系において、収束性が強く、Opt-Z3Wによる基準結果と一致するエネルギー値を達成した。
• エネルギー低下曲線（図4および図5）は収束を確認しており、直交化の遅延によるわずかな振動はあったが、最終的な正確性に影響を及ぼさなかった。
• $n_{org}=2$ を用いることで、直交化頻度を低減し、並列効率が向上した。表3では、$C_{120}$ 系で82%の並列時間使用率を達成した。
• 本手法はスケーラブルかつ効率的である：FAS2系（$N=1293$）では、全時間はOpt-Z3Wの31,647秒からOpt-Parの26,709秒に短縮され、並列軌道更新に73%の時間が費やされた。