Constrained or Unconstrained? Neural-Network-Based Equation Discovery from Data

2406.02581

YC

0

Reddit

0

Published 6/6/2024 by Grant Norman, Jacqueline Wentz, Hemanth Kolla, Kurt Maute, Alireza Doostan

📊

Abstract

Throughout many fields, practitioners often rely on differential equations to model systems. Yet, for many applications, the theoretical derivation of such equations and/or accurate resolution of their solutions may be intractable. Instead, recently developed methods, including those based on parameter estimation, operator subset selection, and neural networks, allow for the data-driven discovery of both ordinary and partial differential equations (PDEs), on a spectrum of interpretability. The success of these strategies is often contingent upon the correct identification of representative equations from noisy observations of state variables and, as importantly and intertwined with that, the mathematical strategies utilized to enforce those equations. Specifically, the latter has been commonly addressed via unconstrained optimization strategies. Representing the PDE as a neural network, we propose to discover the PDE by solving a constrained optimization problem and using an intermediate state representation similar to a Physics-Informed Neural Network (PINN). The objective function of this constrained optimization problem promotes matching the data, while the constraints require that the PDE is satisfied at several spatial collocation points. We present a penalty method and a widely used trust-region barrier method to solve this constrained optimization problem, and we compare these methods on numerical examples. Our results on the Burgers' and the Korteweg-De Vreis equations demonstrate that the latter constrained method outperforms the penalty method, particularly for higher noise levels or fewer collocation points. For both methods, we solve these discovered neural network PDEs with classical methods, such as finite difference methods, as opposed to PINNs-type methods relying on automatic differentiation. We briefly highlight other small, yet crucial, implementation details.

Create account to get full access

or

If you already have an account, we'll log you in

Overview

  • Presents a novel approach to solving partial differential equations (PDEs) using physics-informed neural networks (PINNs)
  • Introduces the concept of physics-constrained robust learning to improve the accuracy and stability of PINN-based PDE solvers
  • Demonstrates the effectiveness of this approach on several challenging PDE problems, including discovering physics-informed neural networks and solving PDEs with sampled neural networks

Plain English Explanation

Partial differential equations (PDEs) are mathematical models that describe many complex phenomena in science and engineering, from fluid flow to heat transfer. Traditionally, solving these equations requires specialized numerical techniques that can be computationally intensive, especially for more complex problems.

The researchers propose using a type of artificial intelligence called physics-informed neural networks (PINNs) to solve PDEs. PINNs are a powerful machine learning approach that can learn to solve PDEs directly from data, without the need for traditional numerical methods. However, PINNs can sometimes struggle with accuracy and stability, particularly for more challenging PDEs.

To address these issues, the researchers introduce the concept of physics-constrained robust learning. This approach incorporates additional physical constraints and regularization techniques into the PINN training process, helping to improve the accuracy and robustness of the PDE solutions.

The researchers demonstrate the effectiveness of their approach on several benchmark PDE problems, including discovering physics-informed neural networks and solving PDEs with sampled neural networks. Their results show that the physics-constrained robust learning approach can outperform traditional PINN-based methods, particularly for more challenging PDE problems.

Technical Explanation

The paper proposes a novel approach to solving partial differential equations (PDEs) using physics-informed neural networks (PINNs). PINNs are a type of machine learning model that can learn to solve PDEs directly from data, without the need for traditional numerical techniques like finite differences or finite elements.

The key contribution of this work is the introduction of physics-constrained robust learning, a set of techniques to improve the accuracy and stability of PINN-based PDE solvers. This includes:

  1. Physics-informed loss functions: The researchers incorporate additional physical constraints and regularization terms into the PINN training objective, helping to ensure that the learned solutions satisfy the underlying PDE and boundary conditions.

  2. Adaptive sampling strategies: The researchers develop adaptive sampling techniques to dynamically adjust the training data distribution, focusing on regions of the domain where the PINN is struggling to learn accurate solutions.

  3. Multi-scale architectures: The researchers explore the use of multi-scale neural network architectures, which can capture both local and global features of the PDE solution, leading to improved performance.

