Search papers, labs, and topics across Lattice.
This paper introduces a neural network designed to learn the inverse operation of Hierarchical Off-Diagonal Low-Rank (HODLR) matrices, leveraging a fast direct solver for efficient computation. The architecture is further enhanced to handle nonlinear solution operators for partial differential equations (PDEs) by integrating deep sub-networks in place of traditional linear layers. Experimental results demonstrate the network's effectiveness in solving various linear and nonlinear problems, achieving superior inference times compared to classical numerical solvers and existing neural operator learning methods.
A neural network that accelerates PDE solutions by learning HODLR matrix inverses outperforms traditional solvers and existing neural operators in speed and accuracy.
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. HODLR (Hierarchical Off-Diagonal Low-Rank) matrices are a subclass of hierarchical matrices in which all off-diagonal submatrices at every level of a recursive binary partition are low-rank. In this article, we present a neural network that learns the inverse operation of HODLR matrices based on the fast direct solver for HODLR matrices developed by Ambikasaran and Darve (2013). We further extend the architecture to learn nonlinear solution operators associated with PDEs by replacing some of the linear layers with deep sub-networks. We demonstrate the performance of the proposed architecture by performing a comprehensive set of experiments that include (i) solving a linear problem such as the Fredholm integral equation of the second kind, (ii) solving PDEs such as the nonlinear Schr枚dinger equation, Burgers' equation, and the steady-state Darcy's flow equation, (iii) generalization study across varying parameter values, (iv) comparing the inference time of the proposed network with the run time of a classical numerical solver, and (v) comparing the proposed network with some of the existing neural operator learning networks.
