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
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