Search papers, labs, and topics across Lattice.
RAPNet, a graph neural network (GNN) framework, learns to generate sparse and robust coarse operators for algebraic multigrid (AMG) from the sparse algebraic system itself. A level-wise training strategy allows RAPNet to learn from small subgraphs and generalize to million-node domains, overcoming limitations of prior neural AMG methods. Results demonstrate that RAPNet outperforms classical non-Galerkin baselines on PDE discretizations and graph Laplacians, particularly benefiting multi-query tasks.
By learning sparse corrections to coarse AMG operators, RAPNet breaks the sparsity-convergence trade-off that has long plagued algebraic multigrid methods.
The scalable solution of large sparse linear systems is a bottleneck in scientific computing and graph analysis. While algebraic multigrid (AMG) offers optimal linear scaling, its performance is severely constrained by the trade-off between the sparsity and convergence quality of coarse-grid operators. Classical AMG heuristics struggle to balance these objectives, often sacrificing stability or performance for sparsity. We propose RAPNet, a graph neural network (GNN) framework that resolves this trade-off by learning to generate sparse, robust coarse operators directly from the sparse algebraic system. Key to our approach is a level-wise training strategy that enables learning from small subgraphs and generalization to million-node domains, bypassing the bottlenecks of prior neural AMG attempts. RAPNet executes exclusively during the solver setup phase, ensuring that the solve phase retains its favorable computational properties. We show that our method outperforms classical non-Galerkin baselines on diverse PDE discretizations and graph Laplacians, making it particularly effective for multi-query tasks such as eigenproblems, time-dependent simulations, and inverse or design problems.
