[论文解读] 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总能量最小化问题。
• 将全局能量最小化分解为N个独立子问题，每个子问题沿单个轨道方向最小化能量。
• 使用Stiefel流形上的梯度型步骤近似求解每个轨道子问题，避免完整的特征值求解。
• 采用两级并行：独立子问题并行求解，且每个子问题在处理器间并行求解。
• 使用不精确线搜索和近似正交化以保持效率，避免在每次迭代中进行完整正交化。
• 引入一种带参数 $ n_{org} $ 的两级并行版本，以控制正交化频率，提升并行效率。

实验结果

研究问题

• RQ1 基于单轨道最小化的并行轨道更新策略，是否能在大规模电子结构问题上实现比传统SCF方法更好的可扩展性？
• RQ2 在大系统上，所提方法与标准优化和SCF方法相比，在收敛速度和精度方面表现如何？
• RQ3 在现代超算上，该轨道更新方法在含数千个电子的系统中可达到的并行效率是多少？
• RQ4 对轨道子问题的不精确求解如何影响整体收敛性和能量精度？
• RQ5 当应用于小带隙或复杂电子结构体系时，该方法是否能保持鲁棒性和可靠性？

主要发现

• 所提出的并行轨道更新优化方法在大系统上实现了超过65%的并行效率，其中在丙氨酸链（$N=132$）和 $C_{120}$（$N=240$）系统中观察到高达82%的效率。
• 对于含3116个电子和180万个网格点的 $C_{1419}H_{556}$ 系统，该方法在155次迭代内收敛至能量 $-8.43085432 \times 10^3$ Hartree，残差为 $4.43 \times 10^{-5}$，实现了61%的并行时间减少。
• 该方法在多种体系中表现出稳健的收敛性，包括苯、阿司匹林、$C_{60}$、$C_{120}$、2JMO、FAS2和大碳链，其能量值与Opt-Z3W的参考结果一致。
• 能量降低曲线（图4和图5）证实了收敛性，尽管由于正交化延迟导致轻微振荡，但对最终精度无影响。
• 使用 $n_{org}=2$ 通过降低正交化频率提高了并行效率，$C_{120}$ 系统在表3中实现了82%的并行时间利用率。
• 该方法具有可扩展性和高效性：对于FAS2系统（$N=1293$），总时间从Opt-Z3W的31,647秒减少至Opt-Par的26,709秒，其中73%的时间用于并行轨道更新。