Solving the wave equation with physics-informed deep learning

Abstract

We investigate the use of Physics-Informed Neural Networks (PINNs) for solving the wave equation. Whilst PINNs have been successfully applied across many physical systems, the wave equation presents unique challenges due to the multi-scale, propagating and oscillatory nature of its solutions, and it is unclear how well they perform in this setting. We use a deep neural network to learn solutions of the wave equation, using the wave equation and a boundary condition as direct constraints in the loss function when training the network. We test the approach by solving the 2D acoustic wave equation for spatially-varying velocity models of increasing complexity, including homogeneous, layered and Earth-realistic models, and find the network is able to accurately simulate the wavefield across these cases. By using the physics constraint in the loss function the network is able to solve for the wavefield far outside of its boundary training data, offering a way to reduce the generalisation issues of existing deep learning approaches. We extend the approach for the Earth-realistic case by conditioning the network on the source location and find that it is able to generalise over this initial condition, removing the need to retrain the network for each solution. In contrast to traditional numerical simulation this approach is very efficient when computing arbitrary space-time points in the wavefield, as once trained the network carries out inference in a single step without needing to compute the entire wavefield. We discuss the potential applications, limitations and further research directions of this work.

13 pages, 9 figures

Cited by in corpus (19)

• Physics-informed Neural Networks (PINNs) for Wave Propagation and Full Waveform Inversions
• Finite Basis Physics-Informed Neural Networks (FBPINNs): a scalable domain decomposition approach for solving differential equations
• Solving the acoustic VTI wave equation using physics-informed neural networks
• A Critical Review of Physics-Informed Machine Learning Applications in Subsurface Energy Systems
• Fourier Neural Operator Surrogate Model to Predict 3D Seismic Waves Propagation
• Finite electro-elasticity with physics-augmented neural networks
• Physics-informed neural networks for one-dimensional sound field predictions with parameterized sources and impedance boundaries
• Physical Activation Functions (PAFs): An Approach for More Efficient Induction of Physics into Physics-Informed Neural Networks (PINNs)
• PINNtomo: Seismic tomography using physics-informed neural networks
• Data-aided Underwater Acoustic Ray Propagation Modeling
• On the Use of Neural Networks for Full Waveform Inversion
• RBF-MGN:Solving spatiotemporal PDEs with Physics-informed Graph Neural Network
• Physics-informed neural network for acoustic resonance analysis in a one-dimensional acoustic tube
• Accelerating Bayesian microseismic event location with deep learning
• Solving Partial Differential Equations with Point Source Based on Physics-Informed Neural Networks
• Towards fast machine-learning-assisted Bayesian posterior inference of microseismic event location and source mechanism
• About optimal loss function for training physics-informed neural networks under respecting causality
• Seismic wave propagation and inversion with Neural Operators
• DynaBench: A benchmark dataset for learning dynamical systems from low-resolution data