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

• URL: http://arxiv.org/abs/2607.26345v1
• Date: Tue, 28 Jul 2026 23:37:55 GMT
• Title: MetaKoopman: Bayesian Meta-Learning of Koopman Operators for Modeling Structured Dynamics under Distribution Shifts
• Abstract summary: We propose MetaKoopman, a meta-learning framework for modeling nonlinear dynamics.<n>We evaluate MetaKoopman on a full-scale autonomous truck and trailer system.
• Score: 0.0
• License: http://creativecommons.org/licenses/by/4.0/
• Abstract: Modeling and forecasting nonlinear dynamics under distribution shifts is essential for robust decision-making in real-world systems. In this work, we propose MetaKoopman, a Bayesian meta-learning framework for modeling nonlinear dynamics through linear latent representations. MetaKoopman learns a Matrix Normal-Inverse Wishart (MNIW) prior over the Koopman operator, enabling closed-form Bayesian updates conditioned on recent trajectory segments. Moreover, it provides a closed-form posterior predictive distribution over future state trajectories, capturing both epistemic and aleatoric uncertainty in the learned dynamics. We evaluate MetaKoopman on a full-scale autonomous truck and trailer system across a wide range of adverse winter scenarios, including snow, ice, and mixed-friction conditions, as well as in simulated control tasks with diverse distribution shifts. MetaKoopman consistently outperforms prior approaches in multi-step prediction accuracy, uncertainty calibration, and robustness to distributional shifts. Field experiments further demonstrate its effectiveness in dynamically feasible motion planning, particularly during evasive maneuvers and operation at the limits of traction. Project website: https://mahmoud-selim.github.io/MetaKoopman/

Related papers

• DynFlowDrive: Flow-Based Dynamic World Modeling for Autonomous Driving [67.83404465171257]
  DynFlowDrive is a latent world model that models the transition of world states under different driving actions.<n>It learns a velocity field that describes how the scene state changes under different driving actions.<n>Building upon this, we introduce a stability-aware multi-mode trajectory selection strategy.
  arXiv  Detail & Related papers  (2026-03-20T06:19:31Z)
• Koopman Autoencoders with Continuous-Time Latent Dynamics for Fluid Dynamics Forecasting [17.98687936773676]
  We introduce a continuous-time Koopman framework that models latent evolution through numerical integration schemes.<n>By allowing variable timesteps at inference, the method demonstrates robustness to temporal resolution and generalizes beyond training regimes.<n>We evaluate the approach on classical CFD benchmarks and report accuracy, stability, and extrapolation properties.
  arXiv  Detail & Related papers  (2026-02-02T21:33:07Z)
• Learnable Koopman-Enhanced Transformer-Based Time Series Forecasting with Spectral Control [1.2604738912025477]
  We introduce four learnable Koopman variants-scalar-gated, per-mode, gated, linear-shaped spectral mapping, and low-rank Koopman operators.<n>Our formulation enables explicit control over the spectrum, stability, and rank of the linear transition operator.<n>Our results demonstrate that learnable Koopman operators are effective, stable, and theoretically principled components for deep forecasting.
  arXiv  Detail & Related papers  (2026-02-01T09:13:49Z)
• Conformal Online Learning of Deep Koopman Linear Embeddings [1.8577594866206437]
  COLoKe is a framework for adaptively updating Koopman-invariant representations from streaming data.<n> COLoKe employs a conformal-style mechanism that shifts the focus from evaluating the conformity of new states to assessing the consistency of the current Koopman model.
  arXiv  Detail & Related papers  (2025-11-16T20:08:48Z)
• KoopCast: Trajectory Forecasting via Koopman Operators [12.103961182264037]
  We present KoopCast, a lightweight model for trajectory forecasting in general dynamic environments.<n>We use Koopman operator theory, which enables a linear representation of nonlinear dynamics by lifting trajectories into a higher-dimensional space.<n>Our model offers three key advantages: (i) competitive accuracy, (ii) interpretability grounded in Koopman theory, and (iii) low-latency deployment.
  arXiv  Detail & Related papers  (2025-09-19T01:27:53Z)
• TrajBooster: Boosting Humanoid Whole-Body Manipulation via Trajectory-Centric Learning [79.59753528758361]
  We present TrajBooster, a cross-embodiment framework that leverages abundant wheeled-humanoid data to boost bipedal VLA.<n>Our key idea is to use end-effector trajectories as a morphology-agnostic interface.<n>Results show that TrajBooster allows existing wheeled-humanoid data to efficiently strengthen bipedal humanoid VLA performance.
  arXiv  Detail & Related papers  (2025-09-15T12:25:39Z)
• Koopa: Learning Non-stationary Time Series Dynamics with Koopman Predictors [85.22004745984253]
  Real-world time series are characterized by intrinsic non-stationarity that poses a principal challenge for deep forecasting models. We tackle non-stationary time series with modern Koopman theory that fundamentally considers the underlying time-variant dynamics. We propose Koopa as a novel Koopman forecaster composed of stackable blocks that learn hierarchical dynamics.
  arXiv  Detail & Related papers  (2023-05-30T07:40:27Z)
• Stochastic Trajectory Prediction via Motion Indeterminacy Diffusion [88.45326906116165]
  We present a new framework to formulate the trajectory prediction task as a reverse process of motion indeterminacy diffusion (MID) We encode the history behavior information and the social interactions as a state embedding and devise a Transformer-based diffusion model to capture the temporal dependencies of trajectories. Experiments on the human trajectory prediction benchmarks including the Stanford Drone and ETH/UCY datasets demonstrate the superiority of our method.
  arXiv  Detail & Related papers  (2022-03-25T16:59:08Z)
• Robust Value Iteration for Continuous Control Tasks [99.00362538261972]
  When transferring a control policy from simulation to a physical system, the policy needs to be robust to variations in the dynamics to perform well. We present Robust Fitted Value Iteration, which uses dynamic programming to compute the optimal value function on the compact state domain. We show that robust value is more robust compared to deep reinforcement learning algorithm and the non-robust version of the algorithm.
  arXiv  Detail & Related papers  (2021-05-25T19:48:35Z)
• Autoregressive Dynamics Models for Offline Policy Evaluation and Optimization [60.73540999409032]
  We show that expressive autoregressive dynamics models generate different dimensions of the next state and reward sequentially conditioned on previous dimensions. We also show that autoregressive dynamics models are useful for offline policy optimization by serving as a way to enrich the replay buffer.
  arXiv  Detail & Related papers  (2021-04-28T16:48:44Z)
• Forecasting Sequential Data using Consistent Koopman Autoencoders [52.209416711500005]
  A new class of physics-based methods related to Koopman theory has been introduced, offering an alternative for processing nonlinear dynamical systems. We propose a novel Consistent Koopman Autoencoder model which, unlike the majority of existing work, leverages the forward and backward dynamics. Key to our approach is a new analysis which explores the interplay between consistent dynamics and their associated Koopman operators.
  arXiv  Detail & Related papers  (2020-03-04T18:24:30Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.