Symmetry adaptation for self-consistent many-body calculations

Basic Information

• Paper ID: 2405.09494
• Title: Symmetry adaptation for self-consistent many-body calculations
• Authors: Xinyang Dong (AI for Science Institute Beijing & University of Michigan), Emanuel Gull (University of Michigan)
• Classification: physics.comp-ph
• Publication Date: May 16, 2024 (Preprint submitted to Computer Physics Communications)
• Paper Link: https://arxiv.org/abs/2405.09494

Abstract

This paper investigates the utilization of space group symmetry to accelerate computations and provide physical insights in numerical calculations of periodic crystalline solids. The authors implement space group symmetry adaptation in finite-temperature self-consistent GW electronic structure calculations and propose an efficient parallelization scheme on accelerators. The implementation employs the simultaneous diagonalization method of Dirac characteristics in the orbital representation. Results demonstrate that symmetry adaptation in self-consistent many-body codes significantly improves runtime, with block diagonalization on the irreducible wedge basis providing additional acceleration benefits.

Research Background and Motivation

1. Problem to be addressed: Modern many-body theoretical calculations (such as self-consistent GW methods) face enormous computational burdens when processing periodic crystalline materials, requiring repeated calculations of complex objects such as frequency-dependent propagators, vertex functions, and screened interactions.
2. Problem significance:
   • Space group symmetry is fundamental to understanding crystalline materials and provides physical insights
   • Exploitation of symmetry can significantly accelerate numerical computations
   • Modern computational architectures such as GPUs can effectively utilize the parallelism exposed by group structures
3. Limitations of existing methods:
   • Standard electronic structure codes (Hartree-Fock, DFT, non-self-consistent GW) are primarily based on single-particle density matrices, with mature symmetry adaptation formalism
   • However, modern many-body techniques require calculations of objects beyond the density matrix, and the formalization of symmetry adaptation is insufficient
4. Research motivation: Extend the symmetry adaptation formalism pioneered by Dovesi et al. in Hartree-Fock and DFT theory to self-consistent GW methods, and implement efficient parallelization on modern GPU architectures.

Core Contributions

1. Method extension: Extend the symmetry adaptation method based on simultaneous diagonalization of Dirac characteristics from single-body theory to self-consistent many-body GW calculations
2. Efficient implementation: Develop an efficient parallelization scheme on GPU accelerators, implementing hybrid MPI and CUDA parallelization
3. Performance improvement: Demonstrate that symmetry adaptation combined with block diagonalization can achieve approximately one order of magnitude reduction in floating-point operations
4. Algorithm optimization: Propose a complete numerical algorithm for handling non-symmorphic space groups and projective representations

Methodology Details

Task Definition

This paper studies how to exploit space group symmetry to accelerate electronic structure calculations of periodic crystalline solids at finite temperature, particularly in self-consistent GW methods. The input consists of crystal structure and Hamiltonian, with output being self-consistent Green's functions and self-energy.

Model Architecture

1. Symmetry Theory Foundation

• Space group operations: Represented as $\hat{\alpha} = \{\alpha|v(\alpha)\}$, where $\alpha$ is a point group operation and $v(\alpha)$ is a translation
• Orbital transformation: The action of symmetry operations on orbitals is: $\hat{\alpha}g^k_{(xj)}(r) = \exp[-i\tilde{k} \cdot v(\alpha x)] \times [O(\alpha)g^{\tilde{k}}_{(xj)}(r)]$

2. Projective Representations

For momentum $k$, the projective representation matrix is defined as: $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. Simultaneous Diagonalization of Dirac Characteristics

• Dirac characteristic definition: $\Omega_c = \frac{n_c}{h} \sum_{\beta \in G} D(\alpha) \cdot D(\gamma) \cdot D(\beta)^{-1}$
• Simultaneous diagonalization: Obtain transformation matrix $U^k$ through simultaneous diagonalization of all relevant Dirac characteristics

Technical Innovations

1. Extension to many-body theory: First systematic application of simultaneous diagonalization of Dirac characteristics to self-consistent GW calculations
2. Tensor transformation: Develop symmetry transformation formulas for three-index interaction tensors: $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})$
3. GPU optimization: Design GPU acceleration scheme with asynchronous stream processing and batched ZGEMM calls

Experimental Setup

Datasets

