Rethinking Forward Processes for Score-Based Data Assimilation in High Dimensions
- URL: http://arxiv.org/abs/2604.02889v1
- Date: Fri, 03 Apr 2026 08:55:38 GMT
- Title: Rethinking Forward Processes for Score-Based Data Assimilation in High Dimensions
- Abstract summary: We propose a measurement-aware score-based filter (MASF) that defines a measurement-aware forward process directly from the measurement equation.<n>For linear measurements, we derive the exact likelihood score and combine it with a learned prior score to obtain the posterior score.<n> Numerical experiments covering a range of settings, including high-dimensional datasets, demonstrate improved accuracy and stability over existing score-based filters.
- Score: 2.3550058325130974
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Data assimilation is the process of estimating the time-evolving state of a dynamical system by integrating model predictions and noisy observations. It is commonly formulated as Bayesian filtering, but classical filters often struggle with accuracy or computational feasibility in high dimensions. Recently, score-based generative models have emerged as a scalable approach for high-dimensional data assimilation, enabling accurate modeling and sampling of complex distributions. However, existing score-based filters often specify the forward process independently of the data assimilation. As a result, the measurement-update step depends on heuristic approximations of the likelihood score, which can accumulate errors and degrade performance over time. Here, we propose a measurement-aware score-based filter (MASF) that defines a measurement-aware forward process directly from the measurement equation. This construction makes the likelihood score analytically tractable: for linear measurements, we derive the exact likelihood score and combine it with a learned prior score to obtain the posterior score. Numerical experiments covering a range of settings, including high-dimensional datasets, demonstrate improved accuracy and stability over existing score-based filters.
Related papers
- Dynamic Online Processor-Native Inference for State Estimation [0.0]
Sensor-rich data-driven applications increasingly use Bayesian approaches to infer latent states of dynamic systems from noisy sensor measurements and physical models.<n>This paper presents a Bayesian filtering technique that uses processor-native uncertainty tracking for both uncertainty propagation and inference.<n> Benchmarks across three nonlinear state-space systems compare the approach against particle filters and Monte-Carlo-based likelihood estimators.
arXiv Detail & Related papers (2026-07-13T19:13:45Z) - Ensemble Score Filtering for Real-Data Energy Consumption Forecast Correction [11.027420128235903]
We study a high-dimensional data assimilation problem for real energy-consumption data.<n>We employ the Ensemble Score Filter (EnSF) to assimilate partial and noisy observations and correct the forecast trajectory over time.
arXiv Detail & Related papers (2026-05-27T20:27:38Z) - Denoising Particle Filters: Learning State Estimation with Single-Step Objectives [2.320417845168326]
We propose a novel particle filtering algorithm in which models are trained from individual state transitions.<n>We evaluate the proposed method on challenging robotic state estimation tasks in simulation.
arXiv Detail & Related papers (2026-02-23T09:53:23Z) - Tempering the Bayes Filter towards Improved Model-Based Estimation [0.5599792629509229]
We develop the tempered Bayes filter, improving estimation performance.<n>A region of the tempering space can be understood as interpolating between the Bayes- and MAP filters.<n>Specializing the results to the linear Gaussian case, we obtain the tempered Kalman filter.
arXiv Detail & Related papers (2025-12-02T14:31:10Z) - DAISI: Data Assimilation with Inverse Sampling using Stochastic Interpolants [12.587156528707796]
We introduce DAISI, a scalable filtering algorithm built on flow-based generative models.<n>We show that DAISI achieves accurate filtering results in regimes with sparse, noisy, and nonlinear observations.
arXiv Detail & Related papers (2025-11-29T00:02:45Z) - Learning IMM Filter Parameters from Measurements using Gradient Descent [45.335821132209766]
In intrinsic parameters of targets under track can be completely unobservable until the system is deployed.
With state-of-the-art sensor systems growing more and more complex, the number of parameters naturally increases.
In this paper, the parameters of an interacting multiple model (IMM) filter are optimized solely using measurements.
arXiv Detail & Related papers (2023-07-13T08:35:40Z) - Score-based Continuous-time Discrete Diffusion Models [102.65769839899315]
We extend diffusion models to discrete variables by introducing a Markov jump process where the reverse process denoises via a continuous-time Markov chain.
We show that an unbiased estimator can be obtained via simple matching the conditional marginal distributions.
We demonstrate the effectiveness of the proposed method on a set of synthetic and real-world music and image benchmarks.
arXiv Detail & Related papers (2022-11-30T05:33:29Z) - Nonlinear Isometric Manifold Learning for Injective Normalizing Flows [58.720142291102135]
We use isometries to separate manifold learning and density estimation.
We also employ autoencoders to design embeddings with explicit inverses that do not distort the probability distribution.
arXiv Detail & Related papers (2022-03-08T08:57:43Z) - Evaluating State-of-the-Art Classification Models Against Bayes
Optimality [106.50867011164584]
We show that we can compute the exact Bayes error of generative models learned using normalizing flows.
We use our approach to conduct a thorough investigation of state-of-the-art classification models.
arXiv Detail & Related papers (2021-06-07T06:21:20Z) - Scalable Marginal Likelihood Estimation for Model Selection in Deep
Learning [78.83598532168256]
Marginal-likelihood based model-selection is rarely used in deep learning due to estimation difficulties.
Our work shows that marginal likelihoods can improve generalization and be useful when validation data is unavailable.
arXiv Detail & Related papers (2021-04-11T09:50:24Z) - Multitarget Tracking with Transformers [21.81266872964314]
Multitarget Tracking (MTT) is a problem of tracking the states of an unknown number of objects using noisy measurements.
In this paper, we propose a high-performing deep-learning method for MTT based on the Transformer architecture.
arXiv Detail & Related papers (2021-04-01T19:14:55Z) - On Maximum Likelihood Training of Score-Based Generative Models [17.05208572228308]
We show that an objective is equivalent to maximum likelihood for certain choices of mixture weighting.
We show that both maximum likelihood training and test-time log-likelihood evaluation can be achieved through parameterization of the score function alone.
arXiv Detail & Related papers (2021-01-22T18:22:29Z) - Self-Tuning Stochastic Optimization with Curvature-Aware Gradient
Filtering [53.523517926927894]
We explore the use of exact per-sample Hessian-vector products and gradients to construct self-tuning quadratics.
We prove that our model-based procedure converges in noisy gradient setting.
This is an interesting step for constructing self-tuning quadratics.
arXiv Detail & Related papers (2020-11-09T22:07:30Z) - Variable Skipping for Autoregressive Range Density Estimation [84.60428050170687]
We show a technique, variable skipping, for accelerating range density estimation over deep autoregressive models.
We show that variable skipping provides 10-100$times$ efficiency improvements when targeting challenging high-quantile error metrics.
arXiv Detail & Related papers (2020-07-10T19:01:40Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.