Machine Learning for Conservative-to-Primitive in Relativistic Hydrodynamics

Abstract

The numerical solution of relativistic hydrodynamics equations in conservative form requires root-finding algorithms that invert the conservative-to-primitive variables map. These algorithms employ the equation of state of the fluid and can be computationally demanding for applications involving sophisticated microphysics models, such as those required to calculate accurate gravitational wave signals in numerical relativity simulations of binary neutron stars. This work explores the use of machine learning methods to speed up the recovery of primitives in relativistic hydrodynamics. Artificial neural networks are trained to replace either the interpolations of a tabulated equation of state or directly the conservative-to-primitive map. The application of these neural networks to simple benchmark problems shows that both approaches improve over traditional root finders with tabular equation-of-state and multi-dimensional interpolations. In particular, the neural networks for the conservative-to-primitive map accelerate the variable recovery by more than an order of magnitude over standard methods while maintaining accuracy. Neural networks are thus an interesting option to improve the speed and robustness of relativistic hydrodynamics algorithms.

17 pages, 12 figures, 2 tables

References in corpus (18)

• Array Programming with NumPy
• Deep Learning in Neural Networks: An Overview
• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• Explosion Mechanisms of Core-Collapse Supernovae
• A high-bias, low-variance introduction to Machine Learning for physicists
• Relativistic Jets in Active Galactic Nuclei
• Binary Neutron Star Mergers: Mass Ejection, Electromagnetic Counterparts and Nucleosynthesis
• The dynamical mass ejection from binary neutron star mergers: Radiation-hydrodynamics study in general relativity
• Primitive Variable Solvers for Conservative General Relativistic Magnetohydrodynamics
• A New Open-Source Code for Spherically-Symmetric Stellar Collapse to Neutron Stars and Black Holes
• The Dynamics of Binary Neutron Star Mergers and of GW170817
• Impact of an improved neutrino energy estimate on outflows in neutron star merger simulations
• Numerical Relativity Simulations of the Neutron Star Merger GW170817: Long-Term Remnant Evolutions, Winds, Remnant Disks, and Nucleosynthesis
• Accretion-induced prompt black hole formation in asymmetric neutron star mergers, dynamical ejecta and kilonova signals
• Neutron Star Merger Remnants
• Robust Recovery of Primitive Variables in Relativistic Ideal Magnetohydrodynamics
• Recovery schemes for primitive variables in general-relativistic magnetohydrodynamics
• Analytic solutions of the Riemann problem in relativistic hydrodynamics and their numerical applications

Cited by in corpus (7)

• Second release of the CoRe database of binary neutron star merger waveforms
• Applications of machine learning in gravitational wave research with current interferometric detectors
• cuHARM : a new GPU accelerated GR-MHD code and its application to ADAF disks
• An Explicit Primitive Conservative Solver for the Euler Equations with Arbitrary Equation of State
• R-process heating implementation in hydrodynamic simulations with neural networks
• Machine learning-driven conservative-to-primitive conversion in hybrid piecewise polytropic and tabulated equations of state
• APRIL: Auxiliary Physically-Redundant Information in Loss -- A physics-informed framework for parameter estimation with a gravitational-wave case study