Tested four III-V and IV group compounds:

• Si (space group 227, non-symmorphic)
• BN (space group 194, non-symmorphic)
• AlP (space group 216, symmorphic)
• GaAs (space group 216, symmorphic)

Computational Parameters

• Basis set: gthdzvp basis set and def2-svp-ri auxiliary basis set
• Temperature grid: 114 imaginary time points, 103 bosonic frequency points
• Momentum grid: $n_k \times n_k \times n_k$ ($n_k = 1,2,4,6$)

Evaluation Metrics

• Floating-point operations (FLOP)
• GPU acceleration ratio
• Memory usage

Comparison Methods

• Full Brillouin zone calculation (Full)
• Irreducible wedge rotation only (Rotation)
• Rotation + block diagonalization (Block Diag)

Experimental Results

Main Results

Floating-point Operation Reduction

Taking Si as an example, FLOP comparison at different $n_k$ values:

GPU Acceleration Performance

• Achieved near-ideal linear speedup on 16 V100 GPUs
• Both P0 and Σ̃ computation kernels demonstrate excellent scalability

Experimental Findings

1. Effect of k-point count: The advantage of inter-point rotation becomes more pronounced as the total number of k-points increases
2. Block diagonalization advantage: Block diagonalization is more effective with fewer k-points, as more points lie on the IBZ boundary
3. Non-symmorphic group advantage: Non-symmorphic space groups (such as Si, BN) show greater speedup compared to symmorphic groups

Related Work

Main Research Directions

1. Traditional symmetry adaptation: Pioneering work by Dovesi et al. in the CRYSTAL code
2. Many-body theory: Hedin's GW method and its self-consistent implementation
3. GPU computing: Accelerator optimization for electronic structure calculations

Advantages of This Work

• First systematic extension of symmetry adaptation to self-consistent many-body calculations
• Provides complete treatment of non-symmorphic space groups
• Implements efficient GPU parallelization

Conclusions and Discussion

Main Conclusions

1. Symmetry adaptation brings significant runtime improvements in self-consistent many-body codes
2. Block diagonalization provides additional acceleration on the irreducible wedge basis
3. GPU architecture effectively utilizes the parallelism exposed by symmetry

Limitations

1. Current implementation is limited to standard space groups, excluding magnetic space groups
2. For systems with very large k-point counts, the advantage of block diagonalization diminishes
3. Requires sufficient GPU memory to store critical data structures

Future Directions

1. Magnetic space groups: Extension to Shubnikov groups for handling magnetic and relativistic systems
2. Optical response: Exploit symmetry knowledge to interpret optical response functions
3. Higher-order methods: Application to more accurate simulation methods including vertex functions

In-depth Evaluation

Strengths

1. Theoretical rigor: Based on mature group theory foundations with complete mathematical derivations
2. Practical value: Achieves approximately one order of magnitude computational acceleration, significant for large-scale calculations
3. Technical completeness: Provides complete solution from theory to implementation
4. Performance verification: Method effectiveness validated across multiple material systems

Weaknesses

1. Scope of applicability: Currently limited to periodic systems; extensibility to surface or defect systems remains unclear
2. Memory requirements: GPU implementation has high memory demands, potentially limiting application to large systems
3. Algorithm stability: Simultaneous diagonalization may encounter numerical stability issues for large orbital representation matrices

Impact

1. Academic contribution: Provides standard paradigm for symmetry exploitation in many-body calculations
2. Practical value: Significantly reduces computational cost of self-consistent GW calculations, enabling calculations on larger systems
3. Reproducibility: Implemented based on open-source software, facilitating community adoption and improvement

Applicable Scenarios

• Periodic crystalline materials with high symmetry
• Electronic structure calculations requiring accurate many-body effects
• Large-scale parallel computing environments, particularly GPU clusters

References

This paper is primarily based on the following key works:

1. Dovesi et al.'s symmetry adaptation theory (Int. J. Quantum Chem. 1986, 1998)
2. Hedin's GW method (Phys. Rev. 1965)
3. Bradley & Cracknell's mathematical theory of solid symmetry
4. Lax's symmetry principles in solid and molecular physics

This paper represents an important contribution to computational physics, successfully combining symmetry theory with modern many-body calculations and GPU acceleration technology, providing a new solution for efficient electronic structure calculations.