Generative models improve accuracy and flexibility in nonlinear filtering

Generative models for simulation based filtering: Formulations and Empirical Comparisons

Machine Learning

Summary

The paper looks at ways to better estimate changing hidden states in systems, like tracking, by using generative models. These methods move a predicted guess closer to what the system's observation shows, using different mathematical transports. The authors introduce three new ways to do this and compare them with existing methods, finding that no single method is best but each has strengths depending on available resources. They show that generative models handle complex, multimodal scenarios better than some traditional approaches.

What this means in practice

  • For data assimilation teams: Enhance state estimation accuracy in weather and climate models by choosing generative filters that better capture complex distributions under computational constraints.
  • For autonomous vehicle engineers: Improve tracking of dynamic targets by using generative model filters which can represent multiple possible target states more effectively than common filtering methods.

Authors

Mohammad Al-Jarrah, Wei Deng, Bamdad Hosseini, Amirhossein Taghvaei

Abstract

This letter presents a unified formulation and a controlled numerical comparison of generative-model approaches to the nonlinear filtering problem. Under this formulation the analysis step is realized by a transport of the forecast distribution to the posterior, the approaches differing only in how that transport is selected and learned. We derive three new filters, based on stochastic interpolants, their deterministic flow-matching limit, and Schrödinger bridges realized through forward--backward SDEs. We develop a two-stage tuning procedure that separates the training of the generative model from its online refinement. The resulting methods are compared against the optimal transport filter (OTF), the Knothe--Rosenblatt filter (KRF), the sequential importance resampling (SIR) particle filter and the ensemble Kalman filter (EnKF), in terms of accuracy, computational time, and sensitivity to ensemble size and state dimension. The results indicate that every generative filter resolves multimodal posteriors that the EnKF and SIR do not, that no single generative framework dominates, the preferred method being set by the available online budget and ensemble size, and that the filters differ in the regularity of the particle trajectories they produce.