2025-11-18T23:52:13.733274

Symmetry adaptation for self-consistent many-body calculations

Dong, Gull

The exploitation of space group symmetries in numerical calculations of periodic crystalline solids accelerates calculations and provides physical insight. We present results for a space-group symmetry adaptation of electronic structure calculations within the finite-temperature self-consistent GW method along with an efficient parallelization scheme on accelerators. Our implementation employs the simultaneous diagonalization of the Dirac characters of the orbital representation. Results show that symmetry adaptation in self-consistent many-body codes results in substantial improvements of the runtime, and that block diagonalization on top of a restriction to the irreducible wedge results in additional speedup.

academic

Symmetry adaptation for self-consistent many-body calculations

基本信息

论文ID: 2405.09494
标题: Symmetry adaptation for self-consistent many-body calculations
作者: Xinyang Dong (AI for Science Institute Beijing & University of Michigan), Emanuel Gull (University of Michigan)
分类: physics.comp-ph
发表时间: May 16, 2024 (Preprint submitted to Computer Physics Communications)
论文链接: https://arxiv.org/abs/2405.09494

摘要

本文研究了在周期性晶体固体的数值计算中利用空间群对称性来加速计算并提供物理洞察。作者在有限温度自洽GW方法的电子结构计算中实现了空间群对称性适应，并提出了在加速器上的高效并行化方案。实现采用轨道表示的Dirac特征的同时对角化方法。结果表明，自洽多体代码中的对称性适应显著改善了运行时间，在不可约楔限制基础上的块对角化提供了额外的加速效果。

研究背景与动机

要解决的问题: 现代多体理论计算（如自洽GW方法）在处理周期性晶体材料时面临巨大的计算负担，需要重复计算频率依赖的传播子、顶点函数和屏蔽相互作用等复杂对象。
问题的重要性:
- 空间群对称性是理解晶体材料的关键，能够提供物理洞察
- 对称性的利用可以显著加速数值计算
- GPU等现代计算架构能够有效利用群结构暴露的并行性
现有方法的局限性:
- 标准电子结构代码（Hartree-Fock、DFT、非自洽GW）主要基于单粒子密度矩阵，对称性适应形式已经成熟
- 但现代多体技术需要计算密度矩阵之外的对象，对称性适应的形式化程度不够
研究动机: 将Dovesi等人在Hartree-Fock和DFT理论中开创的对称性适应形式推广到自洽GW方法，并在现代GPU架构上实现高效并行化。

核心贡献

方法扩展: 将基于Dirac特征同时对角化的对称性适应方法从单体理论扩展到自洽多体GW计算
高效实现: 开发了在GPU加速器上的高效并行化方案，实现了MPI和CUDA的混合并行
性能提升: 证明了对称性适应结合块对角化能够带来约一个数量级的浮点运算减少
算法优化: 提出了处理非同构空间群和投影表示的完整数值算法

空间群操作: 表示为 $\hat{\alpha} = \{\alpha|v(\alpha)\}$ ，其中 $\alpha$ 是点群操作， $v(\alpha)$ 是平移
轨道变换: 对称操作在轨道上的作用为： $\hat{\alpha}g^k_{(xj)}(r) = \exp[-i\tilde{k} \cdot v_{\alpha x}] \times [O(\alpha)g^{\tilde{k}}_{(xj)}(r)]$

2. 投影表示

对于动量 $k$ ，定义投影表示矩阵： $D^k(\alpha) = \exp[ik \cdot v(\alpha)]O^k(\hat{\alpha})$ $\lambda^k(\alpha, \beta) = \exp\{ik \cdot [v(\beta) - \alpha v(\beta)]\}$

3. Dirac特征同时对角化

Dirac特征定义: $\Omega_c = \frac{n_c}{h} \sum_{\beta \in G} D(\alpha) \cdot D(\gamma) \cdot D(\beta)^{-1}$
同时对角化: 通过对所有相关Dirac特征的同时对角化获得变换矩阵 $U^k$

技术创新点

扩展到多体理论: 首次将Dirac特征同时对角化方法系统地应用于自洽GW计算
张量变换: 开发了三指标相互作用张量的对称性变换公式： $V^{\tilde{k}_i\tilde{k}_j} = \bar{O}^q(\hat{\alpha})O^{k_i}(\hat{\alpha})V^{k_ik_j} O^{k_j\dagger}(\hat{\alpha})$
GPU优化: 设计了异步流处理和批量ZGEMM调用的GPU加速方案

实验设置

数据集

测试了四种III-V族和IV族化合物：

Si (空间群227, 非同构)
BN (空间群194, 非同构)
AlP (空间群216, 同构)
GaAs (空间群216, 同构)

计算参数

基组: gthdzvp基组和def2-svp-ri辅助基组
温度网格: 114个虚时间点，103个玻色频率点
动量网格: $n_k \times n_k \times n_k$ ( $n_k = 1,2,4,6$ )

评价指标

浮点运算次数 (FLOP)
GPU加速比
内存使用量

对比方法

完整布里渊区计算 (Full)
仅不可约楔旋转 (Rotation)
旋转+块对角化 (Block Diag)

$n_k$	$n_{ik}$	Full	Rotation	Block Diag	加速比
1	1	$1.31 \times 10^{10}$	$1.31 \times 10^{10}$	$1.50 \times 10^9$	8.7×
2	3	$1.73 \times 10^{12}$	$1.01 \times 10^{12}$	$2.24 \times 10^{11}$	7.7×
4	8	$1.10 \times 10^{14}$	$2.13 \times 10^{13}$	$8.55 \times 10^{12}$	12.9×
6	16	$1.25 \times 10^{15}$	$1.43 \times 10^{14}$	$6.87 \times 10^{13}$	18.2×