PoS - Proceedings of Science
Volume 453 - The 40th International Symposium on Lattice Field Theory (LATTICE2023) - Algorithms and Artificial Intelligence
Gauge-equivariant multigrid neural networks
D. Knüttel, C. Lehner and T. Wettig*
Full text: pdf
Pre-published on: December 27, 2023
Published on: November 06, 2024
Abstract
We show how multigrid preconditioners for the Wilson-clover Dirac operator can be constructed using gauge-equivariant neural networks. For the multigrid solve we employ parallel-transport convolution layers. For the multigrid setup we consider two versions: the standard construction based on the near-null space of the operator and a gauge-equivariant construction using pooling and subsampling layers. We show that both versions eliminate critical slowing down. We also show that transfer learning works and that our approach allows for communication-avoiding algorithms on large machines. In the outlook we discuss how the multigrid setup can be accelerated using gauge-invariant properties of the gauge field.
DOI: https://doi.org/10.22323/1.453.0037
How to cite

Metadata are provided both in "article" format (very similar to INSPIRE) as this helps creating very compact bibliographies which can be beneficial to authors and readers, and in "proceeding" format which is more detailed and complete.

Open Access
Creative Commons LicenseCopyright owned by the author(s) under the term of the Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 International License.