[论文解读] A conjugate gradient method for electronic structure calculations

Xiaoying Dai, Zhuang Liu+2 more|arXiv (Cornell University)|Jan 28, 2016

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

一句话总结

本论文提出了一种用于密度泛函理论中电子结构计算的新型共轭梯度（CG）方法，采用基于海塞矩阵的步长策略，并结合三种正交化技术（WY、QR、PD）以维持轨道正交性。采用QR正交化与海塞矩阵步长的CG方法在收敛速度和稳定性方面优于现有方法（如OptM-QR），在达到相近精度的同时，所需迭代次数更少，计算时间更短。

ABSTRACT

In this paper, we study a conjugate gradient method for electronic structure calculations. We propose a Hessian based step size strategy, which together with three orthogonality approaches yields three algorithms for computing the ground state energy of atomic and molecular systems. Under some mild assumptions, we prove that our algorithms converge locally. It is shown by our numerical experiments that the conjugate gradient method is efficient.

研究动机与目标

• 开发一种比自洽场（SCF）和梯度型方法更高效、更稳定的Kohn-Sham密度泛函理论（DFT）计算替代方法。
• 解决SCF迭代在小带隙大体系中收敛性差和不稳定的难题。
• 通过引入基于海塞矩阵的步长策略以及利用WY、QR和极分解（PD）方法实现鲁棒的正交性保持，改进现有共轭梯度方法。
• 证明所提出的CG方法在温和假设下可实现局部收敛，并在数值实验中优于最先进方法（如OptM-QR）。

提出的方法

• 提出一种在正交性约束下最小化Kohn-Sham总能量泛函的共轭梯度方法，采用基于能量泛函局部二阶泰勒展开的海塞矩阵推导出的步长策略。
• 采用三种正交化策略——WY、QR和极分解（PD）——以在迭代过程中保持Kohn-Sham轨道的正交性。
• 提出一种近似海塞矩阵（17），忽略库仑项和交换相关项，实证表明这些项对海塞矩阵贡献极小，从而在保持精度的同时显著降低计算成本。
• 采用非回溯的、基于海塞矩阵的初始步长，避免了昂贵的精确线搜索以及Barzilai-Borwein（BB）步长的非单调行为。
• 在能量下降性质下进行收敛性分析，证明所有三种算法（基于WY、QR、PD的CG）均具有局部收敛性。
• 采用改进的共轭方向更新策略，结合当前梯度与前一搜索方向，确保下降性和共轭性。

实验结果

研究问题

• RQ1 基于海塞矩阵的步长策略的共轭梯度方法是否能在电子结构计算中实现比现有梯度型方法更快的收敛速度和更高的稳定性？
• RQ2 不同的正交性保持技术（WY、QR、PD）如何影响Kohn-Sham DFT中共轭梯度算法的性能与鲁棒性？
• RQ3 忽略库仑项和交换相关项的近似海塞矩阵（17）是否在步长确定中足够准确，同时显著降低计算成本？
• RQ4 基于海塞矩阵的步长策略是否可消除对回溯线搜索的需求？与BB步长相比，其在收敛性与稳定性方面表现如何？
• RQ5 所提出的CG方法在效率与精度方面与最先进的OptM-QR算法（采用BB步长）相比如何？

主要发现

• 基于QR的共轭梯度算法结合海塞矩阵步长（CG-QR-H）在所有测试体系中表现最佳，所需迭代次数最少，计算时间最短。
• 对于苯（C6H6），CG-QR-H在119次迭代和7.26秒内收敛，而OptM-QR-BB需164次迭代和6.63秒，显示出更高的效率。
• 在大尺寸丙氨酸链（C33H11O11N11）中，CG-QR-H在1232次迭代和1204.82秒内收敛，而OptM-QR-BB需3190次迭代和2338.59秒，展现出显著的速度提升。
• 近似海塞矩阵（17）的收敛行为与精确海塞矩阵（16）几乎完全一致，但计算成本大幅降低，因此推荐使用。
• 基于海塞矩阵的步长避免了BB步长的非单调行为，从而实现更稳定的收敛，尤其在OptM-QR表现出不稳定的体系中效果更明显。
• 采用近似海塞矩阵（17）的算法在精度上与使用精确海塞矩阵（16）的算法相当，且库仑项与交换相关项对海塞矩阵的贡献可忽略，验证了该近似方法的有效性。