Gauge-equivariant pooling layers for preconditioners in lattice QCD

INSPIRE: 2652849
arXiv: 2304.10438
DOI: 10.1103/PhysRevD.110.034517

Authors: Lehner, Christoph, Wettig, Tilo

Submitted: 20 April 2023

Subjects:

• hep-lat
• cs.LG
• cs.NA
• math.NA
• Lattice
• Computing
• Math and Math Physics

Journal reference: Phys.Rev.D 110 (3) 034517 (2024)

Citations: 18

Abstract

We demonstrate that gauge-equivariant pooling and unpooling layers can perform as well as traditional restriction and prolongation layers in multigrid preconditioner models for lattice QCD. These layers introduce a gauge degree of freedom on the coarse grid, allowing for the use of explicitly gauge-equivariant layers on the coarse grid. We investigate the construction of coarse-grid gauge fields and study their efficiency in the preconditioner model. We show that a combined multigrid neural network using a Galerkin construction for the coarse-grid gauge field eliminates critical slowing down.