Physics-informed neural networks for modeling astrophysical shocks

Abstract

Physics-Informed Neural Networks (PINNs) are machine learning models that integrate data-based learning with partial differential equations (PDEs). In this work, for the first time we extend PINNs to model the numerically challenging case of astrophysical shock waves in the presence of a stellar gravitational field. Notably, PINNs suffer from competing losses during gradient descent that can lead to poor performance especially in physical setups involving multiple scales, which is the case for shocks in the gravitationally stratified solar atmosphere. We applied PINNs in three different setups ranging from modeling astrophysical shocks in cases with no or little data to data-intensive cases. Namely, we used PINNs (a) to determine the effective polytropic index controlling the heating mechanism of the space plasma within 1% error, (b) to quantitatively show that data assimilation is seamless in PINNs and small amounts of data can significantly increase the model’s accuracy, and (c) to solve the forward time-dependent problem for different temporal horizons. We addressed the poor performance of PINNs through an effective normalization
approach by reformulating the fluid dynamics PDE system to absorb the gravity- caused variability. This led to a huge improvement in the overall model performance with the density accuracy improving between 2 and 16 times. Finally, we present a detailed critique on the strengths and drawbacks of PINNs in tackling realistic physical problems in astrophysics and conclude that PINNs can be a powerful complimentary modeling approach to classical fluid dynamics solvers.

GRINN: a physics-informed neural network for solving hydrodynamic systems in the presence of self-gravity

“…As a practical matter, FD codes employ various shock-capturing algorithms [48] and/or artificial viscosity terms to reliably model a shock front. Similarly, for PINN-based models, shock-capturing techniques such as clustered collocation points [32], adaptive artificial viscosity [10], and additional input data [35] have been shown to improve performance on shocks. This will be a fruitful topic to explore in follow-up work on the details of shock modeling, but is not our main topic of interest in this work.…”