Uncertainty Propagation in Stochastic Systems via Mixture Models with Error Quantification
Eduardo Figueiredo, Andrea Patane, Morteza Lahijanian, Luca Laurenti
TL;DR
The paper addresses uncertainty propagation in discrete-time non-linear stochastic systems, where exact distributions are typically intractable. It proposes a stochastic-approximation framework that represents the system distribution at each time as a finite mixture $\hat{\mathbb{P}}_{x_t}$ of one-step kernels, with formal total variation ($TV$) distance guarantees to the true distribution $\mathbb{P}_{x_t}$ over a finite horizon $T$ and threshold $\delta$. The core methods include a partition-based propagation scheme, explicit $TV$ bounds that are computable in closed form for Gaussian noise, and an adaptive grid refinement algorithm that concentrates components in high-mass regions to minimize the bound. Theoretical results show convergence of the approximation as the partition becomes finer, and numerical experiments on control benchmarks (including a Dubin's car model) and chance-constrained planning demonstrate improved bound tightness versus uniform grids and practical applicability to safety-critical tasks. Overall, the framework enables tractable, provably correct uncertainty propagation with quantifiable guarantees that support planning and certification under uncertainty.
Abstract
Uncertainty propagation in non-linear dynamical systems has become a key problem in various fields including control theory and machine learning. In this work we focus on discrete-time non-linear stochastic dynamical systems. We present a novel approach to approximate the distribution of the system over a given finite time horizon with a mixture of distributions. The key novelty of our approach is that it not only provides tractable approximations for the distribution of a non-linear stochastic system, but also comes with formal guarantees of correctness. In particular, we consider the total variation (TV) distance to quantify the distance between two distributions and derive an upper bound on the TV between the distribution of the original system and the approximating mixture distribution derived with our framework. We show that in various cases of interest, including in the case of Gaussian noise, the resulting bound can be efficiently computed in closed form. This allows us to quantify the correctness of the approximation and to optimize the parameters of the resulting mixture distribution to minimize such distance. The effectiveness of our approach is illustrated on several benchmarks from the control community.
