Grid or Nodes? Probing Object Memory in World Models Under Partial Observability

Justin McMillen, Yasin Yilmaz

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 ve…