Generative Bayesian Filtering for State Estimation

· Source: stat.ML updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Data Science & Analytics · Depth: Expert, extended

Summary

Generative Bayesian Filtering (GBF) is a novel framework for inferring latent states from high-dimensional, noisy sensor signals in dynamic systems. It replaces traditional linear-Gaussian observation models, common in Kalman filters, with pretrained conditional generative models, specifically Conditional Variational Autoencoders (CVAE). GBF employs a Bayesian prediction-update recursion, where the measurement update is formulated as a score-based posterior sampling problem, combining dynamical priors with CVAE-induced likelihoods. This approach enhances flexibility and uncertainty quantification. Experiments on synthetic temporal MNIST and Fashion-MNIST datasets, along with real-world applications like Laser Powder Bed Fusion (LPBF) manufacturing monitoring and ECG-based arrhythmia diagnosis, demonstrate GBF's improved state estimation accuracy and robustness, particularly against noise and distribution shifts, outperforming several baseline methods.

Key takeaway

For Machine Learning Engineers developing robust state estimation systems from high-dimensional, noisy sensor data, consider Generative Bayesian Filtering (GBF). Its CVAE-based generative observation model and score-based posterior sampling offer superior accuracy and uncertainty quantification compared to traditional or discriminative methods, especially under distribution shifts. You should evaluate GBF for applications like industrial process monitoring or medical diagnostics to enhance reliability and adaptivity.

Key insights

GBF integrates CVAE-based generative models into Bayesian filtering for robust state estimation from high-dimensional, noisy sensor data.

Principles

Method

GBF uses a CVAE decoder as a flexible likelihood model, then applies score-based posterior sampling via Stochastic Gradient Langevin Dynamics (SGLD) to combine dynamical priors with observations for recursive state updates.

In practice

Topics

Code references

Best for: AI Scientist, Machine Learning Engineer, Research Scientist

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by stat.ML updates on arXiv.org.