Assumed Density Filtering and Smoothing with Neural Network Surrogate Models

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Kuang, Simon, Lin, Xinfan
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866917470385733632
author Kuang, Simon
Lin, Xinfan
author_facet Kuang, Simon
Lin, Xinfan
contents The Kalman filter and Rauch-Tung-Striebel (RTS) smoother are optimal for state estimation in linear dynamic systems. With nonlinear systems, the challenge consists in how to propagate uncertainty through the state transitions and output function. For the case of a neural network model, we enable accurate uncertainty propagation using a recent state-of-the-art analytic formula for computing the mean and covariance of a deep neural network with Gaussian input. We argue that cross entropy is a more appropriate performance metric than RMSE for evaluating the accuracy of filters and smoothers. We demonstrate the superiority of our method for state estimation on a stochastic Lorenz system and a Wiener system, and find that our method enables more optimal linear quadratic regulation when the state estimate is used for feedback. Code available at https: //github.com/simontheflutist/analytic-moments.
format Preprint
id arxiv_https___arxiv_org_abs_2511_09016
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Assumed Density Filtering and Smoothing with Neural Network Surrogate Models
Kuang, Simon
Lin, Xinfan
Systems and Control
Machine Learning
The Kalman filter and Rauch-Tung-Striebel (RTS) smoother are optimal for state estimation in linear dynamic systems. With nonlinear systems, the challenge consists in how to propagate uncertainty through the state transitions and output function. For the case of a neural network model, we enable accurate uncertainty propagation using a recent state-of-the-art analytic formula for computing the mean and covariance of a deep neural network with Gaussian input. We argue that cross entropy is a more appropriate performance metric than RMSE for evaluating the accuracy of filters and smoothers. We demonstrate the superiority of our method for state estimation on a stochastic Lorenz system and a Wiener system, and find that our method enables more optimal linear quadratic regulation when the state estimate is used for feedback. Code available at https: //github.com/simontheflutist/analytic-moments.
title Assumed Density Filtering and Smoothing with Neural Network Surrogate Models
topic Systems and Control
Machine Learning
url https://arxiv.org/abs/2511.09016