RAPNet: Accelerating Algebraic Multigrid with Learned Sparse Corrections

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Fink, Yali, Ben-Yair, Ido, Ruthotto, Lars, Treister, Eran
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913163998396416
author Fink, Yali
Ben-Yair, Ido
Ruthotto, Lars
Treister, Eran
author_facet Fink, Yali
Ben-Yair, Ido
Ruthotto, Lars
Treister, Eran
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2605_26854
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle RAPNet: Accelerating Algebraic Multigrid with Learned Sparse Corrections
Fink, Yali
Ben-Yair, Ido
Ruthotto, Lars
Treister, Eran
Machine Learning
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.
title RAPNet: Accelerating Algebraic Multigrid with Learned Sparse Corrections
topic Machine Learning
url https://arxiv.org/abs/2605.26854