arXiv Machine Learning

A Neural Hierarchical-Matrix Preconditioner for Real-Time GPU Solves

The paper presents a neural hierarchical‑matrix preconditioner designed for real‑time GPU solves of sparse symmetric positive‑definite systems that change each frame. By training a graph‑and‑attention network to predict an SPD approximate inverse in H²‑matrix format, the method achieves linear‑time inference and application, outperforming traditional multigrid setup times and local preconditioners. Experiments on 3D mesh diffusion problems show the preconditioner reduces conjugate‑gradient iterations from 116 to 33 and enables 120 fps real‑time performance for up to 3,647 unknowns.

arXiv Machine Learning
Jul 21

One-shot acceleration of transient PDE solvers via online-learned preconditioners

arXiv:2509. 08765v4 Announce Type: replace-cross Abstract: Data-driven acceleration of scientific computing workflows has been a high-profile aim of machine learning (ML) for science, with numerical simulation of transient partial differential equations (PDEs) being one of the main applications.

By Mikhail Khodak, Min Ki Jung, Brian Wynne, Edmond Chow, Egemen Kolemen
arXiv Machine Learning
Jun 19

A fast direct solver based neural network for solving PDEs

arXiv:2606. 19895v1 Announce Type: cross Abstract: The matrices arising from large scale $N$-body problems can be efficiently represented using hierarchical matrices, whose key idea is that the admissible off-diagonal sub-matrices can be well approximated by low-rank matrices across a hierarchy of matrix partitions.

By Jashwanth Reddy Kadaru, Vaishnavi Gujjula