Lie Point Symmetry Data Augmentation for Neural PDE Solvers

Johannes Brandstetter · Max Welling · Daniel Worrall

Room 309
[ Abstract ] [ Livestream: Visit Deep Learning/MISC ]
Thu 21 Jul 10:45 a.m. — 10:50 a.m. PDT
[ Slides [ Paper PDF

Neural networks are increasingly being used to solve partial differential equations (PDEs), replacing slower numerical solvers. However, a critical issue is that neural PDE solvers require high-quality ground truth data, which usually must come from the very solvers they are designed to replace. Thus, we are presented with a proverbial chicken-and-egg problem. In this paper, we present a method, which can partially alleviate this problem, by improving neural PDE solver sample complexity---Lie point symmetry data augmentation (LPSDA). In the context of PDEs, it turns out we are able to quantitatively derive an exhaustive list of data transformations, based on the Lie point symmetry group of the PDEs in question, something not possible in other application areas. We present this framework and demonstrate how it can easily be deployed to improve neural PDE solver sample complexity by an order of magnitude.

Chat is not available.