The Score Kalman Filter

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Iwasaki, Kaito, Bloch, Anthony, Lee, Taeyoung, Ghaffari, Maani
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917502589599744
author Iwasaki, Kaito
Bloch, Anthony
Lee, Taeyoung
Ghaffari, Maani
author_facet Iwasaki, Kaito
Bloch, Anthony
Lee, Taeyoung
Ghaffari, Maani
contents A central obstacle in nonlinear Bayesian filtering is representing the belief distribution. Moment-based filters address this by propagating polynomial moments and reconstructing a density from them. Recent work completes the predict-update loop via the maximum-entropy (MaxEnt) principle, but each step requires the partition function and its gradient, both $n$-dimensional integrals whose cost scales exponentially, restricting the demonstrated MaxEnt moment filtering to $n \le 4$. We avoid the partition function entirely by combining score matching with Stein's identity. In our setting, score matching reduces the density fit to a single linear solve whose coefficients are assembled directly from the propagated moments. The same parameters then drive Stein's identity to close the moment hierarchy during prediction and to recover posterior moments after each Bayesian update, keeping the full predict-update loop free of partition function evaluation. The resulting Score Kalman Filter (SKF) reduces to the classical information-form Kalman filter as a special case and performs every step through linear algebra. On nonlinear coupled-oscillator networks, the SKF runs through $n=20$ and reports lower RMSE than the EKF, UKF, EnKF, and particle-filter baselines on the tested synthetic benchmarks.
format Preprint
id arxiv_https___arxiv_org_abs_2605_16644
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle The Score Kalman Filter
Iwasaki, Kaito
Bloch, Anthony
Lee, Taeyoung
Ghaffari, Maani
Systems and Control
Machine Learning
Optimization and Control
93E11, 93E10, 62M20, 60H10
A central obstacle in nonlinear Bayesian filtering is representing the belief distribution. Moment-based filters address this by propagating polynomial moments and reconstructing a density from them. Recent work completes the predict-update loop via the maximum-entropy (MaxEnt) principle, but each step requires the partition function and its gradient, both $n$-dimensional integrals whose cost scales exponentially, restricting the demonstrated MaxEnt moment filtering to $n \le 4$. We avoid the partition function entirely by combining score matching with Stein's identity. In our setting, score matching reduces the density fit to a single linear solve whose coefficients are assembled directly from the propagated moments. The same parameters then drive Stein's identity to close the moment hierarchy during prediction and to recover posterior moments after each Bayesian update, keeping the full predict-update loop free of partition function evaluation. The resulting Score Kalman Filter (SKF) reduces to the classical information-form Kalman filter as a special case and performs every step through linear algebra. On nonlinear coupled-oscillator networks, the SKF runs through $n=20$ and reports lower RMSE than the EKF, UKF, EnKF, and particle-filter baselines on the tested synthetic benchmarks.
title The Score Kalman Filter
topic Systems and Control
Machine Learning
Optimization and Control
93E11, 93E10, 62M20, 60H10
url https://arxiv.org/abs/2605.16644