cs.LGMay 29, 2026

The Dynamic-Probabilistic Consistency Gap in Chaotic Surrogate Modeling

Authors: Andre Herz, Matthijs Pals, Daniel Durstewitz, Georgia Koppe

Organizations: 1Interdisciplinary Center for Scientific Computing, Heidelberg University, Germany · Faculty of Mathematics and Computer Science, Heidelberg University, Germany · 3Dept. of Theoretical Neuroscience, Central Institute of Mental Health (CIMH), Mannheim, Germany · Faculty of Physics and Astronomy, Heidelberg University, Germany · 5Hector Institute for AI in Psychiatry and Dept. of Psychiatry and Psychotherapy, CIMH, Mannheim, Germany · 6Hertie Institute for AI in Brain Health, University of Tübingen, Germany

Abstract

Dynamical systems reconstruction (DSR) aims to learn surrogate models that capture the dynamics underlying time-series data. Reliably deploying these surrogates requires uncertainty estimates consistent with the learned dynamics. We expose a dynamic-probabilistic consistency (DPC) gap: the pursuit of finite-horizon probabilistic objectives can degrade dynamics or decouple predictive uncertainty from the local tangent dynamics it ought to reflect. We isolate three mechanisms behind this gap: core collapse, noise masking, and blind uncertainty. Specifically, we show that open-loop Gaussian rollout objectives can penalize Jacobian-generated covariance growth in chaotic systems, encouraging optimization shortcuts that weaken physical expansion or decouple uncertainty from it. To mitigate this gap, we propose KAFFEE (Kalman-Aware Framework For Ergodic Emulation), a differentiable extended Kalman filter-based training framework that evaluates likelihood on local predictive residuals (innovations) while transporting covariance through learned local Jacobians. On stochastic hyperchaotic Lorenz-96, KAFFEE reduces the identified failure modes, improves reconstruction of dynamical invariants relative to open-loop objectives, and maintains competitive predictive scores. We further show that the DPC gap appears when probabilistically adapting a DSR foundation model across 13 chaotic systems, where KAFFEE enables in-context Bayesian filtering while largely preserving zero-shot dynamics.

Explore similar work

CardsList
  1. Learning Chaotic Dynamics through Second-Order Geometric Supervision

    Jun 1, 2026Shinhoo Kang, Hai V. Nguyen, Tan Bui-ThanhChaosJacobian