Solution of the Probabilistic Lambert Problem: Connections with Optimal Mass Transport, Schrödinger Bridge and Reaction-Diffusion PDEs

Alexis M.H. Teter,Iman Nodozi,Abhishek Halder
2024-10-03
Abstract:The Lambert problem originated in orbital mechanics. It concerns with determining the initial velocity for a boundary value problem involving the dynamical constraint due to gravitational potential with additional time horizon and endpoint position constraints. Its solution has application in transferring a spacecraft from a given initial to a given terminal position within prescribed flight time via velocity control. We consider a probabilistic variant of the Lambert problem where the knowledge of the endpoint constraints in position vectors are replaced by the knowledge of their respective joint probability density functions. We show that the Lambert problem with endpoint joint probability density constraints is a generalized optimal mass transport (OMT) problem, thereby connecting this classical astrodynamics problem with a burgeoning area of research in modern stochastic control and stochastic machine learning. This newfound connection allows us to rigorously establish the existence and uniqueness of solution for the probabilistic Lambert problem. The same connection also helps to numerically solve the probabilistic Lambert problem via diffusion regularization, i.e., by leveraging further connection of the OMT with the Schrödinger bridge problem (SBP). This also shows that the probabilistic Lambert problem with additive dynamic process noise is a generalized SBP, and can be solved numerically using the so-called Schrödinger factors, as we do in this work. Our analysis leads to solving a system of reaction-diffusion PDEs where the gravitational potential appears as the reaction rate.
Optimization and Control,Machine Learning,Systems and Control,Mathematical Physics
What problem does this paper attempt to address?
### Problems Attempted to Solve by the Paper This paper attempts to solve the **Probabilistic Lambert Problem**. The traditional Lambert problem in orbital mechanics is a deterministic problem aimed at determining the initial velocity given the initial and terminal positions and the flight time constraint. This paper proposes a probabilistic variant where the terminal position constraint is represented by its joint probability density function (PDF) rather than specific vector values. Specifically, the traditional Lambert problem can be formulated as: \[ \ddot{r} = -\nabla_r V(r), \quad r(t_0) = r_0, \quad r(t_1) = r_1 \] where \( V(r) \) is the gravitational potential, and \( r(t_0) \) and \( r(t_1) \) are the given initial and terminal positions. The probabilistic Lambert problem replaces these terminal position constraints with: \[ r(t_0) \sim \rho_0, \quad r(t_1) \sim \rho_1 \] Here, \( \rho_0 \) and \( \rho_1 \) are the joint probability density functions of the initial and terminal positions, respectively, allowing for random uncertainties. Therefore, the problem becomes how to find an appropriate control strategy \( v(r, t) \) that transfers the spacecraft from the statistical distribution \( \rho_0 \) of the initial position to the statistical distribution \( \rho_1 \) of the terminal position, given these statistical information. ### Main Contributions 1. **Establishing the Connection between the Probabilistic Lambert Problem and Optimal Mass Transport (OMT)**: - The authors demonstrate that the probabilistic Lambert problem can be viewed as a generalized optimal mass transport problem. - This connection allows for a rigorous proof of the existence and uniqueness of the solution to the probabilistic Lambert problem. 2. **Further Extension to Stochastic Dynamics**: - When the control equation includes process noise, i.e., \( \dot{r} = v(r, t) \) is replaced by \( dr = v(r, t) dt + \sqrt{2\varepsilon} dw(t) \), the problem can be further extended to a generalized Schrödinger bridge problem (SBP). - This extension allows for the use of methods such as Schrödinger factors to numerically solve the probabilistic Lambert problem. 3. **Derivation of Reaction-Diffusion Partial Differential Equations**: - The authors derive a set of coupled reaction-diffusion partial differential equations where the gravitational potential acts as the reaction rate. - A novel algorithm is proposed to numerically solve this set of partial differential equations and their boundary conditions. 4. **Non-Parametric Computational Methods**: - Through these newly discovered connections, the authors demonstrate how to numerically solve the probabilistic Lambert problem using non-parametric computational methods, avoiding the need for parametric assumptions on nonlinear dynamics or statistics. In summary, by linking the probabilistic Lambert problem with optimal mass transport and Schrödinger bridge problems, this paper not only provides theoretical guarantees but also proposes practical numerical solution methods, thereby offering a new framework for dealing with stochastic uncertainties in orbital mechanics.