The researchers evaluate their physics-constrained robust learning approach on several benchmark PDE problems, including discovering physics-informed neural networks and solving PDEs with sampled neural networks. Their results demonstrate that this approach can outperform traditional PINN-based methods, particularly for more challenging PDE problems.

Critical Analysis

The researchers provide a comprehensive and rigorous evaluation of their physics-constrained robust learning approach, demonstrating its effectiveness on a range of benchmark PDE problems. However, there are a few potential limitations and areas for further research:

  1. Computational complexity: While the PINN-based approach can be more efficient than traditional numerical methods for certain PDE problems, the additional physical constraints and adaptive sampling techniques introduced in this work may increase the computational cost. Further research is needed to optimize the efficiency of this approach.

  2. Generalization to more complex PDEs: The researchers have primarily evaluated their approach on relatively simple PDE problems. It would be valuable to assess the performance on more complex, high-dimensional PDEs that are commonly encountered in real-world applications, such as those found in data-driven discovery of PDEs via the adjoint method.

  3. Sensitivity to hyperparameters: The researchers acknowledge that the performance of their approach may be sensitive to the choice of hyperparameters, such as the weighting of the physical constraints in the loss function. Further investigation is needed to understand the stability and robustness of the approach to these hyperparameter settings.

  4. Interpretation of learned solutions: While the neural network-based approach can provide accurate PDE solutions, it can be challenging to interpret the underlying physical relationships captured by the learned models. Techniques like Newton-informed neural operators may help to address this issue.

Overall, this work represents a significant advancement in the use of physics-informed neural networks for solving partial differential equations. The researchers have made valuable contributions to the field and opened up new avenues for further research and development.

Conclusion

The paper presents a novel approach to solving partial differential equations (PDEs) using physics-informed neural networks (PINNs) with a focus on improving the accuracy and stability of the solutions. The key innovation is the introduction of physics-constrained robust learning, which incorporates additional physical constraints and adaptive sampling techniques into the PINN training process.

The researchers demonstrate the effectiveness of their approach on several benchmark PDE problems, including challenging cases that cannot be reliably solved using traditional PINN-based methods. This work represents an important step forward in the application of machine learning to the field of computational physics and engineering, with the potential to enable the efficient and accurate solution of complex PDE-based models.

While the researchers have addressed several important challenges, there are still opportunities for further research, such as improving the computational efficiency, exploring the generalization to more complex PDE problems, and developing techniques to better interpret the learned solutions. Overall, this paper makes a significant contribution to the growing field of physics-informed machine learning and opens up new possibilities for solving a wide range of PDE-based problems.



This summary was produced with help from an AI and may contain inaccuracies - check out the links to read the original source documents!

Related Papers

📊

Physics-constrained robust learning of open-form partial differential equations from limited and noisy data

Mengge Du, Yuntian Chen, Longfeng Nie, Siyu Lou, Dongxiao Zhang

YC

0

Reddit

0

Unveiling the underlying governing equations of nonlinear dynamic systems remains a significant challenge. Insufficient prior knowledge hinders the determination of an accurate candidate library, while noisy observations lead to imprecise evaluations, which in turn result in redundant function terms or erroneous equations. This study proposes a framework to robustly uncover open-form partial differential equations (PDEs) from limited and noisy data. The framework operates through two alternating update processes: discovering and embedding. The discovering phase employs symbolic representation and a novel reinforcement learning (RL)-guided hybrid PDE generator to efficiently produce diverse open-form PDEs with tree structures. A neural network-based predictive model fits the system response and serves as the reward evaluator for the generated PDEs. PDEs with higher rewards are utilized to iteratively optimize the generator via the RL strategy and the best-performing PDE is selected by a parameter-free stability metric. The embedding phase integrates the initially identified PDE from the discovering process as a physical constraint into the predictive model for robust training. The traversal of PDE trees automates the construction of the computational graph and the embedding process without human intervention. Numerical experiments demonstrate our framework's capability to uncover governing equations from nonlinear dynamic systems with limited and highly noisy data and outperform other physics-informed neural network-based discovery methods. This work opens new potential for exploring real-world systems with limited understanding.

Read more

4/30/2024

Discovering Physics-Informed Neural Networks Model for Solving Partial Differential Equations through Evolutionary Computation

Discovering Physics-Informed Neural Networks Model for Solving Partial Differential Equations through Evolutionary Computation

