Spatio-Temporal Uncertainty-Modulated Physics-Informed Neural Networks for Solving Hyperbolic Conservation Laws with Strong Shocks

Abstract

Physics-Informed Neural Networks (PINNs) frequently encounter difficulties in accurately resolving shock waves within high-speed compressible flows, a failure largely attributed to the "gradient pathology" arising from extreme stiffness at discontinuities. To overcome this limitation, we propose the Spatio-Temporal Uncertainty-Modulated PINN (UM-PINN), a probabilistic framework that reinterprets the training process as a multi-task learning problem governed by homoscedastic aleatoric uncertainty. By integrating a gradient-based spatial mask with learnable variance parameters, our method dynamically balances the conflicting contributions of Partial Differential Equation (PDE) residuals and initial conditions across the spatiotemporal domain, further stabilized by Quasi-Monte Carlo Sobol sampling. We validate the framework against challenging benchmarks, including the one-dimensional (1D) Sod shock tube, the high-frequency Shu-Osher problem, and the complex two-dimensional (2D) Riemann interaction, where standard gradient-based weighting schemes typically fail. Experimental results demonstrate that UM-PINN achieves orders of magnitude improvement in accuracy and shock resolution compared to baseline methods, establishing a robust new paradigm for mesh-free Computational Fluid Dynamics in hyperbolic systems.

References in corpus (9)

• DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operators
• NSFnets (Navier-Stokes Flow nets): Physics-informed neural networks for the incompressible Navier-Stokes equations
• Adaptive activation functions accelerate convergence in deep and physics-informed neural networks
• hp-VPINNs: Variational Physics-Informed Neural Networks With Domain Decomposition
• A comprehensive study of non-adaptive and residual-based adaptive sampling for physics-informed neural networks
• Uncertainty Quantification in Scientific Machine Learning: Methods, Metrics, and Comparisons
• A Point-Cloud Deep Learning Framework for Prediction of Fluid Flow Fields on Irregular Geometries
• Discontinuity Computing using Physics-Informed Neural Network
• LSTM-PINN: An Hybrid Method for Prediction of Steady-State Electrohydrodynamic Flow