Grid or Nodes? Probing Object Memory in World Models Under Partial Observability
cs.RO, cs.CV, cs.LG
Submitted: 2026-10-13
Updated: 2026-10-13
License: http://creativecommons.org/licenses/by/4.0/
The gist: A robot's world model must remember objects that leave the camera's view.
Terminology
Abstract
A robot's world model must remember objects that leave the camera's view. FloWM, a learned world model, keeps them in a spatial grid of features that it moves with five built-in velocities. A node memory instead keeps each object's position and velocity as numbers. We compare them in FloWM's own block world, with an oracle-fed Kalman tracker as the node memory. How the grid is read changes its score: a single-cell probe finds 66% of out-of-view blocks, while FloWM's own probe 89%. FloWM also scores 8 points higher in view although no frame is shown, which we attribute to its write-back of its own predictions. Out of view, the tracker reaches FloWM's range in 14KB against 5.6MB, with no probe. On diagonal motion, outside its five velocities, FloWM finds 11% of blocks even after fine-tuning, while the unchanged tracker keeps 93%. Faster blocks defeat both until one more allowed speed takes the tracker to 94%, where FloWM's built-in shifts would need 25 copies of its grid. A learned node memory must still replace the oracles and handle velocity under noise, long gaps, and the bounce.
Sources
- V-JEPA 2: Self-Supervised Video Models Enable Understanding, Prediction and Planning
- Memory Forcing: Spatio-Temporal Memory for Consistent Scene Generation on Minecraft
- LiveWorld: Simulating Out-of-Sight Dynamics in Generative Video World Models
- Out of Sight but Not Out of Mind: Hybrid Memory for Dynamic Video World Models
- Unsupervised Object-Based Transition Models for 3D Partially Observable Environments
- Current World Models Lack a Persistent State Core
- Out of Sight, Out of Mind? Evaluating State Evolution in Video World Models
- MemoBench: Benchmarking World Modeling in Dynamically Changing Environments
- LoopNav: Benchmarking Spatial Consistency in World Models
- Echo-Memory: A Controlled Study of Memory in Action World Models
Related papers
- FMT x: An Efficient and Asymptotically Optimal Extension of the Fast Marching Tree for Dynamic Replanning
- MPCFormer: A physics-informed data-driven approach for explainable socially-aware autonomous driving
- RoboLab: A High-Fidelity Simulation Benchmark for Analysis of Task Generalist Policies
- ARC: A Reasoning Recipe for Robot Foundation Models
- HRDexDB: A 4D Dexterous Grasping Dataset Across Human and Multiple Robot Embodiments
- APT: Action Expert Pretraining Improves Instruction Generalization of Vision-Language-Action Policies