Data Assimilation with ML
Data assimilation blends a physical model with streaming observations to track a system's state, and ML sharpens every part of the loop.
Combining model and measurement
Data assimilation is the art of keeping a model in step with reality. A physical model predicts how a system evolves; sparse, noisy observations arrive over time; assimilation merges the two to produce the best current estimate of the state, along with its uncertainty. Weather forecasting is the classic example, correcting a simulation with each new round of measurements.
Classical methods
The Kalman filter and its ensemble variants alternate a forecast step, which advances the state and its uncertainty through the model, with an update step, which nudges the estimate toward the observations in proportion to their relative reliability. Variational methods instead find the trajectory that best fits both the model and a window of observations, using adjoint gradients to do so.
Where machine learning helps
- Learned surrogates replace an expensive forecast model in the ensemble
- Networks learn model error, correcting systematic biases from data
- Learned observation operators map complex measurements to model variables
- End-to-end learning tunes the whole assimilation scheme against outcomes
Handling uncertainty
Assimilation is inherently probabilistic: it tracks not just a state but a distribution over states. Machine learning fits naturally here through ensembles and Bayesian methods, which represent and propagate that uncertainty. A well-designed system reports how confident it is, growing its uncertainty between observations and shrinking it when data arrive.
Relevance to operations
Any system that runs while being measured can use assimilation to fuse a physics model with live sensor data for state estimation and short-term prediction. The learned components accelerate the model and correct its biases, while the physics keeps the estimate anchored. As with all surrogates, the learned parts are validated against trusted models and flagged where observations leave the regime they were trained on.