Optimal State Estimation in the Presence of Non-Gaussian Uncertainty via Wasserstein Distance Minimization
Himanshu Prabhat, Raktim Bhattacharya
TL;DR
This work introduces a distribution-agnostic state-estimation framework that minimizes the $2$-Wasserstein distance between the posterior error distribution $p_{e^+}$ and the Dirac delta $ obreak\delta(e^+)$. The approach yields a Wasserstein filter that naturally recovers the Kalman filter for linear Gaussian systems and derives a Gaussian Sum Filter (GSF) with explicit suboptimality bounds when the prior is a Gaussian Mixture. It further proposes a nonlinear extension, nGSF, which improves estimation accuracy via local optimization. Overall, the transport-based formulation provides a principled, non-particle-based method for fusing information under non-Gaussian uncertainty and can operate on general state manifolds.
Abstract
This paper presents a novel distribution-agnostic Wasserstein distance-based estimation framework. The goal is to determine an optimal map combining prior estimate with measurement likelihood such that posterior estimation error optimally reaches the Dirac delta distribution with minimal effort. The Wasserstein metric is used to quantify the effort of transporting from one distribution to another. We hypothesize that minimizing the Wasserstein distance between the posterior error and the Dirac delta distribution results in optimal information fusion and posterior state uncertainty. Framework validation is demonstrated by the successful recovery of the classical Kalman filter for linear systems with Gaussian uncertainties. Notably, the proposed Wasserstein filter does not rely on particle representation of uncertainty. Furthermore, the classical result for the Gaussian Sum Filter (GSF) is retrieved from the Wasserstein framework. This approach analytically exhibits the suboptimality of GSF and enables the use of nonlinear optimization techniques to enhance the accuracy of the Gaussian sum estimator.
