SVD-NO: Learning PDE Solution Operators with SVD Integral Kernels
About
Neural operators have emerged as a promising paradigm for learning solution operators of partial differential equa- tions (PDEs) directly from data. Existing methods, such as those based on Fourier or graph techniques, make strong as- sumptions about the structure of the kernel integral opera- tor, assumptions which may limit expressivity. We present SVD-NO, a neural operator that explicitly parameterizes the kernel by its singular-value decomposition (SVD) and then carries out the integral directly in the low-rank basis. Two lightweight networks learn the left and right singular func- tions, a diagonal parameter matrix learns the singular values, and a Gram-matrix regularizer enforces orthonormality. As SVD-NO approximates the full kernel, it obtains a high de- gree of expressivity. Furthermore, due to its low-rank struc- ture the computational complexity of applying the operator remains reasonable, leading to a practical system. In exten- sive evaluations on five diverse benchmark equations, SVD- NO achieves a new state of the art. In particular, SVD-NO provides greater performance gains on PDEs whose solutions are highly spatially variable. The code of this work is publicly available at https://github.com/2noamk/SVDNO.git.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Learning PDE Solution Operators | 2D Shallow Water | Mean L2 Relative Error0.39 | 20 | |
| Learning PDE Solution Operators | Allen-Cahn 1D | Mean L2 Relative Error0.07 | 12 | |
| Learning PDE Solution Operators | 1D Diffusion-Reaction | Mean L2 Rel Error33 | 12 | |
| Learning PDE Solution Operators | 1D Diffusion-Sorption | Mean L2 Relative Error1.09 | 8 | |
| Learning PDE Solution Operators | 1D Cahn-Hilliard | Mean L2 Relative Error0.47 | 8 | |
| Learning PDE Solution Operators | 3D Maxwell | Mean L2 Relative Error0.6356 | 8 | |
| Learning PDE Solution Operators | 1D Cahn-Hilliard | Training Time per Epoch (s)2.71 | 7 | |
| Learning PDE Solution Operators | 1D Allen-Cahn | Training Time (s)5.87 | 7 | |
| Learning PDE Solution Operators | 1D Diffusion-Sorption | Training Time per Epoch (s)4.95 | 7 | |
| Learning PDE Solution Operators | 1D Diffusion-Reaction | Training Time per Epoch (s)13.64 | 7 |