cs.LGMar 13, 2026

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

Authors: Andrew Kiruluta

Organizations: UC Berkeley School of Information

Abstract

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.

Figures & tables

Explore similar work

CardsList
  1. Amortized Low-Rank Adaptation for Model-Based Reinforcement Learning

    Sep 10, 2026Fernando Palafox, David Fridovich-KeilTest-Time AdaptationWorld Model Learning

  2. Learning, Fast and Slow: Towards LLMs That Adapt Continually

    May 12, 2026Rishabh Tiwari, Kusha Sareen, Lakshya A Agrawal +6Continual Learning for LLMsRL for Language Model Reasoning

  3. Stealthy World Model Manipulation via Data Poisoning

    Jun 17, 2026Yibin Hu, Xiaolin Sun, Zizhan ZhengWorld Model LearningAdversarial Attacks on World Models