[论文解读] ElVibRot-MPI: parallel quantum dynamics with Smolyak algorithm for general molecular simulation

Ahai Chen, André Nauts+1 more|arXiv (Cornell University)|Nov 26, 2021

Advanced NMR Techniques and Applications被引用 5

一句话总结

ElVibRot-MPI 是一个并行量子动力学软件包，采用 Smolyak 稀疏网格算法，可对具有多达数十个自由度的分子体系实现高精度模拟。通过将计算工作量分布在多个处理器上，该软件包实现了优异的并行效率，将 10D 体系的模拟时间从直接乘积法的 65 小时以上缩短至 12 个处理器下的 10 分钟以内。

ABSTRACT

A parallelized quantum dynamics package using the Smolyak algorithm for general molecular simulation is introduced in this work. The program has no limitation of the Hamiltonian form and provides high flexibility on the simulation setup to adapt to different problems. Taking advantage of the Smolyak sparse grids formula, the simulation could be performed with high accuracy, and in the meantime, impressive parallel efficiency. The capability of the simulation could be up to tens of degrees of freedom. The implementation of the algorithm and the package usage are introduced, followed by typical examples and code test results.

研究动机与目标

• 开发一个灵活、可扩展的量子动力学软件包，能够以高精度模拟复杂的分子体系。
• 通过采用 Smolyak 稀疏网格方法，克服量子动力学中的维度灾难问题，相比传统的直接乘积基组，显著降低计算成本。
• 通过 MPI 实现高效的并行化，以在中等到高性能计算集群上处理大规模模拟。
• 提供一个通用工具，适用于多种问题，如振动能级、波包传播和量子门模拟，且不受哈密顿量形式的限制。
• 通过真实分子体系（如吡唑啉）演示在最多 12 个自由度的体系中，ElVibRot-MPI 具备优异的并行效率和可扩展性。

提出的方法

• 程序通过 Smolyak 稀疏网格构造求解时间依赖的薛定谔方程，能够以远少于全张量积网格的网格点数，高效表示高维函数。
• 波函数在基于 Smolyak 的稀疏网格基组中展开，从而在计算成本显著降低的前提下，实现对分子波函数的高精度表示。
• 该方法采用曲线坐标系，并允许任意哈密顿量形式，使其可广泛适用于不同类型的分子体系。
• 通过 MPI 实现并行化，将 Smolyak 网格项的计算分布在多个处理器上，并采用动态负载均衡策略以确保效率。
• 软件包采用 Davidson 迭代法求解大规模特征值问题，并支持切比雪夫时间推进法用于时间依赖模拟。
• 通过灵活的输入文件配置输入参数，支持直接对角化、迭代求解器以及大规模计算中的内存节省选项。

实验结果

研究问题

• RQ1 Smolyak 稀疏网格方法能否通过 MPI 有效并行化，从而实现对具有数十个自由度的分子体系的可扩展量子动力学模拟？
• RQ2 ElVibRot-MPI 在高维体系中的计算性能和精度与标准直接乘积基组方法相比如何？
• RQ3 负载均衡和通信开销在大规模模拟中对 Smolyak 算法的并行效率影响有多大？
• RQ4 ElVibRot-MPI 能否以高精度和合理计算成本模拟复杂分子体系（如吡唑啉，12 个自由度）？
• RQ5 ElVibRot-MPI 在处理器数量增加时的可扩展性如何？其在真实分子电子振动模型中的表现如何？

主要发现

• 对于 10 个自由度的体系，ElVibRot-MPI 在 L=6 时约 10 分钟内完成计算（使用 12 个处理器），而直接乘积法需超过 65 小时。
• 在二次势场中高斯波包的自相关函数模拟中，ElVibRot-MPI 在 L=4（8 个自由度）和 L=6（10 个自由度）时与直接乘积法参考结果高度一致。
• 在吡唑啉（12 个自由度）的模拟中，8 个处理器在首次迭代后即实现良好的工作负载分配，表明负载均衡有效。
• ElVibRot-MPI 的并行效率随节点数量增加而提升，尽管存在通信开销，仍实现了良好的加速比，尤其得益于每个节点的映射表大小显著减小。
• 该软件包在模拟多种分子体系的振动能级、波包传播和量子门操作方面表现出高度灵活性和鲁棒性。
• 采用带内存节省选项的 Davidson 迭代求解器，可高效计算大规模问题的激发态和波函数。