← Back to Machine Learning
cs.LG

Can neural networks fix multigrid's fundamental tradeoff?

Yali Fink, Ido Ben-Yair, Lars Ruthotto, Eran Treister

May 26, 2026

Solving enormous sparse linear systems is foundational to scientific computing, but algebraic multigrid (AMG) solvers face a hard choice: sparse operators are fast but converge slowly; dense ones converge well but waste computation. RAPNet, a graph neural network trained level-by-level, learns to generate sparse coarse operators that sidestep this tradeoff. The method runs only during setup, keeping the solve phase efficient, and generalizes from small subgraphs to million-node problems. Testing on PDE discretizations and graph Laplacians shows 10–50% speedups over classical alternatives, with particular value for repeated-solve tasks like eigenproblems and inverse problems.
Published as RAPNet: Accelerating Algebraic Multigrid with Learned Sparse Corrections arXiv:2605.26854
Read the original paper →