Source-linked AI summary

Densely Multiplied Physics Informed Neural Networks

Feilong Jiang, Xiaonan Hou, Min Xia

arXiv:2402.04390v3cs.LGcs.AI

TL;DR

PINNs can produce low-precision or incorrect solutions to PDEs. This paper introduces a densely multiplied PINN architecture that reuses hidden-layer outputs without adding trainable parameters, and reports more accurate predictions across benchmark problems with the same training resources.

  • Problem

    PINNs can suffer from insufficient precision or incorrect outcomes when solving PDEs.

    Source: §1 · p. 1
  • Method

    DM-PINN reuses hidden-layer outputs through dense multiplication to improve expressive ability without increasing trainable parameters.

    Source: §3 · p. 4
  • Results

    Across several benchmarks, DM-PINN achieved more accurate predictions with the same training resources.

    Source: §5 · p. 12
  • Takeaways & Limitations

    The architecture can serve as a basic PINN structure and is compatible with more advanced training processes for complex problems.

    Source: §5 · p. 13

    The supplied 1D and 2D formulations use bounded domains, so these passages do not establish performance beyond those setups.

    Source: 4.1.Allan-Cahn equation · p. 6, 4.2.Helmholtz equation · p. 7

Abstract

from arXiv · show

Although physics-informed neural networks (PINNs) have shown great potential in dealing with nonlinear partial differential equations (PDEs), it is common that PINNs will suffer from the problem of insufficient precision or obtaining incorrect outcomes. Unlike most of the existing solutions trying to enhance the ability of PINN by optimizing the training process, this paper improved the neural network architecture to improve the performance of PINN. We propose a densely multiply PINN (DM-PINN) architecture, which multiplies the output of a hidden layer with the outputs of all the behind hidden layers. Without introducing more trainable parameters, this effective mechanism can significantly improve the accuracy of PINNs. The proposed architecture is evaluated on four benchmark examples (Allan-Cahn equation, Helmholtz equation, Burgers equation and 1D convection equation). Comparisons between the proposed architecture and different PINN structures demonstrate the superior performance of the DM-PINN in both accuracy and efficiency.

1. Introduction

PINNs can produce low-precision or incorrect PDE solutions, while relatively few studies have explored architectural changes. The paper proposes a densely multiplied network structure and reports improved accuracy without extra trainable parameters.

  • PINNs can suffer from low precision or fail to obtain the correct PDE solution.The introduction identifies imbalanced residual distributions and spatial-temporal causality as possible failure sources.Source: §1 · pp. 1–2
  • Prior approaches address residual imbalance through resampling or by assigning higher weights to high-residual points.The cited approaches differ in whether training points are resampled or kept fixed.Source: §1 · p. 2
  • Other prior approaches address spatial-temporal causality by reformulating the loss or training on sequences of time points.The sequential approach uses each sequence's endpoint as the next sequence's initial state.Source: §1 · p. 2
  • Other structural changes include exact boundary-condition imposition and trainable activation parameters.The introduction also describes ResNet-based PINNs and multiplicative interactions, including a design requiring two additional single-layer networks.Source: §1 · pp. 2–3
  • Only a few researchers have explored improving PINNs by changing their neural-network structure.The original PINN simply used a fully connected neural network, although more complex structures may increase expressive power.Source: §1 · p. 2
  • The proposed densely multiplied network structure improves PINN accuracy without introducing extra trainable parameters.Its effectiveness is demonstrated on benchmark PDE problems.Source: §1 · p. 3

2. Physics-informed neural networks

The PINN framework treats a neural-network output as a PDE solution and trains it by incorporating the governing equation and initial and boundary conditions into a loss. Training adjusts the network parameters to minimize that loss.

  • The PDE is stated in a general form, with initial data specified over the spatial domain and boundary data on its boundary.The initial and boundary conditions are given for times from 0 to T.Source: §2 · p. 3
  • The spatial domain, its boundary, space and time coordinates, and the initial and boundary data are identified in the formulation.The formulation also defines the nonlinear operator and its parameter.Source: §2 · p. 3
  • A PINN represents the latent PDE solution with a neural-network output and incorporates the governing equation, initial condition, and boundary conditions into its loss.The loss combines separate terms for the initial condition, boundary conditions, and PDE residual.Source: §2 · pp. 3–4
  • The loss terms have separate weighting coefficients, which are used to address imbalance between the terms.Collocation points are distributed across the space-time domain.Source: §2 · p. 4
  • Training adjusts the network parameters to minimize the loss, typically using a gradient-descent algorithm such as Adam or SGD.The optimization is expressed as minimizing the loss over the trainable parameters.Source: §2 · p. 4

3. Methodology

DM-PINN reuses hidden-layer outputs through element-wise multiplication with later hidden-layer outputs, aiming to improve expressive ability without adding trainable parameters. The method also describes a skip-multiplied ResNet variant to address possible gradient vanishing in deeper networks.

  • DM-PINN reuses hidden-layer outputs by multiplying each element-wise with the outputs of all subsequent hidden layers.This DenseNet-inspired structure is intended to enhance expressive ability without increasing trainable parameters.Source: §3 · pp. 4–5, Figure 1 · p. 5
  • The dense multiplication may cause gradient vanishing as hidden-layer depth increases, so the method adds batch normalization after the input and adopts a ResNet structure.Each layer performs as a residual block in the adopted structure.Source: §3 · p. 5, Figure 2 · p. 5
  • The method modifies dense multiplication into skip multiplication and names the resulting structure SDM-PINN.Figure 3 identifies the skip densely multiplied structure.Source: §3 · p. 5, Figure 3 · p. 5
  • The reported training setup samples data points with Latin Hypercube, initializes weights with Xavier normal distribution, and uses tanh activation.Training uses one NVIDIA GeForce RTX 4090 GPU, and accuracies are averaged over five independent trials per model.Source: Figure 3 · p. 5

4. Numerical examples

Across four PDE examples, the proposed architectures are compared with vanilla PINN and other structures. Reported outcomes range from improved accuracy to uniquely solving the 1D convection case.

4. Numerical examples

5. Discussion

Discussion reports that DM-PINN achieves the best outcomes among four models across tested learning rates and examines its training dynamics through Hessian eigenvalues and loss curves. The authors describe an architecture that reuses hidden-layer outputs to improve expressive power without adding trainable parameters.

5. Discussion

  • DM-PINN has the smallest λmax, indicating less-stiff gradient flow and fewer learning-rate restrictions, though computing Hessian eigenvalues can be costly.The authors note that an optimum learning rate should lie between 0 and 2/λmax, but they do not use that prescription in practice.Source: §5 · p. 11
  • Near the loss minima, DM-PINN shows the lightest oscillations among four models, a stability pattern associated with easier training.The discussion says the L2-error curves verify this training-stability observation.Source: Figure 8 · p. 11, Figure 9 · p. 12
  • Across tested learning rates, DM-PINN achieves the best outcomes among four models, with L2 error decreasing faster than modified MLP's.Table 5 covers L2 errors across models and learning rates; Figure 9 shows loss and L2-error curves.Source: §5 · p. 12, Table 5 · p. 12, Figure 9 · p. 12
  • Element-wise multiplication reuses hidden-layer outputs to enhance expressive power without extra parameters; the architecture is compatible with advanced training processes.The authors report more accurate predictions with the same training resources.Source: §5 · pp. 12–13
Loading 2402.04390v3…