mlsanityData to 14 Oct 2026

From Gradients to Riccati Geometry: Kalman World Models for Single-Pass Learning

arXiv 2603.13423 ↗First posted 13 Mar 2026cs.LG cs.CLcited by 0

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

  1. 2018World Models27 Mar

Other references · 12

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020Language Models are Few-Shot Learnerscited by 32,419
  3. 2016Overcoming catastrophic forgetting in neural networkscited by 4,787
  4. 2018Neural Ordinary Differential Equationscited by 3,779
  5. 2022Survey of Hallucination in Natural Language Generationcited by 2,075
  6. 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,492
  7. 2016Training Deep Nets with Sublinear Memory Costcited by 848
  8. 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 796
  9. 2017Deep learning for universal linear embeddings of nonlinear dynamicscited by 494
  10. 2019Latent ODEs for Irregularly-Sampled Time Seriescited by 183
  11. 2015Deep Kalman Filterscited by 166
  12. 2013Streaming Variational Bayescited by 48

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Filtering Beats Fine Tuning: A Bayesian Kalman View of In Context Learning in LLMs
  2. 2025Accelerating Model-Based Reinforcement Learning with State-Space World Models
  3. 2025Robotic World Model: A Neural Network Simulator for Robust Policy Optimization in Robotics
  4. 2026Neural Kalman Filtering for Unknown Dynamics: Task-Aware Learning with a Koopman Backbone ↗
  5. 2026MaRK: Markov-adapted Recurrent Kernels for Dynamic Operator Conditioning in State Space Models
  6. 2026WCM: A World Critic Model for Vision-Language-Action Reinforcement Learning
  7. 2026Masked Diffusion Language Models are Strong and Steerable Text-Based World Models for Agentic RL
  8. 2024Uncertainty Representations in State-Space Layers for Deep Reinforcement Learning under Partial Observability