# FREESIA: Covariance-Aware Posterior Transport for Expressive and Scalable Data Assimilation

arXiv: https://arxiv.org/abs/2609.25085 · HTML: https://mlsanity.com/paper/2609.25085/

## Key facts

- No citing papers in the map yet (data through 14 Oct 2026).
- Builds on 13 of its 28 references that are in the map.
- First posted on arXiv 19 Sep 2026.

## Builds on (most cited first, 13 of 13)

- [Flow Matching for Generative Modeling](https://mlsanity.com/paper/2210.02747/) (2022)
- [Improving and generalizing flow-based generative models with minibatch optimal transport](https://mlsanity.com/paper/2302.00482/) (2023)
- [Score-based Data Assimilation](https://mlsanity.com/paper/2306.10574/) (2023)
- [On the Guidance of Flow Matching](https://mlsanity.com/paper/2502.02150/) (2025)
- [An Ensemble Score Filter for Tracking High-Dimensional Nonlinear Dynamical Systems](https://mlsanity.com/paper/2309.00983/) (2023)
- [Data Assimilation: A Mathematical Introduction](https://mlsanity.com/paper/1506.07825/) (2015)
- [Latent-EnSF: A Latent Ensemble Score Filter for High-Dimensional Data Assimilation with Sparse Observation Data](https://mlsanity.com/paper/2409.00127/) (2024)
- [Flow Matching for Efficient and Scalable Data Assimilation](https://mlsanity.com/paper/2508.13313/) (2025)
- [Ensemble score filter with image inpainting for data assimilation in tracking surface quasi-geostrophic dynamics with partial observations](https://mlsanity.com/paper/2501.12419/) (2025)
- [Nonlinear Assimilation via Score-based Sequential Langevin Sampling](https://mlsanity.com/paper/2411.13443/) (2024)
- [Flow-based Bayesian filtering for high-dimensional nonlinear stochastic dynamical systems](https://mlsanity.com/paper/2502.16232/) (2025)
- [A Score-based Diffusion Model Approach for Adaptive Learning of Stochastic Partial Differential Equation Solutions](https://mlsanity.com/paper/2508.06834/) (2025)
- [A Score Filter Enhanced Data Assimilation Framework for Data-Driven Dynamical Systems](https://mlsanity.com/paper/2603.14863/) (2026)

## Abstract (arXiv)

Data assimilation aims to infer the state of complex dynamical systems based on observational data. However, accurate inference of the multimodal posteriors induced by nonlinear or non-injective observation operators remains a key challenge under high-dimensional and sparse observation conditions. Ensemble filters scale to high dimensions but are confined by restrictive distributional assumptions, while training-free generative filters (e.g., EnSF, EnFF) alleviate this limitation but may introduce structural errors and hinder information propagation under sparse observations. To address these issues, we propose a training-free, asymptotically exact posterior transport method. Firstly, a covariance-aware posterior transport scheme is designed, which embeds the forecast cross-covariance into flow-based transport and accurately recovers unobserved states while preserving the non-Gaussian posterior structure. Furthermore, the method combines a tractable observation-adaptive proposal with posterior correction, ensuring accurate approximation of the nonlinear posterior distribution. Finally, we establish the corresponding posterior flow theory, from which the asymptotic exactness of the proposed method relative to finite-ensemble surrogates and the Wasserstein error bound are derived. Experiments on Double-Well, Lorenz-96, and Kolmogorov flow show that the proposed method captures complex posterior structure and remains accurate under sparse, nonlinear, and non-injective observations. In the sparse non-injective setting, it reduces RMSE by 56% relative to the best baseline.
