The Physics-Informed Neural Network Gravity Model: Generation III
John Martin, Hanspeter Schaub
TL;DR
This work introduces PINN-GM-III, a robust third-generation physics-informed neural network gravity model designed to overcome feature divergence, altitude-bias, numerical instability, and extrapolation errors in gravity field modeling. It achieves this through a sequence of architectural innovations: 5D spherical input features, a loss function augmented with relative error to balance high- and low-altitude accuracy, learned numerics via a proxy potential, smooth boundary-condition transitions, and fusion with low-fidelity analytic models. A comprehensive benchmarking suite with six metrics demonstrates that PINN-GM-III attains superior robustness and accuracy on heterogeneous-density asteroids (e.g., Eros, Bennu) compared to analytic models and prior ML methods, while maintaining competitive inference speed. The results establish PINN-GM-III as a strong, data-efficient approach for irregular-body gravity fields and outline a path for integrating analytic priors and physics constraints in future planetary science applications.
Abstract
Scientific machine learning and the advent of the Physics-Informed Neural Network (PINN) have shown high potential in their ability to solve complex differential equations. One example is the use of PINNs to solve the gravity field modeling problem -- learning convenient representations of the gravitational potential from position and acceleration data. These PINN gravity models, or PINN-GMs, have demonstrated advantages in model compactness, robustness to noise, and sample efficiency when compared to popular alternatives; however, further investigation has revealed various failure modes for these and other machine learning gravity models which this manuscript aims to address. Specifically, this paper introduces the third generation Physics-Informed Neural Network Gravity Model (PINN-GM-III) which includes design changes that solve the problems of feature divergence, bias towards low-altitude samples, numerical instability, and extrapolation error. Six evaluation metrics are proposed to expose these past pitfalls and illustrate the PINN-GM-III's robustness to them. This study concludes by evaluating the PINN-GM-III modeling accuracy on a heterogeneous density asteroid, and comparing its performance to other analytic and machine learning gravity models.
