用神经网络做无矩阵预条件器,加速量子色动力学计算
Matrix-free Neural Preconditioner for the Dirac Operator in Lattice Gauge Theory
- 不依赖显式矩阵,通过学习算子映射构造预条件器
- 使迭代次数减半,显著提升求解效率
- 可零样本迁移至不同尺寸格点,通用性强
在格点量子色动力学(QCD)中,生成样本和计算可观测量涉及求解稀疏但病态的厄米正定线性系统,通常使用共轭梯度(CG)等迭代方法,耗时且计算成本高。现有最优的多网格预条件器构建复杂,带来额外开销。本文提出一种基于算子学习的框架,构建无需显式矩阵表示的线性映射作为预条件器,实现高效训练与应用。在1+1维U(1)规范理论、含两重简并质量费米子的施温格模型中,该方法有效降低线性系统的条件数,使收敛所需迭代次数约减半。进一步证明,该框架可学习依赖于格点结构的通用映射,具备对不同尺寸规范场配置的零样本泛化能力。
原文摘要 · Abstract (English)
Linear systems arise in generating samples and in calculating observables in lattice quantum chromodynamics~(QCD). Solving the Hermitian positive definite systems, which are sparse but ill-conditioned, involves using iterative methods, such as Conjugate Gradient (CG), which are time-consuming and computationally expensive. Preconditioners can effectively accelerate this process, with the state-of-the-art being multigrid preconditioners. However, constructing useful preconditioners can be challenging, adding additional computational overhead, especially in large linear systems. We propose a framework, leveraging operator learning techniques, to construct linear maps as effective preconditioners. The method in this work does \emph{not} rely on explicit matrices from either the original linear systems or the produced preconditioners, allowing efficient model training and application in the CG solver. In the context of the Schwinger model U(1) gauge theory in 1+1 spacetime dimensions with two degenerate-mass fermions), this preconditioning scheme effectively decreases the condition number of the linear systems and approximately halves the number of iterations required for convergence in relevant parameter ranges. We further demonstrate the framework learns a general mapping dependent on the lattice structure which leads to zero-shot learning ability for the Dirac operators constructed from gauge field configurations of different sizes.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。