Neural Preconditioner: Learning Preconditioners for Conjugate Gradient PDE Solvers

Our Method is inherently a numerical solver that can produce  arbitrary accuracy.  Here we demonstrate how our method win over end-to-end learning method in terms of accuracy. 

Poisson2D-Ours/GT (as Inviscid Navier-Stokes) 

poisson2d-2.mp4

Ours / Numerical Solver Simulation

MGN / End-to-end Neural Network Simulation

Heat2D-Ours/GT

heat.mp4

Ours / Numerical Solver Simulation

MGN / End-to-end Neural Network Simulation

Wave2D

wave.mp4

Ours / Numerical Solver Simulation

mgn.mp4

MGN / End-to-end Neural Network Simulation