[论文解读] An efficient implementation of Slater-Condon rules

Anthony Scemama, Emmanuel Giner|arXiv (Cornell University)|Nov 25, 2013

Algorithms and Data Compression参考文献 9被引用 13

一句话总结

本文提出了一种使用x86-64 SSE4.2位操作指令优化实现Slater-Condon规则的方法，可快速确定量子化学计算中电子激发的阶数及涉及的自旋轨道。该方法相比基于软件的实现将计算时间减少了6倍以上，每项操作的CPU周期数保持在10–90个，与体系大小无关。

ABSTRACT

Slater-Condon rules are at the heart of any quantum chemistry method as they allow to simplify $3N$-dimensional integrals as sums of 3- or 6-dimensional integrals. In this paper, we propose an efficient implementation of those rules in order to identify very rapidly which integrals are involved in a matrix element expressed in the determinant basis set. This implementation takes advantage of the bit manipulation instructions on x86 architectures that were introduced in 2008 with the SSE4.2 instruction set. Finding which spin-orbitals are involved in the calculation of a matrix element doesn't depend on the number of electrons of the system.

研究动机与目标

• 通过高效识别Slater行列式之间的电子激发，加速基于行列式驱动的量子化学计算。
• 克服由于访问二电子积分时昂贵的随机内存访问导致的矩阵元计算性能瓶颈。
• 利用现代CPU的位操作指令（popcnt、xor）实现激发阶数和轨道替换的常数时间计算。
• 实现一种与电子数无关的可扩展方法，从而在大规模CI和FCI计算中实现高性能。

提出的方法

• 将每个Slater行列式表示为一对64位位串，分别对应α和β自旋轨道，位位置对应轨道编号。
• 通过两个行列式位串之间的XOR运算识别不同的轨道（空穴和占据轨道），再使用硬件popcnt指令计算汉明权重。
• 将激发阶数d计算为不同位数的总和的一半，实现每64位字的O(1)计算，与电子数无关。
• 对于单激发和双激发，使用位操作和掩码技术提取涉及的具体自旋轨道编号。
• 通过奇偶性计算相位因子，利用popcount和位反转逻辑实现。
• 使用AVX/SSE2指令集编译，以启用硬件加速的popcnt，确保执行延迟与位密度无关。

实验结果

研究问题

• RQ1 现代x86-64指令集能否显著加速量子化学方法中行列式之间电子激发的识别？
• RQ2 位级并行性在多大程度上可降低大规模CI或FCI计算中Slater-Condon矩阵元的计算成本？
• RQ3 硬件加速的popcount（popcnt）是否能实现与体系大小无关的常数时间激发阶数计算？
• RQ4 与基于软件的实现相比，该方法在CPU周期数和实际运行时间上的性能表现如何？
• RQ5 该方法在处理大行列式和高阶激发时是否能实现高效扩展并保持低延迟？

主要发现

• n_excitations函数在AVX下约耗时10个周期，在SSE2下约耗时73个周期，与电子数无关。
• get_excitation函数（用于获取单激发和双激发涉及的自旋轨道列表）平均耗时18–89个周期，同样与电子数无关。
• 对于水分子（128个MO），使用AVX指令在单核上对10,000个行列式计算一电子密度矩阵仅耗时0.2秒。
• 与基于软件的popcnt实现相比，该方法实现了超过6倍的加速，尤其在高阶激发情况下增益最大。
• 对于d > 2的情况，平均CPU周期数较低（AVX下为6.7个周期），表明该方法能高效处理绝大多数非平凡激发情况。
• 通过大幅减少对二电子积分的随机内存访问次数，降低了内存访问开销，因为激发检测速度已远快于积分获取速度。