Implicit Neural Solver for Time-dependent Linear PDEs with Convergence Guarantee

  • 2019-10-08 15:20:01
  • Suprosanna Shit, Abinav Ravi, Ivan Ezhov, Jana Lipkova, Marie Piraud, Bjoern Menze
  • 1

Abstract

Fast and accurate solution of time-dependent partial differential equations(PDEs) is of key interest in many research fields including physics,engineering, and biology. Generally, implicit schemes are preferred over theexplicit ones for better stability and correctness. The existing implicitschemes are usually iterative and employ a general-purpose solver which may besub-optimal for a specific class of PDEs. In this paper, we propose a neuralsolver to learn an optimal iterative scheme for a class of PDEs, in adata-driven fashion. We achieve this goal by modifying an iteration of anexisting semi-implicit solver using a deep neural network. Further, we providetheoretical proof that our approach preserves the correctness and convergenceguarantees provided by the existing iterative-solvers. We also demonstrate thatour model generalizes to a different parameter setting than the one seen duringtraining and achieves faster convergence compared to the semi-implicit schemes.

 

Quick Read (beta)

loading the full paper ...