From Gradients to Riccati Geometry: Kalman World Models for Single-Pass Learning
On the map: World models → Latent dynamics models and model-based RL
Abstract (arXiv)
Backpropagation dominates modern machine learning, yet it is not the only principled method for optimizing dynamical systems. We propose Kalman World Models (KWM), a class of learned state-space models trained via recursive Bayesian filtering rather than reverse-mode automatic differentiation. Instead of gradient descent updates, we replace parameter learning with Kalman-style gain adaptation. Training becomes online filtering; error signals become innovations. We further extend this framework to transformer-based large language models (LLMs), where internal activations are treated as latent dynamical states corrected via innovation terms. This yields a gradient-free training and adaptation paradigm grounded in control theory. We derive stability conditions, analyze computational complexity, and provide empirical results on sequence modeling tasks demonstrating competitive performance with improved robustness and continual adaptation properties.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Latent dynamics models and model-based RL, one of 11 directions in World models. Builds on 13 of its 40 references that are in the map. First posted on arXiv 13 Mar 2026.
Builds on
13 of 40 references.
Closest ancestors · 1
Same direction, newest first
- 2018World Models27 Mar
Other references · 12
- 2017Attention Is All You Needcited by 60,361
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2016Overcoming catastrophic forgetting in neural networkscited by 4,787
- 2018Neural Ordinary Differential Equationscited by 3,779
- 2022Survey of Hallucination in Natural Language Generationcited by 2,075
- 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,492
- 2016Training Deep Nets with Sublinear Memory Costcited by 848
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 796
- 2017Deep learning for universal linear embeddings of nonlinear dynamicscited by 494
- 2019Latent ODEs for Irregularly-Sampled Time Seriescited by 183
- 2015Deep Kalman Filterscited by 166
- 2013Streaming Variational Bayescited by 48
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Filtering Beats Fine Tuning: A Bayesian Kalman View of In Context Learning in LLMs
- 2025Accelerating Model-Based Reinforcement Learning with State-Space World Models
- 2025Robotic World Model: A Neural Network Simulator for Robust Policy Optimization in Robotics
- 2026Neural Kalman Filtering for Unknown Dynamics: Task-Aware Learning with a Koopman Backbone ↗
- 2026MaRK: Markov-adapted Recurrent Kernels for Dynamic Operator Conditioning in State Space Models
- 2026WCM: A World Critic Model for Vision-Language-Action Reinforcement Learning
- 2026Masked Diffusion Language Models are Strong and Steerable Text-Based World Models for Agentic RL
- 2024Uncertainty Representations in State-Space Layers for Deep Reinforcement Learning under Partial Observability