MetaKoopman: Bayesian Meta-Learning of Koopman Operators for Modeling Structured Dynamics under Distribution Shifts

arXiv:2607.26345 2026 Dynamics 1 ideas extracted · analyzed Aug 31, 2026

What the math gives to ML

The paper's transferable core is the combination of Koopman linearization and conjugate Bayesian regression: a nonlinear encoder maps observations into a latent space where dynamics are linear, while a Matrix Normal-Inverse Wishart prior can be updated in closed form from a short trajectory segment. This gives a computationally cheap alternative to repeatedly adapting all neural-network weights under distribution shift, and it produces calibrated epistemic plus aleatoric uncertainty for multi-step rollouts. The most practical neural-network transfer is a meta-learned latent linear dynamics head whose prior is adapted online by sufficient-statistic updates, with the encoder trained across tasks.

Ideas from this paper

✓✓ Beats tuned baseline 2026

Conjugate Bayesian Latent Dynamics Head

Replace the final nonlinear transition network of a latent world model with a linear Koopman-style transition whose coefficients have a Matrix Normal-Inverse Wishart prior. Meta-learn the prior across tasks, then adapt only closed-form sufficient statistics from a few recent transitions; this should be more data-efficient and uncertainty-aware than gradient fine-tuning under distribution shift.

Useful7/10
Difficulty5/10
Novelty6/10
Paper: MetaKoopman: Bayesian Meta-Learning of Koopman Operators for Modeling Structured Dynamics under Distribution Shifts arXiv:2607.26345