Jay McMahon

IM
h-index21
3papers
12citations
Novelty65%
AI Score42

3 Papers

11.2SYJun 3
Training with Hard Constraints: Learning Neural Certificates and Controllers for SDEs

Chun-Wei Kong, Sebastian Escobar, Ibon Gracia et al.

Due to their expressive power, neural networks (NNs) are promising templates for functional optimization problems, particularly for reach-avoid certificate generation for systems governed by stochastic differential equations (SDEs). However, ensuring hard-constraint satisfaction remains a major challenge. In this work, we propose two constraint-driven training frameworks with guarantees for supermartingale-based neural certificate construction and controller synthesis for SDEs. The first approach enforces certificate inequalities via domain discretization and a bound-based loss, guaranteeing global validity once the loss reaches zero. We show that this method also enables joint NN controller-certificate synthesis with hard guarantees. For high-dimensional systems where discretization becomes prohibitive, we introduce a partition-free, scenario-based training method that provides arbitrarily tight PAC guarantees for certificate constraint satisfaction. Benchmarks demonstrate scalability of the bound-based method up to 5D, outperforming the state of the art, and scalability of the scenario-based approach to at least 10D with high-confidence guarantees.

LGOct 28, 2024
Error Bounds for Physics-Informed Neural Networks in Fokker-Planck PDEs

Chun-Wei Kong, Luca Laurenti, Jay McMahon et al.

Stochastic differential equations are commonly used to describe the evolution of stochastic processes. The state uncertainty of such processes is best represented by the probability density function (PDF), whose evolution is governed by the Fokker-Planck partial differential equation (FP-PDE). However, it is generally infeasible to solve the FP-PDE in closed form. In this work, we show that physics-informed neural networks (PINNs) can be trained to approximate the solution PDF. Our main contribution is the analysis of PINN approximation error: we develop a theoretical framework to construct tight error bounds using PINNs. In addition, we derive a practical error bound that can be efficiently constructed with standard training methods. We discuss that this error-bound framework generalizes to approximate solutions of other linear PDEs. Empirical results on nonlinear, high-dimensional, and chaotic systems validate the correctness of our error bounds while demonstrating the scalability of PINNs and their significant computational speedup in obtaining accurate PDF solutions compared to the Monte Carlo approach.

IMFeb 13, 2022
Gravity Estimation at Small Bodies via Optical Tracking of Hopping Artificial Probes

Jacopo Villa, Andrew French, Jay McMahon et al.

Despite numerous successful missions to small celestial bodies, the gravity field of such targets has been poorly characterized so far. Gravity estimates can be used to infer the internal structure and composition of small bodies and, as such, have strong implications in the fields of planetary science, planetary defense, and in-situ resource utilization. Current gravimetry techniques at small bodies mostly rely on tracking the spacecraft orbital motion, where the gravity observability is low. To date, only lower-degree and order spherical harmonics of small-body gravity fields could be resolved. In this paper, we evaluate gravimetry performance for a novel mission architecture where artificial probes repeatedly hop across the surface of the small body and perform low-altitude, suborbital arcs. Such probes are tracked using optical measurements from the mothership's onboard camera and orbit determination is performed to estimate the probe trajectories, the small body's rotational kinematics, and the gravity field. The suborbital motion of the probes provides dense observations at low altitude, where the gravity signal is stronger. We assess the impact of observation parameters and mission duration on gravity observability. Results suggest that the gravitational spherical harmonics of a small body with the same mass as the asteroid Bennu, can be observed at least up to degree 40 within months of observations. Measurement precision and frequency are key to achieve high-performance gravimetry.