Physics informed neural networks improve earthquake wave simulations
Solving the Elastic Wave Equation with Physics-Informed Neural Networks: A Robust and Critical Assessment
Machine Learning
Summary
Solving equations that describe how seismic waves move through the Earth is hard because traditional methods can be slow or complicated. The authors studied a type of AI called physics-informed neural networks (PINNs) that learn solutions while respecting physical laws, avoiding some typical problems. They tested different designs and found that adding parts based on wave physics made the AI much more accurate. This improvement could help with faster and better understanding of earthquakes and seismic hazards.
Physics-Informed Neural NetworksPartial Differential EquationsElastic Wave EquationSeismologyNeural Network ArchitectureSpectral BiasMeshfree MethodsAcoustic Wave EquationSeismic SourcesSeismic Hazard Detection
Authors
Davide Staub, Ben Moseley
Abstract
Physics-Informed Neural Networks (PINNs) have recently emerged as a promising approach for solving Partial Differential Equations (PDEs), offering a meshfree alternative that integrates physical principles into the learning process. This presents a new paradigm compared to traditional discretization methods and purely data-driven machine learning techniques. While promising, PINNs are not a panacea; they inherit challenges such as spectral bias and unstable convergence. Moreover, their potential in seismology remains largely unexplored. In this work, we provide a robust and critical assessment of PINNs for solving the elastic wave equation in seismology. We investigate the performance of PINNs on problems with varying degrees of complexity across various seismic sources and parameter models, from constant to highly heterogeneous settings. A pivotal aspect of our work involves investigating whether embedding physical principles directly into the network architecture enhances convergence and accuracy. We test an extensive range of neural architecture designs, from unrestricted, uninformed PINNs to highly specialized ones. We find that integrating an understanding of wave physics into the network design significantly improves accuracy. For instance, introducing a custom wavelet or plane wave layer, coupled with encoder and decoder layers, consistently yields a relative $L_2$ error approximately half that of the standard PINN, as evidenced across numerous experiments. We further demonstrate that this novel architecture enhances accuracy when applied to the acoustic wave equation, underlying the versatility of our network. Another key contribution of our research is the successful conditioning of PINNs on seismic source locations. This signifies a considerable advancement towards rapid seismic hazard detection and seismic analysis.