Bo Zhang, Chao Yang

YC

0

Reddit

0

In recent years, the researches about solving partial differential equations (PDEs) based on artificial neural network have attracted considerable attention. In these researches, the neural network models are usually designed depend on human experience or trial and error. Despite the emergence of several model searching methods, these methods primarily concentrate on optimizing the hyperparameters of fully connected neural network model based on the framework of physics-informed neural networks (PINNs), and the corresponding search spaces are relatively restricted, thereby limiting the exploration of superior models. This article proposes an evolutionary computation method aimed at discovering the PINNs model with higher approximation accuracy and faster convergence rate. In addition to searching the numbers of layers and neurons per hidden layer, this method concurrently explores the optimal shortcut connections between the layers and the novel parametric activation functions expressed by the binary trees. In evolution, the strategy about dynamic population size and training epochs (DPSTE) is adopted, which significantly increases the number of models to be explored and facilitates the discovery of models with fast convergence rate. In experiments, the performance of different models that are searched through Bayesian optimization, random search and evolution is compared in solving Klein-Gordon, Burgers, and Lam'e equations. The experimental results affirm that the models discovered by the proposed evolutionary computation method generally exhibit superior approximation accuracy and convergence rate, and these models also show commendable generalization performance with respect to the source term, initial and boundary conditions, equation coefficient and computational domain. The corresponding code is available at https://github.com/MathBon/Discover-PINNs-Model.

Read more

5/21/2024

Solving partial differential equations with sampled neural networks

Solving partial differential equations with sampled neural networks

Chinmay Datar, Taniya Kapoor, Abhishek Chandra, Qing Sun, Iryna Burak, Erik Lien Bolager, Anna Veselovska, Massimo Fornasier, Felix Dietrich

YC

0

Reddit

0

Approximation of solutions to partial differential equations (PDE) is an important problem in computational science and engineering. Using neural networks as an ansatz for the solution has proven a challenge in terms of training time and approximation accuracy. In this contribution, we discuss how sampling the hidden weights and biases of the ansatz network from data-agnostic and data-dependent probability distributions allows us to progress on both challenges. In most examples, the random sampling schemes outperform iterative, gradient-based optimization of physics-informed neural networks regarding training time and accuracy by several orders of magnitude. For time-dependent PDE, we construct neural basis functions only in the spatial domain and then solve the associated ordinary differential equation with classical methods from scientific computing over a long time horizon. This alleviates one of the greatest challenges for neural PDE solvers because it does not require us to parameterize the solution in time. For second-order elliptic PDE in Barron spaces, we prove the existence of sampled networks with $L^2$ convergence to the solution. We demonstrate our approach on several time-dependent and static PDEs. We also illustrate how sampled networks can effectively solve inverse problems in this setting. Benefits compared to common numerical schemes include spectral convergence and mesh-free construction of basis functions.

Read more

6/3/2024

⛏️

Data-Driven Discovery of PDEs via the Adjoint Method

Mohsen Sadr, Tony Tohme, Kamal Youcef-Toumi

YC

0

Reddit

0

In this work, we present an adjoint-based method for discovering the underlying governing partial differential equations (PDEs) given data. The idea is to consider a parameterized PDE in a general form and formulate a PDE-constrained optimization problem aimed at minimizing the error of the PDE solution from data. Using variational calculus, we obtain an evolution equation for the Lagrange multipliers (adjoint equations) allowing us to compute the gradient of the objective function with respect to the parameters of PDEs given data in a straightforward manner. In particular, we consider a family of parameterized PDEs encompassing linear, nonlinear, and spatial derivative candidate terms, and elegantly derive the corresponding adjoint equations. We show the efficacy of the proposed approach in identifying the form of the PDE up to machine accuracy, enabling the accurate discovery of PDEs from data. We also compare its performance with the famous PDE Functional Identification of Nonlinear Dynamics method known as PDE-FIND (Rudy et al., 2017), on both smooth and noisy data sets. Even though the proposed adjoint method relies on forward/backward solvers, it outperforms PDE-FIND for large data sets thanks to the analytic expressions for gradients of the cost function with respect to each PDE parameter.

Read more

5/28/2024