F1: Neural Network Preconditioning for U(1) Wilson-type Dirac Operators

28 Jul 2021, 15:00
1h
Poster Algorithms (including Machine Learning, Quantum Computing, Tensor Networks) Poster

Speakers

Brian Xiao (MIT) Yin Lin (University of Chicago)

Description

We report progress in preconditioning Wilson-type Dirac operators in 1+1 dimensional U(1) lattice field theory using a neural network. We have developed a convolutional network that produces a preconditioner of comparable sparsity to the input operator. Once the model is trained, applying it to produce preconditioners is computationally cheap; with an optimized implementation, the neural network approach may provide a practical improvement over algebraic multigrid-based preconditioning, since the network is cheaper to apply and tends to produce much sparser preconditioners compared to AMG. So far, we have achieved a consistent reduction in iteration count during the solution of Dirac-operator linear systems using the conjugate gradient algorithm. Furthermore, the purely convolutional network architecture generalizes well across lattice volumes.

Primary author

Co-authors

Prof. Phiala Shanahan (Massachusetts Institute of Technology) Dr Daniel Hackett (MIT) Salvatore Cali (MIT) Yin Lin (University of Chicago)

Presentation materials