Publications

2025

1. 

   DAISI: Data Assimilation with Inverse Sampling using Stochastic Interpolants

   Martin Andrae, Erik Larsson, So Takao, Tomas Landelius, and Fredrik Lindsten

   2025

   Abs Bib PDF

   Data assimilation (DA) is a cornerstone of scientific and engineering applications, combining model forecasts with sparse and noisy observations to estimate latent system states. Classical DA methods, such as the ensemble Kalman filter, rely on Gaussian approximations and heuristic tuning (e.g., inflation and localization) to scale to high dimensions. While often successful, these approximations can make the methods unstable or inaccurate when the underlying distributions of states and observations depart significantly from Gaussianity. To address this limitation, we introduce DAISI, a scalable filtering algorithm built on flow-based generative models that enables flexible probabilistic inference using data-driven priors. The core idea is to use a stationary, pre-trained generative prior to assimilate observations via guidance-based conditional sampling while incorporating forecast information through a novel inverse-sampling step. This step maps the forecast ensemble into a latent space to provide initial conditions for the conditional sampling, allowing us to encode model dynamics into the DA pipeline without having to retrain or fine-tune the generative prior at each assimilation step. Experiments on challenging nonlinear systems show that DAISI achieves accurate filtering results in regimes with sparse, noisy, and nonlinear observations where traditional methods struggle.

   @misc{andrae2025daisi,
     title = {DAISI: Data Assimilation with Inverse Sampling using Stochastic Interpolants},
     author = {Andrae, Martin and Larsson, Erik and Takao, So and Landelius, Tomas and Lindsten, Fredrik},
     year = {2025},
     eprint = {2512.00252},
     archiveprefix = {arXiv},
     primaryclass = {stat.ML},
   }

2. 

   Continuous Ensemble Weather Forecasting with Diffusion models

   Martin Andrae, Tomas Landelius, Joel Oskarsson, and Fredrik Lindsten

   In The Thirteenth International Conference on Learning Representations, 2025

   Abs Bib PDF Blog Code Poster

   Weather forecasting has seen a shift in methods from numerical simulations to data-driven systems. While initial research in the area focused on deterministic forecasting, recent works have used diffusion models to produce skillful ensemble forecasts. These models are trained on a single forecasting step and rolled out autoregressively. However, they are computationally expensive and accumulate errors for high temporal resolution due to the many rollout steps. We address these limitations with Continuous Ensemble Forecasting, a novel and flexible method for sampling ensemble forecasts in diffusion models. The method can generate temporally consistent ensemble trajectories completely in parallel, with no autoregressive steps. Continuous Ensemble Forecasting can also be combined with autoregressive rollouts to yield forecasts at an arbitrary fine temporal resolution without sacrificing accuracy. We demonstrate that the method achieves competitive results for global weather forecasting with good probabilistic properties.

   @inproceedings{andrae2025continuous,
     title = {Continuous Ensemble Weather Forecasting with Diffusion models},
     author = {Andrae, Martin and Landelius, Tomas and Oskarsson, Joel and Lindsten, Fredrik},
     booktitle = {The Thirteenth International Conference on Learning Representations},
     year = {2025},
   }