Towards fast weak adversarial training to solve high dimensional parabolic partial differential equations using XNODE-WAN

by   Paul Valsecchi Oliva, et al.

Due to the curse of dimensionality, solving high dimensional parabolic partial differential equations (PDEs) has been a challenging problem for decades. Recently, a weak adversarial network (WAN) proposed in (Y.Zang et al., 2020) offered a flexible and computationally efficient approach to tackle this problem defined on arbitrary domains by leveraging the weak solution. WAN reformulates the PDE problem as a generative adversarial network, where the weak solution (primal network) and the test function (adversarial network) are parameterized by the multi-layer deep neural networks (DNNs). However, it is not yet clear whether DNNs are the most effective model for the parabolic PDE solutions as they do not take into account the fundamentally different roles played by time and spatial variables in the solution. To reinforce the difference, we design a novel so-called XNODE model for the primal network, which is built on the neural ODE (NODE) model with additional spatial dependency to incorporate the a priori information of the PDEs and serve as a universal and effective approximation to the solution. The proposed hybrid method (XNODE-WAN), by integrating the XNODE model within the WAN framework, leads to significant improvement in the performance and efficiency of training. Numerical results show that our method can reduce the training time to a fraction of that of the WAN model.



There are no comments yet.


page 10

page 22

page 25


Weak Adversarial Networks for High-dimensional Partial Differential Equations

Solving general high-dimensional partial differential equations (PDE) is...

Friedrichs Learning: Weak Solutions of Partial Differential Equations via Deep Learning

This paper proposes Friedrichs learning as a novel deep learning methodo...

Adversarial Multi-task Learning Enhanced Physics-informed Neural Networks for Solving Partial Differential Equations

Recently, researchers have utilized neural networks to accurately solve ...

A Deep Neural Network Surrogate for High-Dimensional Random Partial Differential Equations

Developing efficient numerical algorithms for high dimensional random Pa...

SelectNet: Self-paced Learning for High-dimensional Partial Differential Equations

The residual method with deep neural networks as function parametrizatio...

Neural Network Based Variational Methods for Solving Quadratic Porous Medium Equations in High Dimensions

In this paper, we propose and study neural network based methods for sol...

Understanding Loss Landscapes of Neural Network Models in Solving Partial Differential Equations

Solving partial differential equations (PDEs) by parametrizing its solut...

Code Repositories


We develop a custom ML routine XNODE-WAN to solve parabolic PDEs with high efficiency.

view repo
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.