[論文レビュー] 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のビット操作命令を用いたスレーター・コンドン規則の最適化実装を提示しており、量子化学計算における電子励起次数および関与するスピン軌道の高速特定を可能にしている。この手法により、ソフトウェアベースのアプローチと比較して計算時間を6倍以上短縮し、システムサイズに依存しない10–90 CPUサイクル/操作を達成している。

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.

研究の動機と目的

• スレーター行列式間の電子励起を効率的に特定することで、行列式駆動型の量子化学計算を高速化すること。
• 二電子積分への高コストなランダムメモリアクセスが行列要素計算の性能ボトルネックとなっている問題を克服すること。
• 現代のCPUビット操作命令（popcnt、xor）を活用し、励起次数および軌道置換の定数時間計算を実現すること。
• 電子数に依存しないスケーリングが可能な手法を実装することで、大規模なCIおよびFCI計算における高性能を実現すること。

提案手法

• スピン軌道のαおよびβに対し、64ビットのビットストリングのペアとして各スレーター行列式を表現し、ビット位置を軌道インデックスに対応させる。
• 二つの行列式のビットストリング間にXOR演算を適用して異なる軌道（ホールおよび粒子）を特定し、ハードウェアのpopcnt命令を用いてハミング重みを数える。
• 差異ビットの合計数の半分として励起次数dを計算し、電子数に依存しないO(1)の計算が64ビットワード単位で可能になる。
• 単一および二重励起の場合、ビット演算およびマスク技術を用いて関与する具体的なスピン軌道インデックスを抽出する。
• パリティを用いた置換の位相因子計算を、カウント数およびビット反転論理を用いて実装する。
• AVX/SSE2命令セットでコンパイルし、ハードウェア加速されたpopcntを有効化することで、ビット密度に依存しない低遅延実行を実現する。

実験結果

リサーチクエスチョン

• RQ1 現代のx86-64命令セットは、量子化学手法における行列式間の電子励起の特定を顕著に高速化できるか？
• RQ2 ビットレベル並列処理は、大規模なCIまたはFCI計算におけるスレーター・コンドン行列要素の評価コストをどの程度低減できるか？
• RQ3 ハードウェア加速されたカウント（popcnt）は、システムサイズに依存せず、励起次数の定数時間計算を可能にするか？
• RQ4 本実装のパフォーマンスは、ソフトウェアベースの代替手法と比較して、CPUサイクルおよびウォールクロック時間の観点でどの程度優れているか？
• RQ5 本手法は、大規模な行列式および高次の励起に対して、低遅延を維持したまま効率的にスケーリング可能か？

主な発見

• n_excitations関数は、二つの行列式間の励起次数を、AVXでは約10サイクル、SSE2では73サイクルで計算可能であり、電子数に依存しない。
• get_excitation関数（単一および二重励起に必要なスピン軌道インデックスのリスト取得）は、平均して18–89サイクルで実行され、再び電子数に依存しない。
• 水分子（128 MOs）の10,000個の行列式に対する1電子密度行列計算は、AVX命令を用いた1コアでわずか0.2秒で完了した。
• ソフトウェアベースのpopcntと比較して6倍以上のスピードアップを達成しており、特に高次の励起ケースで顕著な向上が見られた。
• d > 2のケースの平均CPUサイクル数は低く（AVXでは6.7サイクル）、これは本手法が多数の非自明な励起ケースを効率的に処理できることを示している。
• 励起検出が積分の取得よりもはるかに高速になったため、二電子積分のランダムメモリフェッチのオーバーヘッドが大幅に削減された。